leaderboard/vote.py

from enum import Enum

class Vote(Enum):
    GOOD = "GOOD"
    BAD = "BAD"
    NEUTRAL = "NEUTRAL"

def vote_for_model(model_a: str, model_a_vote: Vote, model_b: str, model_b_vote: Vote):
    print(f"Voting for {model_a}: {model_a_vote} | Voting for {model_b}: {model_b_vote}")

    return "", ""