chore(deps): update dependency sqlalchemy to v2.0.4

.gitignore

@@ -164,3 +164,5 @@ cython_debug/
 .DS_Store
 
 whisperbox-transcribe.sqlite*
+*shm
+*wal

app/shared/db/alembic/env.py

@@ -62,7 +62,7 @@ def run_migrations_online() -> None:
     """
 
     connectable = engine_from_config(
-        config.get_section(config.config_ini_section),
+        config.get_section(config.config_ini_section),  # type: ignore
         prefix="sqlalchemy.",
         poolclass=pool.NullPool,
     )

app/shared/db/models.py

@@ -3,8 +3,7 @@ from typing import Optional
 
 from sqlalchemy import JSON, VARCHAR, Column, DateTime, Enum, ForeignKey, String, func
 from sqlalchemy.dialects.postgresql import UUID
-from sqlalchemy.ext.declarative import declarative_base, declared_attr
-from sqlalchemy.orm import Mapped, declarative_mixin  # type: ignore
+from sqlalchemy.orm import Mapped, declarative_base, declarative_mixin, declared_attr
 
 from .schemas import ArtifactType, JobStatus, JobType
 

app/tests/test_api.py

@@ -87,7 +87,7 @@ def test_get_artifacts_pass(
     auth_headers: Dict[str, str], db_session: Session, mock_job: models.Job
 ) -> None:
     artifact = models.Artifact(
-        data=[], job_id=mock_job.id, type=schemas.ArtifactType.raw_transcript
+        data=[], job_id=str(mock_job.id), type=schemas.ArtifactType.raw_transcript
     )
 
     db_session.add(artifact)

mypy.ini

@@ -1,4 +1,4 @@
 [mypy]
-plugins = sqlmypy
+plugins = sqlalchemy.ext.mypy.plugin
 ignore_missing_imports = True
 disallow_untyped_defs = True

pyproject.toml

@@ -5,7 +5,7 @@ version = "0.1.0"
 
 dependencies=[
   "celery[redis] ==5.2.7",
-  "sqlalchemy[mypy] ==1.4.46",
+  "sqlalchemy[mypy] ==2.0.4",
   "pydantic ==1.10.5"
 ]
 
@@ -36,7 +36,6 @@ tooling = [
   "pytest ==7.2.1",
   # types
   "mypy ==1.0.1",
-  "sqlalchemy-stubs ==0.4",
   "types-requests ==2.28.11.15"
 ]