chore: Update RobustHopper environment in evaluate.py

SuReLI · May 12, 2024 · 9073d13 · 9073d13
1 parent e875c0d
commit 9073d13
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/rrls/evaluate.py b/rrls/evaluate.py
@@ -14,6 +14,7 @@
     InvertedPendulumParamsBound,
     RobustAnt,
     RobustHalfCheetah,
+    RobustHopper,
     RobustHumanoidStandUp,
     RobustInvertedPendulum,
     RobustWalker2d,
@@ -139,19 +140,19 @@ def generate_evaluation_set(
 )
 
 EVALUATION_ROBUST_HOPPER_1D = generate_evaluation_set(
-    modified_env=RobustWalker2d,  # type: ignore
+    modified_env=RobustHopper,  # type: ignore
     param_bounds=HopperParamsBound.ONE_DIM.value,
     nb_mesh_dim=10,
 )
 
 EVALUATION_ROBUST_HOPPER_2D = generate_evaluation_set(
-    modified_env=RobustWalker2d,  # type: ignore
+    modified_env=RobustHopper,  # type: ignore
     param_bounds=HopperParamsBound.TWO_DIM.value,
     nb_mesh_dim=10,
 )
 
 EVALUATION_ROBUST_HOPPER_3D = generate_evaluation_set(
-    modified_env=RobustWalker2d,  # type: ignore
+    modified_env=RobustHopper,  # type: ignore
     param_bounds=HopperParamsBound.THREE_DIM.value,
     nb_mesh_dim=10,
 )