diff --git a/api/models/task.py b/api/models/task.py index 548a142273..8bc7372523 100644 --- a/api/models/task.py +++ b/api/models/task.py @@ -1,8 +1,8 @@ from datetime import UTC, datetime from celery import states # type: ignore - from sqlalchemy.orm import mapped_column + from models.base import Base from .engine import db