Fix rank not found

src/wandb_data.py

@@ -159,6 +159,12 @@ def _add_runs(wandb_runs: list[wapi.Run]):
                 benchmark_metrics = benchmark["metrics"]
                 if hotkey not in submission_info:
                     continue
+                ranks = metrics["ranks"]
+                if hotkey not in ranks:
+                    continue
+                scores = metrics["scores"]
+                if hotkey not in scores:
+                    continue
                 submissions[hotkey] = Submission(
                     info=submission_info[hotkey],
                     metrics=MetricData(
@@ -170,8 +176,8 @@ def _add_runs(wandb_runs: list[wapi.Run]):
                     ),
                     average_similarity=float(benchmark["average_similarity"]),
                     min_similarity=float(benchmark["min_similarity"]),
-                    rank=int(metrics["ranks"][hotkey]),
-                    score=float(metrics["scores"][hotkey]),
+                    rank=int(ranks[hotkey]),
+                    score=float(scores[hotkey]),
                 )
 
         if "invalid_submissions" in metrics: