tgorordo/smithy
1
0
Fork 0
mirror of https://github.com/tgorordo/smithy.git synced 2026-06-13 02:42:14 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update rcv.py to handle ties correctly

Browse source
This commit is contained in:
Thomas (Tom) C. Gorordo 2026-06-12 15:05:27 -07:00 • committed by GitHub
parent 8a324243b3
commit 38c2845c3c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/smithy/rcv.py
View file

@ -59,6 +59,9 @@ def pmg_from_rcv_bigslow(ballots: pl.DataFrame) -> rwx.PyDiGraph:
pmg.add_edge(nodes[a], nodes[b], a_wins - b_wins)
elif b_wins > a_wins:
pmg.add_edge(nodes[b], nodes[a], b_wins - a_wins)
else: # tie
pmg.add_edge(nodes[a], nodes[b], 0)
pmg.add_edge(nodes[b], nodes[a], 0)
return pmg
@ -105,6 +108,9 @@ def pmg_from_rcv_smallfast(ballots: pl.DataFrame) -> rwx.PyDiGraph:
pmg.add_edge(nodes[a], nodes[b], int(a_wins - b_wins))
elif b_wins > a_wins:
pmg.add_edge(nodes[b], nodes[a], int(b_wins - a_wins))
else: # tie
pmg.add_edge(nodes[a], nodes[b], 0)
pmg.add_edge(nodes[b], nodes[a], 0)
return pmg