Changed UnknownTicket to UnknownBet in classifier.py

This commit is contained in:
Chlupaty
2026-03-22 11:49:34 +01:00
parent fa19b30601
commit d8e71d3483

View File

@@ -13,7 +13,7 @@ from beaky.datamodels.ticket import (
GoalAmount, GoalAmount,
GoalHandicap, GoalHandicap,
Ticket, Ticket,
UnknownTicket, UnknownBet,
WinDrawLose, WinDrawLose,
WinDrawLoseDouble, WinDrawLoseDouble,
WinLose, WinLose,
@@ -41,7 +41,7 @@ def classify(text: str) -> Bet:
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
if not text: if not text:
return UnknownTicket( return UnknownBet(
ticketType=BetType.UNKNOWN, ticketType=BetType.UNKNOWN,
team1Name="N/A", team1Name="N/A",
team2Name="N/A", team2Name="N/A",
@@ -129,7 +129,7 @@ def classify(text: str) -> Bet:
return WinDrawLose(ticketType=BetType.WIN_DRAW_LOSE, betType=bet_type, **base_args) return WinDrawLose(ticketType=BetType.WIN_DRAW_LOSE, betType=bet_type, **base_args)
# Fallback Unknown # Fallback Unknown
return UnknownTicket(ticketType=BetType.UNKNOWN, raw_text=text, **base_args) return UnknownBet(ticketType=BetType.UNKNOWN, raw_text=text, **base_args)
def img_classify(paths: list[str], ticket_id: int) -> Ticket: def img_classify(paths: list[str], ticket_id: int) -> Ticket:
@@ -153,7 +153,7 @@ def img_classify(paths: list[str], ticket_id: int) -> Ticket:
except Exception as exc: # pragma: no cover - defensive fallback except Exception as exc: # pragma: no cover - defensive fallback
# Ensure result is always defined so downstream code cannot reference an unbound name # Ensure result is always defined so downstream code cannot reference an unbound name
print(f"classify() raised an exception: {exc}") print(f"classify() raised an exception: {exc}")
result = UnknownTicket( result = UnknownBet(
ticketType=BetType.UNKNOWN, ticketType=BetType.UNKNOWN,
team1Name="N/A", team1Name="N/A",
team2Name="N/A", team2Name="N/A",