ScenarioControlEnv: Stop running simulation in "reset" if necessary

WaterFutures · Jan 14, 2025 · df8024d · df8024d
1 parent 3372ee0
commit df8024d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/epyt_flow/gym/scenario_control_env.py b/epyt_flow/gym/scenario_control_env.py
@@ -107,6 +107,14 @@ def reset(self) -> Union[tuple[ScadaData, bool], ScadaData]:
             Current SCADA data (i.e. sensor readings).
         """
         if self._scenario_sim is not None:
+            # Abort current simulation if any is runing
+            try:
+                next(self._sim_generator)
+                self._sim_generator.send(True)
+            except StopIteration:
+                pass
+
+            # Close scenario
             self._scenario_sim.close()
 
         self._scenario_sim = ScenarioSimulator(