13 lines
608 B
Python
13 lines
608 B
Python
from legged_gym import LEGGED_GYM_ROOT_DIR, LEGGED_GYM_ENVS_DIR
|
|
|
|
from legged_gym.envs.go2.go2_config import GO2RoughCfg, GO2RoughCfgPPO
|
|
from legged_gym.envs.h1.h1_config import H1RoughCfg, H1RoughCfgPPO
|
|
from legged_gym.envs.h1_2.h1_2_config import H1_2RoughCfg, H1_2RoughCfgPPO
|
|
from .base.legged_robot import LeggedRobot
|
|
|
|
from legged_gym.utils.task_registry import task_registry
|
|
|
|
task_registry.register("go2", LeggedRobot, GO2RoughCfg(), GO2RoughCfgPPO())
|
|
task_registry.register("h1", LeggedRobot, H1RoughCfg(), H1RoughCfgPPO())
|
|
task_registry.register("h1_2", LeggedRobot, H1_2RoughCfg(), H1_2RoughCfgPPO())
|