diff --git a/strategy.py b/strategy.py index d13750b..9992f81 100644 --- a/strategy.py +++ b/strategy.py @@ -91,7 +91,7 @@ def run(self, env: simpy.Environment, duration: int): if self.interruptible: # Get the points in time with the lowest CI - worst_ci_to_include = forecast.sort_values()[duration_steps - 1] + worst_ci_to_include = forecast.sort_values().iloc[duration_steps - 1] # Get their indices in the forecast dataframe lowest_indices = np.where(forecast <= worst_ci_to_include)[0] last_index = None