fixing broken test for the Runner - bad import

BDonnot · Jun 4, 2021 · 53cfa3d · 53cfa3d
1 parent ea9c09c
commit 53cfa3d
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -30,6 +30,9 @@ Change Log
   than just boolean.
 - [BREAKING] now the observation stores the information related to shunts by default. This means old logs computed with
   the runner might not work with this new version.
+- [BREAKING] the "Runner.py" file has been renamed, following pep convention "runner.py". You should rename your
+  import `from grid2op.Runner.Runner import Runner` to `from grid2op.Runner.runner import Runner`
+  (**NB** we higly recommend importing the `Runner` like `from grid2op.Runner import Runner` though !)
 - [FIXED]: some bugs in the `action_space.get_all_unitary_redispatch` and `action_space.get_all_unitary_curtail`
 - [FIXED]: some bugs in the `GreedyAgent` and `TopologyGreedy`
 - [ADDED] support for the "alarm operator" / "attention budget" feature

diff --git a/grid2op/tests/test_AlarmFeature.py b/grid2op/tests/test_AlarmFeature.py
@@ -423,6 +423,7 @@ def test_runner(self):
             ep_dat = EpisodeData.from_disk(agent_path=f, name=res[0][1])
             assert len(ep_dat) == 10
             assert ep_dat.observations[0].attention_budget == 3
+            assert ep_dat.observations[1].attention_budget == 3 + 1. / (12. * 8.)
 
     def test_kwargs(self):
         """test the get_kwargs function properly foward the attention budget"""

diff --git a/grid2op/tests/test_Runner.py b/grid2op/tests/test_Runner.py
@@ -20,7 +20,7 @@
 from grid2op.Reward import L2RPNReward
 from grid2op.Backend import PandaPowerBackend
 from grid2op.MakeEnv import make
-from grid2op.Runner.Runner import _aux_one_process_parrallel
+from grid2op.Runner.aux_fun import _aux_one_process_parrallel
 from grid2op.Runner import Runner
 from grid2op.dtypes import dt_float
 from grid2op.Agent import RandomAgent

diff --git a/grid2op/tests/test_RunnerFast.py b/grid2op/tests/test_RunnerFast.py
@@ -16,9 +16,6 @@
 PATH_PREVIOUS_RUNNER = os.path.join(data_dir, "runner_data")
 
 import grid2op
-from grid2op.Chronics import Multifolder
-from grid2op.Reward import L2RPNReward
-from grid2op.Backend import PandaPowerBackend
 from grid2op.MakeEnv import make
 from grid2op.Runner import Runner
 from grid2op.dtypes import dt_float