fix label mapping

tasks/text.py

@@ -127,7 +127,7 @@ async def evaluate_text(request: TextEvaluationRequest):
         "2_not_human": 2,
         "3_not_bad": 3,
         "4_solutions_harmful_unnecessary": 4,
-        "5_science_is_unreliable": 5,
+        "5_science_unreliable": 5,
         "6_proponents_biased": 6,
         "7_fossil_fuels_needed": 7
     }
@@ -136,8 +136,16 @@ async def evaluate_text(request: TextEvaluationRequest):
         # Load and prepare the dataset
         dataset = load_dataset("QuotaClimat/frugalaichallenge-text-train", token=HF_TOKEN)
         
-        # Convert string labels to integers
-        dataset = dataset.map(lambda x: {"label": LABEL_MAPPING[x["label"]]})
+        # Convert string labels to integers with error handling
+        def convert_label(example):
+            try:
+                return {"label": LABEL_MAPPING[example["label"]]}
+            except KeyError as e:
+                print(f"Warning: Unknown label {example['label']}")
+                # Return default label or raise exception
+                return {"label": 0}  # or raise e if you want to fail on unknown labels
+                
+        dataset = dataset.map(convert_label)
 
         # Split dataset
         test_dataset = dataset["test"]
@@ -220,5 +228,4 @@ async def evaluate_text(request: TextEvaluationRequest):
 
     except Exception as e:
         print(f"Error in evaluate_text: {str(e)}")
-        raise Exception(f"Failed to process request: {str(e)}")
-
+        raise Exception(f"Failed to process request: {str(e)}")