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