Skip to content

Commit 2bc55f8

Browse files
committed
Debug: Fix collection_wrapper_after_run test
1 parent d67b516 commit 2bc55f8

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

grid2op/tests/test_CompactEpisodeData.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -181,8 +181,11 @@ def test_collection_wrapper_after_run(self):
181181
agentClass=OneChange,
182182
use_compact_episode_data=True,
183183
)
184-
*_, episode_data = runner.run_one_episode(
185-
max_iter=self.max_iter, detailed_output=True,
184+
with warnings.catch_warnings():
185+
warnings.filterwarnings("ignore")
186+
*_, episode_data = runner.run_one_episode(
187+
max_iter=self.max_iter, detailed_output=True,
188+
)
186189
# Check that the type of first action is set bus
187190
assert episode_data.action_space.from_vect(episode_data.actions[0]).get_types()[2]
188191

@@ -298,4 +301,4 @@ def test_can_return_ep_data(self):
298301

299302

300303
if __name__ == "__main__":
301-
unittest.main()
304+
unittest.main()

0 commit comments

Comments
 (0)