Fix get_contestant_ranks when only one score is available

womboai · Nov 29, 2024 · e27e804 · e27e804
1 parent 5d6008c
commit e27e804
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/validator/weight_setting/winner_selection.py b/validator/weight_setting/winner_selection.py
@@ -22,9 +22,15 @@ def get_contestant_ranks(scores: dict[Key, float]) -> dict[Key, int]:
     if not scores:
         return {}
 
-    i = 0
     rank = 0
 
+    if len(scores) == 1:
+        hotkey = next(iter(scores))
+
+        return { hotkey: rank }
+
+    i = 0
+
     scores = list(sorted(scores.items(), key=itemgetter(1), reverse=True))
     score_values = list(map(itemgetter(1), scores))