Update base image (#1677)

Signed-off-by: Kevin Su <pingsutw@apache.org>
flyteorg · May 13, 2024 · a4cd567 · a4cd567
1 parent 3245aac
commit a4cd567
Show file tree

Hide file tree

Showing 8 changed files with 8 additions and 12 deletions.
diff --git a/.github/workflows/checks.yml b/.github/workflows/checks.yml
@@ -53,7 +53,7 @@ jobs:
           # TODO: Register and update the examples below. (onnx_plugin, feast_integration, etc)
           echo "PACKAGES=$(find examples -maxdepth 1 -type d -exec basename '{}' \; \
           | grep -v -e 'testing' -e 'examples' \
-          | grep -v -e 'airflow_plugin' -e 'forecasting_sales' -e 'onnx_plugin' -e 'feast_integration' -e 'modin_plugin' -e 'sagemaker_inference_agent' \
+          | grep -v -e 'airflow_plugin' -e 'forecasting_sales' -e 'onnx_plugin' -e 'feast_integration' -e 'modin_plugin' -e 'sagemaker_inference_agent' -e 'mnist_classifier' \
           | sort \
           | jq --raw-input . \
           | jq --slurp . \

diff --git a/examples/databricks_agent/Dockerfile b/examples/databricks_agent/Dockerfile
@@ -1,4 +1,4 @@
-FROM databricksruntime/standard:12.2-LTS
+FROM databricksruntime/standard:14.3-LTS
 LABEL org.opencontainers.image.source=https://github.com/flyteorg/flytesnacks
 
 ENV VENV /opt/venv

diff --git a/examples/databricks_plugin/Dockerfile b/examples/databricks_plugin/Dockerfile
@@ -1,4 +1,4 @@
-FROM databricksruntime/standard:12.2-LTS
+FROM databricksruntime/standard:14.3-LTS
 LABEL org.opencontainers.image.source=https://github.com/flyteorg/flytesnacks
 
 ENV VENV /opt/venv

diff --git a/examples/k8s_dask_plugin/Dockerfile b/examples/k8s_dask_plugin/Dockerfile
@@ -1,7 +1,7 @@
 # ######################
 # NOTE: For CI/CD only #
 ########################
-FROM ubuntu:focal
+FROM ubuntu:latest
 LABEL org.opencontainers.image.source https://github.com/flyteorg/flytesnacks
 
 WORKDIR /root

diff --git a/examples/k8s_dask_plugin/k8s_dask_plugin/dask_example.py b/examples/k8s_dask_plugin/k8s_dask_plugin/dask_example.py
@@ -12,7 +12,7 @@
 # %% [markdown]
 # Create an `ImageSpec` to encompass all the dependencies needed for the Dask task.
 # %%
-custom_image = ImageSpec(name="flyte-dask-plugin", registry="ghcr.io/flyteorg", packages=["flytekitplugins-dask"])
+custom_image = ImageSpec(registry="ghcr.io/flyteorg", packages=["flytekitplugins-dask"])
 
 # %% [markdown]
 # :::{important}

diff --git a/examples/k8s_spark_plugin/Dockerfile b/examples/k8s_spark_plugin/Dockerfile
@@ -1,7 +1,7 @@
 # ######################
 # NOTE: For CI/CD only #
 ########################
-FROM apache/spark-py:3.3.1
+FROM apache/spark-py:v3.4.0
 LABEL org.opencontainers.image.source https://github.com/flyteorg/flytesnacks
 
 WORKDIR /root

diff --git a/examples/k8s_spark_plugin/k8s_spark_plugin/dataframe_passing.py b/examples/k8s_spark_plugin/k8s_spark_plugin/dataframe_passing.py
@@ -14,11 +14,7 @@
 from flytekit import ImageSpec, Resources, kwtypes, task, workflow
 from flytekit.types.structured.structured_dataset import StructuredDataset
 from flytekitplugins.spark import Spark
-
-try:
-    from typing import Annotated
-except ImportError:
-    from typing_extensions import Annotated
+from typing_extensions import Annotated
 
 # %% [markdown]
 # Create an `ImageSpec` to automate the retrieval of a prebuilt Spark image.

diff --git a/examples/mnist_classifier/Dockerfile b/examples/mnist_classifier/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
+FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
 LABEL org.opencontainers.image.source https://github.com/flyteorg/flytesnacks
 
 WORKDIR /root