mypy fixes

src/baby_monitor/models/db/user.py

@@ -1,9 +1,14 @@
 """Database models."""
 
+from typing import TYPE_CHECKING
 from sqlalchemy import Column, Integer, String, DateTime, Boolean
 from datetime import datetime
 
-from baby_monitor.repositories.dependencies.get_database import Base
+if TYPE_CHECKING:
+    from sqlalchemy.orm import DeclarativeBase
+    Base = DeclarativeBase
+else:
+    from baby_monitor.repositories.dependencies.get_database import Base
 
 
 class User(Base):

src/baby_monitor/models/invitation.py

@@ -5,10 +5,15 @@ SQLite will store as naive UTC, which is handled in the repository layer.
 """
 
 from datetime import datetime
+from typing import TYPE_CHECKING
 from sqlalchemy import Boolean, DateTime, Integer, String
 from sqlalchemy.orm import Mapped, mapped_column
 
-from baby_monitor.repositories.dependencies.get_database import Base
+if TYPE_CHECKING:
+    from sqlalchemy.orm import DeclarativeBase
+    Base = DeclarativeBase
+else:
+    from baby_monitor.repositories.dependencies.get_database import Base
 
 
 class Invitation(Base):