from easydict import EasyDict
from . import gym_lunarlander_v2
from . import gym_pongnoframeskip_v4
from . import gym_qbertnoframeskip_v4
from . import gym_spaceInvadersnoframeskip_v4

supported_env_cfg = {
    gym_lunarlander_v2.cfg.env.env_id: gym_lunarlander_v2.cfg,
    gym_pongnoframeskip_v4.cfg.env.env_id: gym_pongnoframeskip_v4.cfg,
    gym_qbertnoframeskip_v4.cfg.env.env_id: gym_qbertnoframeskip_v4.cfg,
    gym_spaceInvadersnoframeskip_v4.cfg.env.env_id: gym_spaceInvadersnoframeskip_v4.cfg,
}

supported_env_cfg = EasyDict(supported_env_cfg)

supported_env = {
    gym_lunarlander_v2.cfg.env.env_id: gym_lunarlander_v2.env,
    gym_pongnoframeskip_v4.cfg.env.env_id: gym_pongnoframeskip_v4.env,
    gym_qbertnoframeskip_v4.cfg.env.env_id: gym_qbertnoframeskip_v4.env,
    gym_spaceInvadersnoframeskip_v4.cfg.env.env_id: gym_spaceInvadersnoframeskip_v4.env,
}

supported_env = EasyDict(supported_env)