Merge pull request #81 from Nixtla/fix/input_size_check

[FIX] Use min size
Nixtla · Aug 16, 2023 · dc4f524 · dc4f524
2 parents f390d9c + 8b87d2c
commit dc4f524
Show file tree

Hide file tree

Showing 2 changed files with 535 additions and 52 deletions.
diff --git a/nbs/timegpt.ipynb b/nbs/timegpt.ipynb
diff --git a/nixtlats/timegpt.py b/nixtlats/timegpt.py
@@ -329,7 +329,7 @@ def _validate_input_size(
         require_history: bool,
     ):
         if require_history:
-            min_history = Y_df.groupby("unique_id").size().mean()
+            min_history = Y_df.groupby("unique_id").size().min()
             if min_history < input_size + model_horizon:
                 raise Exception(
                     "Your time series data is too short "