From ec8d80bab900d654358ddce06097d2d08ab9b84b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jos=C3=A9=20Morales?= Date: Mon, 27 May 2024 10:45:06 -0600 Subject: [PATCH] propagate model alias to fallback model in fit --- nbs/src/core/core.ipynb | 1 + statsforecast/core.py | 1 + 2 files changed, 2 insertions(+) diff --git a/nbs/src/core/core.ipynb b/nbs/src/core/core.ipynb index c1f8b294..0adcfc5b 100644 --- a/nbs/src/core/core.ipynb +++ b/nbs/src/core/core.ipynb @@ -174,6 +174,7 @@ " except Exception as error:\n", " if fallback_model is not None:\n", " new_fallback_model = fallback_model.new()\n", + " new_fallback_model.alias = model.alias\n", " fm[i, i_model] = new_fallback_model.fit(y=y, X=X)\n", " else:\n", " raise error\n", diff --git a/statsforecast/core.py b/statsforecast/core.py index 3d4dd41e..9416aa84 100644 --- a/statsforecast/core.py +++ b/statsforecast/core.py @@ -62,6 +62,7 @@ def fit(self, models, fallback_model=None): except Exception as error: if fallback_model is not None: new_fallback_model = fallback_model.new() + new_fallback_model.alias = model.alias fm[i, i_model] = new_fallback_model.fit(y=y, X=X) else: raise error