style: fix type hint

EspenAlbert · Feb 9, 2024 · 6484bfb · 6484bfb
1 parent 5affd69
commit 6484bfb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/model_lib/src/model_lib/pydantic_utils.py b/model_lib/src/model_lib/pydantic_utils.py
@@ -166,7 +166,7 @@ def ensure_timezone(value: datetime):
             return value.replace(tzinfo=timezone.utc)
         return value
 
-    utc_datetime = Annotated[datetime, AfterValidator(ensure_timezone)]
+    utc_datetime: TypeAlias = Annotated[datetime, AfterValidator(ensure_timezone)]
 else:
 
     class _utc_datetime(datetime):
@@ -181,7 +181,7 @@ def ensure_utc(cls, value: datetime):
                 return value.replace(tzinfo=timezone.utc)
             return value
 
-    utc_datetime = Union[_utc_datetime, datetime]
+    utc_datetime: TypeAlias = Union[_utc_datetime, datetime]
 
 if IS_PYDANTIC_V2:
     utc_datetime_ms: TypeAlias = Annotated[