Fix Notebook01 Dataset column_rename (#543)

adapter-hub · May 10, 2023 · 1ae5287 · 1ae5287
1 parent db09025
commit 1ae5287
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/notebooks/01_Adapter_Training.ipynb b/notebooks/01_Adapter_Training.ipynb
@@ -433,7 +433,7 @@
         "# Encode the input data\n",
         "dataset = dataset.map(encode_batch, batched=True)\n",
         "# The transformers model expects the target class column to be named \"labels\"\n",
-        "dataset.rename_column_(\"label\", \"labels\")\n",
+        "dataset = dataset.rename_column(\"label\", \"labels\")\n",
         "# Transform to pytorch tensors and only output the required columns\n",
         "dataset.set_format(type=\"torch\", columns=[\"input_ids\", \"attention_mask\", \"labels\"])"
       ]
@@ -4209,4 +4209,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}
diff --git a/notebooks/03_Adapter_Fusion.ipynb b/notebooks/03_Adapter_Fusion.ipynb
@@ -266,7 +266,7 @@
     "# Encode the input data\n",
     "dataset = dataset.map(encode_batch, batched=True)\n",
     "# The transformers model expects the target class column to be named \"labels\"\n",
-    "dataset.rename_column_(\"label\", \"labels\")\n",
+    "dataset = dataset.rename_column(\"label\", \"labels\")\n",
     "# Transform to pytorch tensors and only output the required columns\n",
     "dataset.set_format(type=\"torch\", columns=[\"input_ids\", \"attention_mask\", \"labels\"])"
    ]
@@ -728,4 +728,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 1
-}
+}
diff --git a/notebooks/04_Cross_Lingual_Transfer.ipynb b/notebooks/04_Cross_Lingual_Transfer.ipynb
@@ -356,7 +356,7 @@
     "  # Encode the input data\n",
     "  dataset = dataset.map(encode_batch, batched=True)\n",
     "  # The transformers model expects the target class column to be named \"labels\"\n",
-    "  dataset.rename_column_(\"label\", \"labels\")\n",
+    "  dataset = dataset.rename_column(\"label\", \"labels\")\n",
     "  # Transform to pytorch tensors and only output the required columns\n",
     "  dataset.set_format(columns=[\"input_ids\", \"attention_mask\", \"labels\"])\n",
     "  return dataset\n",

diff --git a/notebooks/05_Adapter_Drop_Training.ipynb b/notebooks/05_Adapter_Drop_Training.ipynb
@@ -169,7 +169,7 @@
     "# Encode the input data\n",
     "dataset = dataset.map(encode_batch, batched=True)\n",
     "# The transformers model expects the target class column to be named \"labels\"\n",
-    "dataset.rename_column_(\"label\", \"labels\")\n",
+    "dataset = dataset.rename_column(\"label\", \"labels\")\n",
     "# Transform to pytorch tensors and only output the required columns\n",
     "dataset.set_format(type=\"torch\", columns=[\"input_ids\", \"attention_mask\", \"labels\"])"
    ]
@@ -475,4 +475,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 1
-}
+}