fix_wrong_import_doc (#967)

Nixtla · Apr 11, 2024 · 0a772c5 · 0a772c5
1 parent 99bd5c5
commit 0a772c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nbs/models.patchtst.ipynb b/nbs/models.patchtst.ipynb
@@ -904,7 +904,7 @@
     "import matplotlib.pyplot as plt\n",
     "\n",
     "from neuralforecast import NeuralForecast\n",
-    "from neuralforecast.models import MLP\n",
+    "from neuralforecast.models import PatchTST\n",
     "from neuralforecast.losses.pytorch import MQLoss, DistributionLoss\n",
     "from neuralforecast.tsdataset import TimeSeriesDataset\n",
     "from neuralforecast.utils import AirPassengers, AirPassengersPanel, AirPassengersStatic, augment_calendar_df\n",