Skip to content

Commit 3d8db41

Browse files
🎨 Auto format
1 parent 0b86f18 commit 3d8db41

2 files changed

Lines changed: 7 additions & 5 deletions

File tree

‎sqlmodel/__init__.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,5 +145,5 @@
145145
from .sql.expression import type_coerce as type_coerce
146146
from .sql.expression import within_group as within_group
147147
from .sql.sqltypes import AutoString as AutoString
148-
from .sql.sqltypes import UTCDateTime as UTCDateTime
149148
from .sql.sqltypes import IntEnum as IntEnum
149+
from .sql.sqltypes import UTCDateTime as UTCDateTime

‎sqlmodel/sql/sqltypes.py‎

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,8 @@ def load_dialect_impl(self, dialect: Dialect) -> "types.TypeEngine[Any]":
6464

6565

6666
_TIntEnum = TypeVar("_TIntEnum", bound="_IntEnum")
67+
68+
6769
class IntEnum(types.TypeDecorator[Optional[_TIntEnum]]):
6870
impl = types.SmallInteger
6971
cache_ok = True
@@ -79,14 +81,14 @@ def __init__(self, enum_type: type[_TIntEnum], *args: Any, **kwargs: Any):
7981

8082
def process_result_value(
8183
self,
82-
value: Optional[int],
84+
value: int | None,
8385
dialect: Dialect,
84-
) -> Optional[_TIntEnum]:
86+
) -> _TIntEnum | None:
8587
return None if (value is None) else self.enum_type(value)
8688

8789
def process_bind_param(
8890
self,
89-
value: Optional[_TIntEnum],
91+
value: _TIntEnum | None,
9092
dialect: Dialect,
91-
) -> Optional[int]:
93+
) -> int | None:
9294
return None if (value is None) else value.value

0 commit comments

Comments
 (0)