Fix test

Nixtla · Dec 11, 2024 · eb17a4d · eb17a4d
1 parent 2e3d7c6
commit eb17a4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nbs/core.ipynb b/nbs/core.ipynb
@@ -3172,12 +3172,12 @@
     "    mean = default_optimizer_predict.loc[:, nf_model.__name__].mean()\n",
     "\n",
     "    # using a customized optimizer\n",
+    "    models2 = [nf_model(**params)]\n",
     "    optimizer = torch.optim.Adadelta(params=models2[0].parameters(), rho=0.75)\n",
     "    scheduler=torch.optim.lr_scheduler.StepLR(\n",
     "        optimizer=optimizer, step_size=10e7, gamma=0.5\n",
     "    )\n",
     "\n",
-    "    models2 = [nf_model(**params)]\n",
     "    models2[0].set_configure_optimizers(\n",
     "        optimizer=optimizer,\n",
     "        scheduler=scheduler,\n",