fixing broken tests again (-:

Grid2op · May 3, 2024 · cc9fd62 · cc9fd62
1 parent 8c3448e
commit cc9fd62
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/grid2op/tests/test_RunnerFast.py b/grid2op/tests/test_RunnerFast.py
@@ -20,8 +20,6 @@
 from grid2op.Runner import Runner
 from grid2op.dtypes import dt_float
 
-warnings.simplefilter("error")
-
 
 class TestRunner(HelperTests, unittest.TestCase):
     def setUp(self):
@@ -56,7 +54,7 @@ def setUp(self):
     def test_one_episode(self):
         with warnings.catch_warnings():
             warnings.filterwarnings("ignore")
-            _, cum_reward, timestep, max_ts = self.runner.run_one_episode(
+            _, _, cum_reward, timestep, max_ts = self.runner.run_one_episode(
             max_iter=self.max_iter
         )
         assert int(timestep) == self.max_iter
@@ -65,7 +63,7 @@ def test_one_episode(self):
     def test_one_episode_detailed(self):
         with warnings.catch_warnings():
             warnings.filterwarnings("ignore")
-            _, cum_reward, timestep, max_ts, episode_data = self.runner.run_one_episode(
+            _, _, cum_reward, timestep, max_ts, episode_data = self.runner.run_one_episode(
             max_iter=self.max_iter, detailed_output=True
         )
         assert int(timestep) == self.max_iter