Install git on initializer images

Signed-off-by: Andrey Velichkevich <[email protected]>
kubeflow · Dec 10, 2024 · 7c4cb22 · 7c4cb22
1 parent 8e8243c
commit 7c4cb22
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/cmd/initializer_v2/dataset/Dockerfile b/cmd/initializer_v2/dataset/Dockerfile
@@ -9,6 +9,9 @@ COPY pkg/initializer_v2 pkg/initializer_v2
 # Install the needed packages.
 RUN pip install -r requirements.txt
 
+# Git is needed for the Kubeflow Training SDK to download JobSet Python models.
+RUN apk update && apk add --no-cache git
+
 # Copy and install the Kubeflow Training SDK for the configs.
 COPY sdk_v2 sdk_v2
 COPY LICENSE LICENSE

diff --git a/cmd/initializer_v2/model/Dockerfile b/cmd/initializer_v2/model/Dockerfile
@@ -9,6 +9,9 @@ COPY pkg/initializer_v2 pkg/initializer_v2
 # Install the needed packages.
 RUN pip install -r requirements.txt
 
+# Git is needed for the Kubeflow Training SDK to download JobSet Python models.
+RUN apk update && apk add --no-cache git
+
 # Copy and install the Kubeflow Training SDK for the configs.
 COPY sdk_v2 sdk_v2
 COPY LICENSE LICENSE