From 9a03f5112717adc93683b014349ba49aa6553ca7 Mon Sep 17 00:00:00 2001
From: Future-Outlier <eric901201@gmail.com>
Date: Wed, 20 Nov 2024 11:48:51 +0800
Subject: [PATCH] add pydantic>2

Signed-off-by: Future-Outlier <eric901201@gmail.com>
---
 dev-requirements.in                                            | 2 +-
 examples/data_types_and_io/Dockerfile                          | 2 +-
 .../data_types_and_io/pydantic_basemodel_attribute_access.py   | 3 +--
 examples/data_types_and_io/requirements.in                     | 2 +-
 4 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/dev-requirements.in b/dev-requirements.in
index d31cfd256..203c9c2f5 100644
--- 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
index c4b7079b3..1d9af3994 100644
--- 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
index b1b388c39..8ab19694c 100644
--- 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
index b2026eec6..502469658 100644
--- 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