Lock file maintenance

mex/common/types/email.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-from pydantic import GetJsonSchemaHandler, json_schema
+from pydantic import GetCoreSchemaHandler, GetJsonSchemaHandler, json_schema
 from pydantic_core import core_schema
 
 EMAIL_PATTERN = r"^[^@ \t\r\n]+@[^@ \t\r\n]+\.[^@ \t\r\n]+$"
@@ -10,7 +10,9 @@ class Email(str):
     """Email address of a person, organization or other entity."""
 
     @classmethod
-    def __get_pydantic_core_schema__(cls, source: type[Any]) -> core_schema.CoreSchema:
+    def __get_pydantic_core_schema__(
+        cls, source_type: Any, handler: GetCoreSchemaHandler
+    ) -> core_schema.CoreSchema:
         """Modify the core schema to add the email regex."""
         return core_schema.str_schema(pattern=EMAIL_PATTERN)
 

mex/common/types/identifier.py

@@ -3,7 +3,7 @@
 from typing import Any, Self
 from uuid import UUID, uuid4
 
-from pydantic import GetJsonSchemaHandler, json_schema
+from pydantic import GetCoreSchemaHandler, GetJsonSchemaHandler, json_schema
 from pydantic_core import core_schema
 
 MEX_ID_ALPHABET = string.ascii_letters + string.digits
@@ -42,7 +42,9 @@ def validate(cls, value: Any) -> Self:
         raise ValueError(f"Cannot parse {type(value)} as {cls.__name__}")
 
     @classmethod
-    def __get_pydantic_core_schema__(cls, source: type[Any]) -> core_schema.CoreSchema:
+    def __get_pydantic_core_schema__(
+        cls, source_type: Any, handler: GetCoreSchemaHandler
+    ) -> core_schema.CoreSchema:
         """Modify the core schema to add the ID regex."""
         return core_schema.no_info_before_validator_function(
             cls.validate, core_schema.str_schema(pattern=MEX_ID_PATTERN)

mex/common/types/path.py

@@ -2,6 +2,7 @@
 from pathlib import Path
 from typing import Any, Self, Union
 
+from pydantic import GetCoreSchemaHandler
 from pydantic_core import core_schema
 
 
@@ -49,7 +50,9 @@ def is_relative(self) -> bool:
         return not self._path.is_absolute()
 
     @classmethod
-    def __get_pydantic_core_schema__(cls, source: type[Any]) -> core_schema.CoreSchema:
+    def __get_pydantic_core_schema__(
+        cls, source_type: Any, handler: GetCoreSchemaHandler
+    ) -> core_schema.CoreSchema:
         """Set schema to str schema."""
         from_str_schema = core_schema.chain_schema(
             [

mex/common/types/temporal_entity.py

@@ -6,7 +6,7 @@
 from typing import Any, Literal, Union, cast, overload
 
 from pandas._libs.tslibs import parsing
-from pydantic import GetJsonSchemaHandler, json_schema
+from pydantic import GetCoreSchemaHandler, GetJsonSchemaHandler, json_schema
 from pydantic_core import core_schema
 from pytz import timezone
 
@@ -189,7 +189,9 @@ def _validate_precision(cls, precision: TemporalEntityPrecision) -> None:
             raise ValueError(error_str)
 
     @classmethod
-    def __get_pydantic_core_schema__(cls, source: type[Any]) -> core_schema.CoreSchema:
+    def __get_pydantic_core_schema__(
+        cls, source_type: Any, handler: GetCoreSchemaHandler
+    ) -> core_schema.CoreSchema:
         """Modify the core schema to add validation and serialization rules."""
         from_str_schema = core_schema.chain_schema(
             [