diff --git a/src/teilchensammler_cli/models.py b/src/teilchensammler_cli/models.py index 30bfbff..7a97299 100644 --- a/src/teilchensammler_cli/models.py +++ b/src/teilchensammler_cli/models.py @@ -1,3 +1,5 @@ +from typing import Any +from sqlmodel.sql.expression import Select import logging import uuid @@ -91,7 +93,9 @@ async def load_initial_data(engine) -> Sequence[Teilchen]: List of Teilchen, potentially empty """ with Session(engine) as session: - statement = select(Teilchen.id, Teilchen.name, Teilchen.description, Teilchen.tags) # ty:ignore[no-matching-overload] + statement: Select[Any] = select( + Teilchen.id, Teilchen.name, Teilchen.description, Teilchen.tags + ) all_teilchen = session.exec(statement).all() logger.debug("Loading initial data: found %s records", len(all_teilchen)) diff --git a/tests.py b/tests.py index c60d834..55b40b8 100644 --- a/tests.py +++ b/tests.py @@ -1,12 +1,12 @@ -from teilchensammler_cli.database import create_db_and_tables -from sqlalchemy.sql import text import uuid from typing import Generator import pytest from sqlalchemy import Engine +from sqlalchemy.sql import text from sqlmodel import Session, SQLModel, create_engine +from teilchensammler_cli.database import create_db_and_tables from teilchensammler_cli.models import ( Teilchen, TeilchenCreate, @@ -46,7 +46,7 @@ def TC(**kwargs) -> TeilchenCreate | None: """ if kwargs: arguments = empty_teilchen_data | kwargs - return TeilchenCreate(**arguments) # ty:ignore[invalid-argument-type] + return TeilchenCreate(**arguments) @pytest.mark.parametrize(