Changed UnknownTicket to UnknownBet in classifier.py
This commit is contained in:
@@ -13,7 +13,7 @@ from beaky.datamodels.ticket import (
|
||||
GoalAmount,
|
||||
GoalHandicap,
|
||||
Ticket,
|
||||
UnknownTicket,
|
||||
UnknownBet,
|
||||
WinDrawLose,
|
||||
WinDrawLoseDouble,
|
||||
WinLose,
|
||||
@@ -41,7 +41,7 @@ def classify(text: str) -> Bet:
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
if not text:
|
||||
return UnknownTicket(
|
||||
return UnknownBet(
|
||||
ticketType=BetType.UNKNOWN,
|
||||
team1Name="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)
|
||||
|
||||
# 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:
|
||||
@@ -153,7 +153,7 @@ def img_classify(paths: list[str], ticket_id: int) -> Ticket:
|
||||
except Exception as exc: # pragma: no cover - defensive fallback
|
||||
# Ensure result is always defined so downstream code cannot reference an unbound name
|
||||
print(f"classify() raised an exception: {exc}")
|
||||
result = UnknownTicket(
|
||||
result = UnknownBet(
|
||||
ticketType=BetType.UNKNOWN,
|
||||
team1Name="N/A",
|
||||
team2Name="N/A",
|
||||
|
||||
Reference in New Issue
Block a user