Test: Clean superfluous text

Signed-off-by: Xavier Weiss <[email protected]>
Grid2op · Nov 22, 2024 · 3e47d8b · 3e47d8b
1 parent 35f279f
commit 3e47d8b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 8 deletions.
diff --git a/grid2op/tests/_aux_test_gym_compat.py b/grid2op/tests/_aux_test_gym_compat.py
@@ -321,9 +321,6 @@ def test_chain_converter(self):
         else:
             # it's linux
             if GYM_VERSION <= _MAX_GYM_VERSION_RANDINT:
-                res_disp = np.array(
-                    [-0.8333335,-1.666667,0.,0.,0.,2.5], dtype=dt_float
-                )
                 res_disp = np.array(
                     [-0.8333335,-1.666667,0.,0.,0.,2.5], dtype=dt_float
                 )
@@ -333,10 +330,6 @@ def test_chain_converter(self):
                 res_disp = np.array(
                     [-0.8333335,-1.666667,0.,0.,0.,2.5], dtype=dt_float
                 )
-                res_disp = np.array(
-                    [-0.8333335,-1.666667,0.,0.,0.,2.5], dtype=dt_float
-                )
-                res_tup = (4, 4, 0, 0, 0, 6)
         assert (
             act_gym["redispatch"] == res_tup
         ), f'error. redispatch is {act_gym["redispatch"]}'

diff --git a/grid2op/tests/test_issue_418.py b/grid2op/tests/test_issue_418.py
@@ -37,7 +37,7 @@ def test_seed(self):
             obs = gymenv.reset(seed=42)
             curt = np.array([1,1.,0.18852758,0.5537014,0.43770432,1])
             curt = np.array([-1,-1.,0.18852758,0.5537014,0.43770432,-1])
-            if "actual_flex" in self.env.observation_space.attr_list_vect:
+            if "actual_flex" in gymenv.observation_space.spaces:
                 year = 229
                 day = 13
             else: