Convert SNTV winners to Python ints in all cases

endolith · Jul 5, 2024 · c54c52c · c54c52c
1 parent 1b58417
commit c54c52c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/elsim/methods/fptp.py b/elsim/methods/fptp.py
@@ -198,7 +198,10 @@ def sntv(election, n=1, tiebreaker=None):
         tie_winners = tiebreak(list(tied_candidates), n_needed)
         if tie_winners == [None]:
             return None
-        return set(untied_winners) | set(tie_winners)
+        winners = set(untied_winners) | set(tie_winners)
     else:
-        # TODO: Maybe should use arrays for deterministic randomness?
-        return set(int(candidate) for candidate in top_candidates)
+        winners = set(top_candidates)
+
+    # Convert to set of Python ints
+    # TODO: Maybe should use arrays for deterministic randomness?
+    return set(int(winner) for winner in winners)