diff --git a/rofunc/learning/RofuncRL/tasks/__init__.py b/rofunc/learning/RofuncRL/tasks/__init__.py index 23f01bd16..99d147275 100644 --- a/rofunc/learning/RofuncRL/tasks/__init__.py +++ b/rofunc/learning/RofuncRL/tasks/__init__.py @@ -18,8 +18,8 @@ def __init__(self, env_type="isaacgym"): from .isaacgym.ase.humanoid_location import HumanoidLocationTask from .isaacgym.ase.humanoid_reach import HumanoidReachTask from .isaacgym.ase.humanoid_strike import HumanoidStrikeTask - from .isaacgym.ase.humanoid_view_motion import HumanoidViewMotionTask - from .isaacgym.hotu.humanoid_view_motion import HumanoidViewMotionTask + from .isaacgym.ase.humanoid_view_motion import HumanoidASEViewMotionTask + from .isaacgym.hotu.humanoid_view_motion import HumanoidHOTUViewMotionTask self.task_map = { "Ant": AntTask, @@ -38,8 +38,8 @@ def __init__(self, env_type="isaacgym"): "HumanoidASELocationSwordShield": HumanoidLocationTask, "HumanoidASEReachSwordShield": HumanoidReachTask, "HumanoidASEStrikeSwordShield": HumanoidStrikeTask, - "HumanoidASEViewMotion": HumanoidViewMotionTask, - "HumanoidHOTUViewMotion": HumanoidViewMotionTask, + "HumanoidASEViewMotion": HumanoidASEViewMotionTask, + "HumanoidHOTUViewMotion": HumanoidHOTUViewMotionTask, } elif env_type == "omniisaacgym": # OmniIsaacGym tasks diff --git a/rofunc/learning/RofuncRL/tasks/isaacgym/ase/humanoid_view_motion.py b/rofunc/learning/RofuncRL/tasks/isaacgym/ase/humanoid_view_motion.py index 35d2123e9..1bf965fb0 100644 --- a/rofunc/learning/RofuncRL/tasks/isaacgym/ase/humanoid_view_motion.py +++ b/rofunc/learning/RofuncRL/tasks/isaacgym/ase/humanoid_view_motion.py @@ -33,16 +33,16 @@ from rofunc.learning.RofuncRL.tasks.isaacgym.ase.humanoid_amp import HumanoidAMP -class HumanoidViewMotionTask(HumanoidAMP): +class HumanoidASEViewMotionTask(HumanoidAMP): def __init__( - self, - cfg, - rl_device, - sim_device, - graphics_device_id, - headless, - virtual_screen_capture, - force_render, + self, + cfg, + rl_device, + sim_device, + graphics_device_id, + headless, + virtual_screen_capture, + force_render, ): self.cfg = cfg control_freq_inv = cfg["env"]["controlFrequencyInv"] diff --git a/rofunc/learning/RofuncRL/tasks/isaacgym/hotu/humanoid_view_motion.py b/rofunc/learning/RofuncRL/tasks/isaacgym/hotu/humanoid_view_motion.py index 4725f199a..873019766 100644 --- a/rofunc/learning/RofuncRL/tasks/isaacgym/hotu/humanoid_view_motion.py +++ b/rofunc/learning/RofuncRL/tasks/isaacgym/hotu/humanoid_view_motion.py @@ -19,7 +19,7 @@ from rofunc.learning.RofuncRL.tasks.isaacgym.hotu.humanoid_hotu import HumanoidHOTU -class HumanoidViewMotionTask(HumanoidHOTU): +class HumanoidHOTUViewMotionTask(HumanoidHOTU): def __init__( self, cfg,