diff --git a/backend/app/dao/base.py b/backend/app/dao/base.py index db4341a31..2222d8ba2 100644 --- a/backend/app/dao/base.py +++ b/backend/app/dao/base.py @@ -2,7 +2,7 @@ from collections.abc import AsyncGenerator, Sequence from contextlib import asynccontextmanager, contextmanager from contextvars import ContextVar -from typing import Any, Generic, Type, TypeVar +from typing import Any, Generic, TypeVar from sqlalchemy import event, select from sqlalchemy.ext.asyncio import AsyncSession @@ -16,7 +16,7 @@ class BaseDAO(Generic[ModelType]): """Base class for data access objects, managing session context and basic CRUD.""" - def __init__(self, model: Type[ModelType]): + def __init__(self, model: type[ModelType]): self.model = model @asynccontextmanager diff --git a/backend/tests/test_base_dao.py b/backend/tests/test_base_dao.py index d1ed2f582..71dcfe03b 100644 --- a/backend/tests/test_base_dao.py +++ b/backend/tests/test_base_dao.py @@ -1,5 +1,5 @@ -from types import SimpleNamespace import uuid +from types import SimpleNamespace import pytest from sqlalchemy import String, create_engine, select