File size: 1,125 Bytes
079c32c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from easydict import EasyDict
from . import gym_bipedalwalker_v3
from . import gym_halfcheetah_v3
from . import gym_hopper_v3
from . import gym_lunarlandercontinuous_v2
from . import gym_pendulum_v1
from . import gym_walker2d_v3

supported_env_cfg = {
    gym_bipedalwalker_v3.cfg.env.env_id: gym_bipedalwalker_v3.cfg,
    gym_halfcheetah_v3.cfg.env.env_id: gym_halfcheetah_v3.cfg,
    gym_hopper_v3.cfg.env.env_id: gym_hopper_v3.cfg,
    gym_lunarlandercontinuous_v2.cfg.env.env_id: gym_lunarlandercontinuous_v2.cfg,
    gym_pendulum_v1.cfg.env.env_id: gym_pendulum_v1.cfg,
    gym_walker2d_v3.cfg.env.env_id: gym_walker2d_v3.cfg,
}

supported_env_cfg = EasyDict(supported_env_cfg)

supported_env = {
    gym_bipedalwalker_v3.cfg.env.env_id: gym_bipedalwalker_v3.env,
    gym_halfcheetah_v3.cfg.env.env_id: gym_halfcheetah_v3.env,
    gym_hopper_v3.cfg.env.env_id: gym_hopper_v3.env,
    gym_lunarlandercontinuous_v2.cfg.env.env_id: gym_lunarlandercontinuous_v2.env,
    gym_pendulum_v1.cfg.env.env_id: gym_pendulum_v1.env,
    gym_walker2d_v3.cfg.env.env_id: gym_walker2d_v3.env,
}

supported_env = EasyDict(supported_env)