adjust_slicing

MaiBe-ctrl · MaiBe-ctrl · commit 1ca20853768e · 2024-08-13T10:59:45.000-07:00
diff --git a/neuralprophet/time_dataset.py b/neuralprophet/time_dataset.py
@@ -158,9 +158,9 @@ def get_sample_seasonalities(self, df_tensors, origin_index, n_forecasts, max_la
 
         # Determine the range of indices based on whether lags are used
         if max_lags == 0:
-            indices = [origin_index]
+            indices = slice(origin_index, origin_index + 1)
         else:
-            indices = list(range(origin_index - n_lags + 1, origin_index + n_forecasts + 1))
+            indices = slice(origin_index - n_lags + 1, origin_index + n_forecasts + 1)
 
         # Extract the precomputed seasonalities from self.seasonalities
         for name, features in self.seasonalities.items():
