Fix context space

automl · Dec 11, 2023 · 778e383 · 778e383
1 parent 134f161
commit 778e383
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/carl/envs/gymnax/classic_control/carl_gymnax_mountaincar.py b/carl/envs/gymnax/classic_control/carl_gymnax_mountaincar.py
@@ -39,13 +39,13 @@ class CARLGymnaxMountainCarContinuous(CARLGymnaxMountainCar):
     def get_context_features() -> dict[str, ContextFeature]:
         return {
             "max_speed": UniformFloatContextFeature(
-                "max_speed", lower=0, upper=np.inf, default_value=0.07
+                "max_speed", lower=1e-3, upper=10, default_value=0.07
             ),
             "goal_position": UniformFloatContextFeature(
-                "goal_position", lower=-np.inf, upper=np.inf, default_value=0.45
+                "goal_position", lower=-2, upper=2, default_value=0.45
             ),
             "goal_velocity": UniformFloatContextFeature(
-                "goal_velocity", lower=-np.inf, upper=np.inf, default_value=0
+                "goal_velocity", lower=-10, upper=10, default_value=0
             ),
             "power": UniformFloatContextFeature(
                 "power", lower=1e-6, upper=10, default_value=0.001