add uv in next minor versions as requested by partner team (#783)

TRNWWZ · web-flow · commit 99f4f4ddedfe · 2025-07-18T16:23:22.000-07:00
Approve
diff --git a/build_artifacts/v2/v2.9/v2.9.0/cpu.additional_packages_env.in b/build_artifacts/v2/v2.9/v2.9.0/cpu.additional_packages_env.in
@@ -1 +1,2 @@
 conda-forge::mcp
+conda-forge::uv
diff --git a/build_artifacts/v2/v2.9/v2.9.0/gpu.additional_packages_env.in b/build_artifacts/v2/v2.9/v2.9.0/gpu.additional_packages_env.in
@@ -1 +1,2 @@
 conda-forge::mcp
+conda-forge::uv
diff --git a/build_artifacts/v3/v3.4/v3.4.0/cpu.additional_packages_env.in b/build_artifacts/v3/v3.4/v3.4.0/cpu.additional_packages_env.in
@@ -1 +1,2 @@
 conda-forge::mcp
+conda-forge::uv
diff --git a/build_artifacts/v3/v3.4/v3.4.0/gpu.additional_packages_env.in b/build_artifacts/v3/v3.4/v3.4.0/gpu.additional_packages_env.in
@@ -1 +1,2 @@
 conda-forge::mcp
+conda-forge::uv
diff --git a/test/test_artifacts/v2/uv.test.Dockerfile b/test/test_artifacts/v2/uv.test.Dockerfile
@@ -0,0 +1,9 @@
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
+
+ARG MAMBA_DOCKERFILE_ACTIVATE=1
+
+RUN echo 'import requests; print(requests.get("https://astral.sh"))' > example.py
+
+RUN uv add --script example.py requests
+CMD uv run example.py
diff --git a/test/test_artifacts/v3/uv.test.Dockerfile b/test/test_artifacts/v3/uv.test.Dockerfile
@@ -0,0 +1,9 @@
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
+
+ARG MAMBA_DOCKERFILE_ACTIVATE=1
+
+RUN echo 'import requests; print(requests.get("https://astral.sh"))' > example.py
+
+RUN uv add --script example.py requests
+CMD uv run example.py
