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 "", ""