fix dataframe_passing.py (#1728)

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Sep 9, 2024 · beab2f9 · beab2f9
1 parent 8613c49
commit beab2f9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/examples/k8s_spark_plugin/k8s_spark_plugin/dataframe_passing.py b/examples/k8s_spark_plugin/k8s_spark_plugin/dataframe_passing.py
@@ -19,7 +19,9 @@
 # %% [markdown]
 # Create an `ImageSpec` to automate the retrieval of a prebuilt Spark image.
 # %%
-custom_image = ImageSpec(python_version="3.9", registry="ghcr.io/flyteorg", packages=["flytekitplugins-spark"])
+custom_image = ImageSpec(
+    python_version="3.9", registry="ghcr.io/flyteorg", packages=["flytekitplugins-spark", "pyarrow"]
+)
 
 # %% [markdown]
 # :::{important}