🎨 Cleans up the environments

rickstaa · Apr 11, 2021 · acd28f5 · acd28f5
1 parent 7b7dbe0
commit acd28f5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/simzoo/envs/classic_control/cart_pole_cost/cart_pole_cost.py b/simzoo/envs/classic_control/cart_pole_cost/cart_pole_cost.py
@@ -198,7 +198,7 @@ def __init__(
         self.observation_space = spaces.Box(-obs_high, obs_high, dtype=np.float32)
         self.reward_range = spaces.Box(
             np.array([0.0], dtype=np.float32),
-            np.array([np.finfo(np.float32).max], dtype=np.float32),
+            np.array([100], dtype=np.float32),
             dtype=np.float32,
         )
 

diff --git a/simzoo/envs/classic_control/cart_pole_cost/cart_pole_disturber.py b/simzoo/envs/classic_control/cart_pole_cost/cart_pole_disturber.py
@@ -43,9 +43,8 @@
         # The range of values you want to use for each disturbance iteration
         "variable_range": np.linspace(0.5, 2.0, num=5, dtype=np.float32),
         # Label used in robustness plots.
-        "label": "r: %s",
+        "label": "l: %s",
     },
-    "test": {},
 }