fix: wandb missing final scores (#88)

* fix: wandb now storing final scores correctly * fix: cast scores from tensor.float to primitive float
tensorplex-labs · Dec 9, 2024 · d992fd9 · d992fd9
1 parent e63481d
commit d992fd9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/commons/scoring.py b/commons/scoring.py
@@ -641,8 +641,8 @@ def calculate_score(
                 ground_truth = gt_score[i]
 
                 # NOTE: just use ground truth for now
-                hotkey_to_final_score[r.axon.hotkey] = ground_truth / len(
-                    criteria_types
+                hotkey_to_final_score[r.axon.hotkey] = float(
+                    ground_truth / len(criteria_types)
                 )
 
             criteria_to_miner_scores[criteria.type] = Score(