Bug fix pop fails when timesteps_pfa is not provided

openego · Oct 18, 2023 · f50460c · f50460c
1 parent 8df403d
commit f50460c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/edisgo/flex_opt/reinforce_grid.py b/edisgo/flex_opt/reinforce_grid.py
@@ -699,8 +699,7 @@ def reinforce():
     logger.debug("Start 'catch-convergence' reinforcement.")
 
     # Get the timesteps from kwargs and then remove it to set it later manually
-    timesteps_pfa = kwargs.get("timesteps_pfa")
-    kwargs.pop("timesteps_pfa")
+    timesteps_pfa = kwargs.pop("timesteps_pfa", None)
     selected_timesteps = timesteps_pfa
 
     # Initial try