From 0a772c5a43fc4275ab5c6ed7670ff641d41be83f Mon Sep 17 00:00:00 2001
From: Olivier Sprangers <45119856+elephaint@users.noreply.github.com>
Date: Thu, 11 Apr 2024 18:53:17 +0200
Subject: [PATCH] fix_wrong_import_doc (#967)

---
 nbs/models.patchtst.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nbs/models.patchtst.ipynb b/nbs/models.patchtst.ipynb
index 6fba91284..a5c74be3f 100644
--- 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",