diff --git a/nbs/common.base_auto.ipynb b/nbs/common.base_auto.ipynb index 516878c87..788c1d446 100644 --- a/nbs/common.base_auto.ipynb +++ b/nbs/common.base_auto.ipynb @@ -62,6 +62,7 @@ "outputs": [], "source": [ "#| export\n", + "import warnings\n", "from copy import deepcopy\n", "from os import cpu_count\n", "\n", @@ -169,7 +170,11 @@ " callbacks=None,\n", " ):\n", " super(BaseAuto, self).__init__()\n", - " self.save_hyperparameters() # Allows instantiation from a checkpoint from class\n", + " with warnings.catch_warnings(record=False):\n", + " warnings.filterwarnings('ignore')\n", + " # the following line issues a warning about the loss attribute being saved\n", + " # but we do want to save it\n", + " self.save_hyperparameters() # Allows instantiation from a checkpoint from class\n", "\n", " if backend == 'ray':\n", " if not isinstance(config, dict):\n", diff --git a/neuralforecast/common/_base_auto.py b/neuralforecast/common/_base_auto.py index f566a9e92..ba1bafede 100644 --- a/neuralforecast/common/_base_auto.py +++ b/neuralforecast/common/_base_auto.py @@ -4,6 +4,7 @@ __all__ = ['BaseAuto'] # %% ../../nbs/common.base_auto.ipynb 5 +import warnings from copy import deepcopy from os import cpu_count @@ -101,7 +102,11 @@ def __init__( callbacks=None, ): super(BaseAuto, self).__init__() - self.save_hyperparameters() # Allows instantiation from a checkpoint from class + with warnings.catch_warnings(record=False): + warnings.filterwarnings("ignore") + # the following line issues a warning about the loss attribute being saved + # but we do want to save it + self.save_hyperparameters() # Allows instantiation from a checkpoint from class if backend == "ray": if not isinstance(config, dict):