add pydantic>2

Signed-off-by: Future-Outlier <[email protected]>
flyteorg · Nov 20, 2024 · 9a03f51 · 9a03f51
1 parent 844d2d3
commit 9a03f51
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/dev-requirements.in b/dev-requirements.in
@@ -4,7 +4,7 @@ coverage
 pre-commit
 codespell
 mock
-pydantic
+pydantic>2
 pytest
 mypy
 mashumaro

diff --git a/examples/data_types_and_io/Dockerfile b/examples/data_types_and_io/Dockerfile
@@ -19,7 +19,7 @@ RUN python3 -m venv ${VENV}
 ENV PATH="${VENV}/bin:$PATH"
 
 RUN --mount=type=cache,sharing=locked,mode=0777,target=/root/.cache/pip,id=pip \
-    pip install flytekit pandas pyarrow pydantic
+    pip install flytekit pandas pyarrow pydantic>2
 RUN --mount=type=cache,sharing=locked,mode=0777,target=/root/.cache/pip,id=pip \
     pip install torch --index-url https://download.pytorch.org/whl/cpu
 

diff --git a/examples/data_types_and_io/data_types_and_io/pydantic_basemodel_attribute_access.py b/examples/data_types_and_io/data_types_and_io/pydantic_basemodel_attribute_access.py
@@ -3,9 +3,8 @@
 
 image_spec = ImageSpec(
     registry="ghcr.io/flyteorg",
-    packages=["pydantic"],
+    packages=["pydantic>2"],
 )
-image_spec = "localhost:30000/flytekit:dev"
 
 
 @task(container_image=image_spec)

diff --git a/examples/data_types_and_io/requirements.in b/examples/data_types_and_io/requirements.in
@@ -3,4 +3,4 @@ torch
 tabulate
 tensorflow
 pyarrow
-pydantic
+pydantic>2