Wrap pydantic schema in no_info_after_validator_function

PiperOrigin-RevId: 642561049

Author: Joshua Bambrick

CHANGELOG.md

@@ -8,6 +8,11 @@ Changelog follow https://keepachangelog.com/ format.
 
 ## [Unreleased]
 
+## [1.9.2] - 2024-06-12
+
+*   `epath`:
+    *   Support pydantic serialization of epath.Path
+
 ## [1.9.1] - 2024-06-04
 
 *   `epath`:

etils/__init__.py

@@ -16,7 +16,7 @@
 
 # A new PyPI release will be pushed everytime `__version__` is increased
 # When changing this, also update the CHANGELOG.md
-__version__ = '1.9.1'
+__version__ = '1.9.2'
 
 # Do NOT add anything to this file. This is left empty on purpose.
 # Users should import subprojects directly.

etils/epath/abstract_path.py

@@ -242,5 +242,10 @@ def __get_pydantic_core_schema__(cls, source_type, handler):
       pydantic_core.CoreSchema
 
     """
+    from pydantic_core import core_schema  # pylint: disable=g-import-not-at-top  # pytype: disable=import-error
+
     del source_type  # Unused in this implementation.
-    return handler(pathlib.PurePosixPath)
+
+    return core_schema.no_info_after_validator_function(
+        cls, handler(pathlib.PurePosixPath)
+    )

etils/epath/abstract_path_test.py

@@ -0,0 +1,30 @@
+# Copyright 2024 The etils Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Tests pydantic serialization of etils.epath.Path."""
+
+from etils import epath
+import pydantic
+
+
+def test_pydantic_serialize():
+
+  class _Model(pydantic.BaseModel):
+    path_field: epath.Path
+
+  model = _Model(path_field=epath.Path("/example/path"))
+  serialized = model.model_dump_json()
+  deserialized = _Model.model_validate_json(serialized)
+  assert isinstance(deserialized.path_field, epath.Path)
+  assert model == deserialized

pyproject.toml

@@ -134,6 +134,7 @@ dev = [
     "torch",
     "optree",  # For `etree.optree`
     "dataclass_array",  # To test lazy_imports
+    "pydantic",  # To test pydantic serialization of epath.Path
 ]
 docs = [
     "sphinx-apitree[ext]",