From c1723eb7c9f6eb41c9dab4271e80c6f4cf84915d Mon Sep 17 00:00:00 2001 From: FelixFehse Date: Fri, 14 Jun 2024 11:39:24 +0200 Subject: [PATCH] fix: name split explicitly to avoid code execution --- src/documentation/evaluation.ipynb | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/documentation/evaluation.ipynb b/src/documentation/evaluation.ipynb index 0a80cc5c0..1baf85991 100644 --- 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" ] },