diff --git a/examples/data_types_and_io/Dockerfile b/examples/data_types_and_io/Dockerfile
index d93ea429b..c4b7079b3 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
+    pip install flytekit pandas pyarrow pydantic
 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