diff --git a/legged_gym/envs/__init__.py b/legged_gym/envs/__init__.py index f87d9b2..550f533 100644 --- a/legged_gym/envs/__init__.py +++ b/legged_gym/envs/__init__.py @@ -1,4 +1,3 @@ - from legged_gym import LEGGED_GYM_ROOT_DIR, LEGGED_GYM_ENVS_DIR from legged_gym.envs.go2.go2_config import GO2RoughCfg, GO2RoughCfgPPO @@ -6,7 +5,6 @@ from legged_gym.envs.h1.h1_config import H1RoughCfg, H1RoughCfgPPO 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("go2", LeggedRobot, GO2RoughCfg(), GO2RoughCfgPPO()) +task_registry.register("h1", LeggedRobot, H1RoughCfg(), H1RoughCfgPPO())