Merge branch 'master' into fix/nbeats-nhits-TODOs

unit8co · May 18, 2022 · 3cc47f1 · 3cc47f1
2 parents 19a1cc2 + 8d9bd19
commit 3cc47f1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/darts/models/filtering/gaussian_process_filter.py b/darts/models/filtering/gaussian_process_filter.py
@@ -71,4 +71,5 @@ def filter(self, series: TimeSeries, num_samples: int = 1) -> TimeSeries:
         else:
             filtered_values = self.model.sample_y(times, n_samples=num_samples)
 
+        filtered_values = filtered_values.reshape(len(times), -1, num_samples)
         return TimeSeries.from_times_and_values(series.time_index, filtered_values)