fix wrong print (huggingface#5256)

print `encoded_dataset.column_names` not `dataset.column_names`
andreaschandra · Nov 18, 2022 · 4eccb22 · 4eccb22
1 parent 731f3af
commit 4eccb22
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebooks/Overview.ipynb b/notebooks/Overview.ipynb
@@ -2204,13 +2204,13 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "['id', 'title', 'context', 'question', 'answers']\n"
+      "['id', 'title', 'context', 'question', 'answers', 'input_ids', 'token_type_ids', 'attention_mask']\n"
      ]
     }
    ],
    "source": [
     "# we have added additional columns\n",
-    "pprint(dataset.column_names)"
+    "pprint(encoded_dataset.column_names)"
    ]
   },
   {