fix: fixed wandb missing final scores (#88)

- wandb now storing final scores correctly - cast scores from tensor.float to primitive float
tensorplex-labs · Dec 9, 2024 · 11ce44a · 11ce44a
1 parent 383d505
commit 11ce44a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 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(

diff --git a/dojo/__init__.py b/dojo/__init__.py
@@ -59,4 +59,4 @@ def get_dojo_api_base_url() -> str:
     if base_url is None:
         raise ValueError("DOJO_API_BASE_URL is not set in the environment.")
 
-    return base_url
+    return base_url