Correctly parse empty columns in /leaderboard_space/utils.py

yourbench_space/leaderboard_space/utils.py

@@ -43,9 +43,12 @@ def extract_dataviz() -> Tuple[list, list]:
 
             for ix, row in enumerate(ds):
                 prompt = row["full_prompt"]
-                gold = row["gold"]
+                gold = row.get("gold", "")
+                if isinstance(gold, list):
+                    gold = gold[0] if gold else ""
                 score = list(row["metrics"].values())[0]
-                prediction = row["predictions"][0]
+                predictions = row.get("predictions", [])
+                prediction = predictions[0] if predictions else ""
 
                 # We store flattened samples in a dict
                 # ix -> ix, prompt, gold, model_score for each model, model_prediction for each model