Update text.py

tasks/text.py

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