diff --git a/views_stepshifter/models/hurdle_model.py b/views_stepshifter/models/hurdle_model.py index e8845d5..bf60c40 100644 --- a/views_stepshifter/models/hurdle_model.py +++ b/views_stepshifter/models/hurdle_model.py @@ -157,7 +157,7 @@ def predict(self, run_type: str, eval_type: str = "standard") -> pd.DataFrame: ForecastingModelManager._resolve_evaluation_sequence_number(eval_type) ) if self.get_device_params().get("device") == "cuda": - pred = [] + preds = [] for sequence_number in tqdm.tqdm( range(ForecastingModelManager._resolve_evaluation_sequence_number(eval_type)), desc="Predicting for sequence number",