From 8af24f6cc9b05d2bf952817e2c7c4e3d4d22ff0b Mon Sep 17 00:00:00 2001
From: Apoorva Kulkarni <kuapoorv@amazon.com>
Date: Wed, 18 Sep 2024 15:27:51 -0700
Subject: [PATCH] chore: Update cmd-shell image to python3.11 (#656)

---
 .../examples/docker/Dockerfile-cmd-shell                   | 7 +++++++
 ai-ml/trainium-inferentia/fsx-for-lustre.tf                | 2 +-
 2 files changed, 8 insertions(+), 1 deletion(-)
 create mode 100644 ai-ml/trainium-inferentia/examples/docker/Dockerfile-cmd-shell

diff --git a/ai-ml/trainium-inferentia/examples/docker/Dockerfile-cmd-shell b/ai-ml/trainium-inferentia/examples/docker/Dockerfile-cmd-shell
new file mode 100644
index 000000000..15e3773bb
--- /dev/null
+++ b/ai-ml/trainium-inferentia/examples/docker/Dockerfile-cmd-shell
@@ -0,0 +1,7 @@
+FROM public.ecr.aws/amazonlinux/amazonlinux:2023
+
+RUN yum update -y && yum install -y awscli-2
+
+RUN python3 -m ensurepip --upgrade
+
+RUN pip3 install -U "huggingface_hub[cli]"
diff --git a/ai-ml/trainium-inferentia/fsx-for-lustre.tf b/ai-ml/trainium-inferentia/fsx-for-lustre.tf
index f6a2b45fd..6c88aa2cb 100644
--- a/ai-ml/trainium-inferentia/fsx-for-lustre.tf
+++ b/ai-ml/trainium-inferentia/fsx-for-lustre.tf
@@ -101,7 +101,7 @@ metadata:
 spec:
   containers:
   - name: app
-    image: public.ecr.aws/aws-cli/aws-cli:2.13.1
+    image: public.ecr.aws/data-on-eks/cmd-shell:py3-awscli2
     command: ["/bin/sh", "-c"]
     args: ["while true; do sleep 30; done"]
     volumeMounts: