Fix last type errors

Author: diofeher

README.md

@@ -10,6 +10,7 @@ Backend:
 - Containerization with Docker and Uvicorn for serving
 - Ruff for style checks
 - Github Actions as CI/CD for automatic code checking, using Ruff and ESLint
+- mypy for static type checking 
 
 Frontend:
 - Usage of React and Next.js
@@ -19,7 +20,6 @@ Frontend:
 
 TODO:
 - Alembi for Database migrations
-- mypy for static type checking 
 
 
 # Installation

backend/app/app.py

@@ -10,7 +10,7 @@
 
 
 @asynccontextmanager
-async def lifespan(app: FastAPI) -> AsyncGenerator[None]:
+async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
     # Load the ML model
     create_db_and_tables()
     yield

backend/app/db.py

@@ -1,5 +1,6 @@
 from sqlalchemy_utils import database_exists, create_database
-from sqlmodel import Session, create_engine, SQLModel, Engine
+from sqlmodel import Session, create_engine, SQLModel
+from sqlalchemy.engine import Engine
 from .settings import settings