diff --git a/src/patito/_pydantic/dtypes/dtypes.py b/src/patito/_pydantic/dtypes/dtypes.py index 3422487..24333d1 100644 --- a/src/patito/_pydantic/dtypes/dtypes.py +++ b/src/patito/_pydantic/dtypes/dtypes.py @@ -2,7 +2,7 @@ from collections.abc import Mapping from functools import cache, reduce -from operator import and_, or_ +from operator import or_ from typing import TYPE_CHECKING, Any import polars as pl diff --git a/tests/test_dtypes.py b/tests/test_dtypes.py index 74e1a08..4406939 100644 --- a/tests/test_dtypes.py +++ b/tests/test_dtypes.py @@ -51,9 +51,7 @@ def test_valids_basic_annotations() -> None: with pytest.raises(TypeError, match="Mixed type enums not supported"): DtypeResolver(Literal[1, 2, "3"]).valid_polars_dtypes() # pyright: ignore - assert DtypeResolver( - Literal["a", "b", "c"] - ).valid_polars_dtypes() == { # pyright: ignore + assert DtypeResolver(Literal["a", "b", "c"]).valid_polars_dtypes() == { # pyright: ignore pl.Enum(["a", "b", "c"]), pl.String, }