fix: name split explicitly to avoid code execution

Aleph-Alpha · Jun 14, 2024 · c1723eb · c1723eb
1 parent c93eada
commit c1723eb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/documentation/evaluation.ipynb b/src/documentation/evaluation.ipynb
@@ -152,9 +152,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "dataset = load_dataset(\"cardiffnlp/tweet_topic_multi\")\n",
-    "test_set_name = \"validation_random\"\n",
-    "all_data = list(dataset[test_set_name])\n",
+    "dataset = load_dataset(\"cardiffnlp/tweet_topic_multi\", split=\"validation_random\")\n",
+    "all_data = list(dataset)\n",
     "data = all_data[:25]  # this has 573 datapoints, let's take a look at 25 for now"
    ]
   },