Use pydantic compact to handle v1/v2 transition

polyaxon · Jul 29, 2023 · 03e0852 · 03e0852
1 parent b259fb2
commit 03e0852
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/traceml/traceml/artifacts/schemas.py b/traceml/traceml/artifacts/schemas.py
@@ -2,8 +2,8 @@
 
 from typing import Dict, List, Optional
 
+from clipped.compact.pydantic import StrictStr
 from clipped.types.uuids import UUIDStr
-from pydantic import StrictStr
 
 from polyaxon.schemas.base import BaseSchemaModel
 from traceml.artifacts.kinds import V1ArtifactKind

diff --git a/traceml/traceml/events/schemas.py b/traceml/traceml/events/schemas.py
@@ -4,14 +4,14 @@
 from collections import namedtuple
 from typing import Dict, List, Mapping, Optional, Union
 
+from clipped.compact.pydantic import StrictStr, root_validator
 from clipped.config.parser import ConfigParser
 from clipped.config.schema import skip_partial
 from clipped.utils.csv import validate_csv
 from clipped.utils.dates import parse_datetime
 from clipped.utils.enums import PEnum
 from clipped.utils.np import sanitize_np_types
 from clipped.utils.tz import now
-from pydantic import StrictStr, root_validator
 
 from polyaxon.schemas.base import BaseSchemaModel
 from traceml.artifacts.kinds import V1ArtifactKind

diff --git a/traceml/traceml/logging/schemas.py b/traceml/traceml/logging/schemas.py
@@ -2,11 +2,11 @@
 
 from typing import List, Optional, Text
 
+from clipped.compact.pydantic import StrictStr
 from clipped.utils.csv import validate_csv
 from clipped.utils.dates import parse_datetime
 from clipped.utils.json import orjson_dumps, orjson_loads
 from clipped.utils.tz import now
-from pydantic import StrictStr
 
 from polyaxon.schemas.base import BaseSchemaModel
 from traceml.logging.parser import (