quantile_maybe_used

Nixtla · Oct 17, 2024 · 8ee4592 · 8ee4592
1 parent a4c8b54
commit 8ee4592
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/nbs/common.base_model.ipynb b/nbs/common.base_model.ipynb
@@ -1333,6 +1333,11 @@
     "        \"\"\"\n",
     "        self._check_exog(dataset)\n",
     "        self._restart_seed(random_seed)\n",
+    "        if \"quantile\" in data_module_kwargs:\n",
+    "            warnings.warn(\"The 'quantile' argument will be deprecated, use 'quantiles' instead.\")\n",
+    "            if quantiles is not None:\n",
+    "                raise ValueError(\"You can't specify quantile and quantiles.\")\n",
+    "            quantiles = [data_module_kwargs.pop(\"quantile\")]\n",
     "        self._set_quantiles(quantiles)\n",
     "\n",
     "        self.predict_step_size = step_size\n",

diff --git a/neuralforecast/common/_base_model.py b/neuralforecast/common/_base_model.py
@@ -1465,6 +1465,13 @@ def predict(
         """
         self._check_exog(dataset)
         self._restart_seed(random_seed)
+        if "quantile" in data_module_kwargs:
+            warnings.warn(
+                "The 'quantile' argument will be deprecated, use 'quantiles' instead."
+            )
+            if quantiles is not None:
+                raise ValueError("You can't specify quantile and quantiles.")
+            quantiles = [data_module_kwargs.pop("quantile")]
         self._set_quantiles(quantiles)
 
         self.predict_step_size = step_size