fix depracation clause

Nixtla · Oct 31, 2024 · edf1ecb · edf1ecb
1 parent b5861d6
commit edf1ecb
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/nbs/common.base_model.ipynb b/nbs/common.base_model.ipynb
@@ -368,7 +368,7 @@
     "                f\"`dataloader_kwargs={{'num_workers': {self.num_workers_loader}}}`\",\n",
     "                category=FutureWarning,\n",
     "            )\n",
-    "        dataloader_kwargs['num_workers'] = self.num_workers_loader\n",
+    "            dataloader_kwargs['num_workers'] = self.num_workers_loader\n",
     "\n",
     "        datamodule = datamodule_constructor(\n",
     "            dataset=dataset, \n",

diff --git a/nbs/common.base_recurrent.ipynb b/nbs/common.base_recurrent.ipynb
@@ -550,7 +550,7 @@
     "                f\"`dataloader_kwargs={{'num_workers': {self.num_workers_loader}}}`\",\n",
     "                category=FutureWarning,\n",
     "            )         \n",
-    "        data_module_kwargs['num_workers'] = self.num_workers_loader\n",
+    "            data_module_kwargs['num_workers'] = self.num_workers_loader\n",
     "        \n",
     "        if step_size > 1:\n",
     "            raise Exception('Recurrent models do not support step_size > 1')\n",

diff --git a/neuralforecast/common/_base_model.py b/neuralforecast/common/_base_model.py
@@ -344,7 +344,7 @@ def _fit(
                 f"`dataloader_kwargs={{'num_workers': {self.num_workers_loader}}}`",
                 category=FutureWarning,
             )
-        dataloader_kwargs["num_workers"] = self.num_workers_loader
+            dataloader_kwargs["num_workers"] = self.num_workers_loader
 
         datamodule = datamodule_constructor(
             dataset=dataset,

diff --git a/neuralforecast/common/_base_recurrent.py b/neuralforecast/common/_base_recurrent.py
@@ -574,7 +574,7 @@ def predict(self, dataset, step_size=1, random_seed=None, **data_module_kwargs):
                 f"`dataloader_kwargs={{'num_workers': {self.num_workers_loader}}}`",
                 category=FutureWarning,
             )
-        data_module_kwargs["num_workers"] = self.num_workers_loader
+            data_module_kwargs["num_workers"] = self.num_workers_loader
 
         if step_size > 1:
             raise Exception("Recurrent models do not support step_size > 1")