enforce positive

Nixtla · Nov 11, 2024 · 7367581 · 7367581
1 parent 328c7b5
commit 7367581
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mlforecast/core.py b/mlforecast/core.py
@@ -438,7 +438,7 @@ def _transform(
         if (
             self.keep_last_n is None
             and update_samples
-            and all(samples > -1 for samples in update_samples)
+            and all(samples > 0 for samples in update_samples)
         ):
             # user didn't set keep_last_n and we can infer it from the transforms
             self.keep_last_n = max(update_samples)

diff --git a/nbs/core.ipynb b/nbs/core.ipynb
@@ -931,7 +931,7 @@
     "        if (\n",
     "            self.keep_last_n is None\n",
     "            and update_samples\n",
-    "            and all(samples > -1 for samples in update_samples)\n",
+    "            and all(samples > 0 for samples in update_samples)\n",
     "        ):\n",
     "            # user didn't set keep_last_n and we can infer it from the transforms\n",
     "            self.keep_last_n = max(update_samples)\n",