Changed UnknownTicket to UnknownBet in classifier.py
This commit is contained in:
@@ -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",
|
||||||
|
|||||||
Reference in New Issue
Block a user