Corrected notebook to point to test datasets

notebooks/template-audio.ipynb

@@ -148,8 +148,8 @@
     "dataset = load_dataset(request.dataset_name)\n",
     "\n",
     "# Split dataset\n",
-    "train_test = dataset[\"train\"].train_test_split(test_size=request.test_size, seed=request.test_seed)\n",
-    "test_dataset = train_test[\"test\"]"
+    "train_test = dataset[\"train\"]\n",
+    "test_dataset = dataset[\"test\"]"
    ]
   },
   {

notebooks/template-image.ipynb

@@ -212,8 +212,8 @@
     "dataset = load_dataset(request.dataset_name)\n",
     "\n",
     "# Split dataset\n",
-    "train_test = dataset[\"train\"].train_test_split(test_size=request.test_size, seed=request.test_seed)\n",
-    "test_dataset = train_test[\"test\"]"
+    "train_test = dataset[\"train\"]\n",
+    "test_dataset = dataset[\"test\"]"
    ]
   },
   {

notebooks/template-text.ipynb

@@ -157,8 +157,8 @@
     "dataset = dataset.map(lambda x: {\"label\": LABEL_MAPPING[x[\"label\"]]})\n",
     "\n",
     "# Split dataset\n",
-    "train_test = dataset[\"train\"].train_test_split(test_size=request.test_size, seed=request.test_seed)\n",
-    "test_dataset = train_test[\"test\"]"
+    "train_test = dataset[\"train\"]\n",
+    "test_dataset = dataset[\"test\"]"
    ]
   },
   {

tasks/text.py

@@ -44,8 +44,8 @@ async def evaluate_text(request: TextEvaluationRequest):
     dataset = dataset.map(lambda x: {"label": LABEL_MAPPING[x["label"]]})
 
     # Split dataset
-    train_test = train_test["train"]
-    test_dataset = train_test["test"]
+    train_test = dataset["train"]
+    test_dataset = dataset["test"]
     
     # Start tracking emissions
     tracker.start()