fix

awslabs · Dec 5, 2023 · 0ad33c2 · 0ad33c2
1 parent da9dc3a
commit 0ad33c2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/gluonts/ext/rotbaum/_preprocess.py b/src/gluonts/ext/rotbaum/_preprocess.py
@@ -452,8 +452,12 @@ def make_features(self, time_series: Dict, starting_index: int) -> List:
         end_index = starting_index + self.context_window_size
         if starting_index < 0:
             prefix = [None] * abs(starting_index)
+            time_series_window = time_series["target"]
         else:
             prefix = []
+            time_series_window = time_series["target"][
+                starting_index:end_index
+            ]
         time_series_window = time_series["target"][starting_index:end_index]
         only_lag_features, transform_dict = self._pre_transform(
             time_series_window, self.subtract_mean, self.count_nans
@@ -480,10 +484,10 @@ def make_features(self, time_series: Dict, starting_index: int) -> List:
             list(
                 chain(
                     *[
-                        list(ent[0]) + list(ent[1].values())
+                        prefix + list(ent[0]) + list(ent[1].values())
                         for ent in [
                             self._pre_transform(
-                                ts[starting_index:end_index],
+                                ts if prefix else ts[starting_index:end_index],
                                 self.subtract_mean,
                                 self.count_nans,
                             )