2020from grid2op .Runner import Runner
2121from grid2op .dtypes import dt_float
2222
23- warnings .simplefilter ("error" )
24-
2523
2624class TestRunner (HelperTests , unittest .TestCase ):
2725 def setUp (self ):
@@ -56,7 +54,7 @@ def setUp(self):
5654 def test_one_episode (self ):
5755 with warnings .catch_warnings ():
5856 warnings .filterwarnings ("ignore" )
59- _ , cum_reward , timestep , max_ts = self .runner .run_one_episode (
57+ _ , _ , cum_reward , timestep , max_ts = self .runner .run_one_episode (
6058 max_iter = self .max_iter
6159 )
6260 assert int (timestep ) == self .max_iter
@@ -65,7 +63,7 @@ def test_one_episode(self):
6563 def test_one_episode_detailed (self ):
6664 with warnings .catch_warnings ():
6765 warnings .filterwarnings ("ignore" )
68- _ , cum_reward , timestep , max_ts , episode_data = self .runner .run_one_episode (
66+ _ , _ , cum_reward , timestep , max_ts , episode_data = self .runner .run_one_episode (
6967 max_iter = self .max_iter , detailed_output = True
7068 )
7169 assert int (timestep ) == self .max_iter
0 commit comments