diff --git a/notebooks/hfdemo/patch_tsmixer_blog.ipynb b/notebooks/hfdemo/patch_tsmixer_blog.ipynb
index 56a6edac..c9380b21 100644
--- a/notebooks/hfdemo/patch_tsmixer_blog.ipynb
+++ b/notebooks/hfdemo/patch_tsmixer_blog.ipynb
@@ -240,12 +240,7 @@
     "tsp = TimeSeriesPreprocessor(\n",
     "    timestamp_column=timestamp_column,\n",
     "    id_columns=id_columns,\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    scaling=True,\n",
     ")\n",
     "tsp = tsp.train(train_data)"
@@ -261,12 +256,7 @@
     "    tsp.preprocess(train_data),\n",
     "    id_columns=id_columns,\n",
     "    timestamp_column=\"date\",\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    context_length=context_length,\n",
     "    prediction_length=forecast_horizon,\n",
     ")\n",
@@ -274,12 +264,7 @@
     "    tsp.preprocess(valid_data),\n",
     "    id_columns=id_columns,\n",
     "    timestamp_column=\"date\",\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    context_length=context_length,\n",
     "    prediction_length=forecast_horizon,\n",
     ")\n",
@@ -287,12 +272,7 @@
     "    tsp.preprocess(test_data),\n",
     "    id_columns=id_columns,\n",
     "    timestamp_column=\"date\",\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    context_length=context_length,\n",
     "    prediction_length=forecast_horizon,\n",
     ")"
@@ -878,12 +858,7 @@
     "tsp = TimeSeriesPreprocessor(\n",
     "    timestamp_column=timestamp_column,\n",
     "    id_columns=id_columns,\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    scaling=True,\n",
     ")\n",
     "tsp = tsp.train(train_data)"
@@ -898,36 +873,21 @@
     "train_dataset = ForecastDFDataset(\n",
     "    tsp.preprocess(train_data),\n",
     "    id_columns=id_columns,\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    context_length=context_length,\n",
     "    prediction_length=forecast_horizon,\n",
     ")\n",
     "valid_dataset = ForecastDFDataset(\n",
     "    tsp.preprocess(valid_data),\n",
     "    id_columns=id_columns,\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    context_length=context_length,\n",
     "    prediction_length=forecast_horizon,\n",
     ")\n",
     "test_dataset = ForecastDFDataset(\n",
     "    tsp.preprocess(test_data),\n",
     "    id_columns=id_columns,\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    context_length=context_length,\n",
     "    prediction_length=forecast_horizon,\n",
     ")"
diff --git a/notebooks/hfdemo/patch_tst_getting_started.ipynb b/notebooks/hfdemo/patch_tst_getting_started.ipynb
index 4dd9fe7c..975c09b6 100644
--- a/notebooks/hfdemo/patch_tst_getting_started.ipynb
+++ b/notebooks/hfdemo/patch_tst_getting_started.ipynb
@@ -246,12 +246,7 @@
     "tsp = TimeSeriesPreprocessor(\n",
     "    timestamp_column=timestamp_column,\n",
     "    id_columns=id_columns,\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    scaling=True,\n",
     ")\n",
     "tsp.train(train_data)"
@@ -267,36 +262,21 @@
     "train_dataset = ForecastDFDataset(\n",
     "    tsp.preprocess(train_data),\n",
     "    id_columns=id_columns,\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    context_length=context_length,\n",
     "    prediction_length=forecast_horizon,\n",
     ")\n",
     "valid_dataset = ForecastDFDataset(\n",
     "    tsp.preprocess(valid_data),\n",
     "    id_columns=id_columns,\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    context_length=context_length,\n",
     "    prediction_length=forecast_horizon,\n",
     ")\n",
     "test_dataset = ForecastDFDataset(\n",
     "    tsp.preprocess(test_data),\n",
     "    id_columns=id_columns,\n",
-<<<<<<< HEAD
     "    target_columns=forecast_columns,\n",
-=======
-    "    input_columns=forecast_columns,\n",
-    "    output_columns=forecast_columns,\n",
->>>>>>> main
     "    context_length=context_length,\n",
     "    prediction_length=forecast_horizon,\n",
     ")"