Fix a bug

Skylark0924 · Nov 23, 2023 · d3f9a9b · d3f9a9b
1 parent 79c8555
commit d3f9a9b
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 14 deletions.
diff --git 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
@@ -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
@@ -18,7 +18,7 @@
 from rofunc.learning.RofuncRL.tasks.isaacgym.hotu.humanoid_hotu import HumanoidHOTU
 
 
-class HumanoidViewMotionTask(HumanoidHOTU):
+class HumanoidHOTUViewMotionTask(HumanoidHOTU):
     def __init__(
             self,
             cfg,