fix notebook

etna-team · Aug 29, 2024 · ea07e7e · ea07e7e
1 parent c4a7adc
commit ea07e7e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/examples/210-embedding_models.ipynb b/examples/210-embedding_models.ipynb
@@ -658,8 +658,7 @@
     "from etna.datasets import load_dataset\n",
     "\n",
     "ts = load_dataset(\"m3_monthly\")\n",
-    "ts.drop_features(features=[\"origin_timestamp\"])\n",
-    "ts.df_exog = None\n",
+    "ts = TSDataset(ts.to_pandas(features=[\"target\"]), freq=None)\n",
     "ts.head()"
    ]
   },
@@ -1502,8 +1501,7 @@
     "HORIZON = 12\n",
     "\n",
     "ts = load_dataset(\"tourism_monthly\")\n",
-    "ts.drop_features(features=[\"origin_timestamp\"])\n",
-    "ts.df_exog = None\n",
+    "ts = TSDataset(ts.to_pandas(features=[\"target\"]), freq=None)\n",
     "ts.head()"
    ]
   },