unitree_rl_gym/legged_gym/envs/__init__.py

11 lines
454 B
Python
Raw Normal View History

2023-10-11 15:38:49 +08:00
from legged_gym import LEGGED_GYM_ROOT_DIR, LEGGED_GYM_ENVS_DIR
from legged_gym.envs.go2.go2_config import GO2RoughCfg, GO2RoughCfgPPO
2024-01-12 10:18:20 +08:00
from legged_gym.envs.h1.h1_config import H1RoughCfg, H1RoughCfgPPO
2023-10-11 15:38:49 +08:00
from .base.legged_robot import LeggedRobot
from legged_gym.utils.task_registry import task_registry
2024-07-17 16:39:25 +08:00
task_registry.register("go2", LeggedRobot, GO2RoughCfg(), GO2RoughCfgPPO())
task_registry.register("h1", LeggedRobot, H1RoughCfg(), H1RoughCfgPPO())