Bugfix: set_pump_speed

WaterFutures · Dec 31, 2024 · 7bedb01 · 7bedb01
1 parent 2eea39d
commit 7bedb01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/epyt_flow/gym/scenario_control_env.py b/epyt_flow/gym/scenario_control_env.py
@@ -194,7 +194,7 @@ def set_pump_speed(self, pump_id: str, speed: float) -> None:
         if pattern_idx == 0:
             warnings.warn(f"No pattern for pump '{pump_id}' found -- a new pattern is created")
             pattern_idx = self._scenario_sim.epanet_api.addPattern(f"pump_speed_{pump_id}")
-            self._scenario_sim.epanet_api.setLinkPumpPatternIndex(pump_idx, pattern_idx)
+            self._scenario_sim.epanet_api.setLinkPumpPatternIndex(pump_idx + 1, pattern_idx)
 
         self._scenario_sim.epanet_api.setPattern(pattern_idx, np.array([speed]))