From e348748a5ca7f23f8d0d568c1986a13f71fc898c Mon Sep 17 00:00:00 2001 From: bronsen Date: Wed, 8 Apr 2026 23:46:50 +0200 Subject: [PATCH] codestyle: fiddle with type hints and LSP hints --- src/teilchensammler_cli/models.py | 6 +++++- tests.py | 6 +++--- 2 files changed, 8 insertions(+), 4 deletions(-) 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(