diff --git a/sqlmodel/main.py b/sqlmodel/main.py index 2e558647f1..31f2e334e3 100644 --- a/sqlmodel/main.py +++ b/sqlmodel/main.py @@ -14,6 +14,7 @@ Any, Callable, ClassVar, + Literal, Optional, TypeVar, Union, @@ -48,7 +49,7 @@ from sqlalchemy.orm.instrumentation import is_instrumented from sqlalchemy.sql.schema import MetaData from sqlalchemy.sql.sqltypes import LargeBinary, Time, Uuid -from typing_extensions import Literal, TypeAlias, deprecated, get_origin +from typing_extensions import TypeAlias, deprecated, get_origin from ._compat import ( # type: ignore[attr-defined] IS_PYDANTIC_V2, diff --git a/sqlmodel/sql/expression.py b/sqlmodel/sql/expression.py index 7732336118..72d7b1de7d 100644 --- a/sqlmodel/sql/expression.py +++ b/sqlmodel/sql/expression.py @@ -1,6 +1,7 @@ from collections.abc import Iterable, Mapping, Sequence from typing import ( Any, + Literal, Optional, TypeVar, Union, @@ -34,7 +35,6 @@ UnaryExpression, ) from sqlalchemy.sql.type_api import TypeEngine -from typing_extensions import Literal from ._expression_select_cls import Select as Select from ._expression_select_cls import SelectOfScalar as SelectOfScalar diff --git a/tests/test_pydantic/test_field.py b/tests/test_pydantic/test_field.py index 9d7bc77625..140b02fd9b 100644 --- a/tests/test_pydantic/test_field.py +++ b/tests/test_pydantic/test_field.py @@ -1,10 +1,9 @@ from decimal import Decimal -from typing import Optional, Union +from typing import Literal, Optional, Union import pytest from pydantic import ValidationError from sqlmodel import Field, SQLModel -from typing_extensions import Literal def test_decimal():