Add check for requires_even_sampling in forecast() (#109)

salesforce · Jun 28, 2022 · e0a79f0 · e0a79f0
1 parent 593ecb9
commit e0a79f0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/merlion/models/forecast/base.py b/merlion/models/forecast/base.py
@@ -119,6 +119,10 @@ def resample_time_stamps(self, time_stamps: Union[int, List[int]], time_series_p
             tf = resampled[-1]
             time_stamps = to_timestamp(resampled)
 
+        elif not self.require_even_sampling:
+            resampled = to_pd_datetime(time_stamps)
+            tf = resampled[-1]
+
         # Handle the cases where we don't have a max_forecast_steps
         elif self.max_forecast_steps is None:
             tf = to_pd_datetime(time_stamps[-1])