change: Updated argument name for sagemaker-distribution image in tes…

…ts (#274)
aws · Mar 19, 2024 · 68d52e9 · 68d52e9
1 parent ecddf40
commit 68d52e9
Show file tree

Hide file tree

Showing 39 changed files with 77 additions and 77 deletions.
diff --git a/test/test_artifacts/v0/autogluon.test.Dockerfile b/test/test_artifacts/v0/autogluon.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE as base
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE as base
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v0/boto3.test.Dockerfile b/test/test_artifacts/v0/boto3.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN sudo apt-get update && sudo apt-get install -y git && \

diff --git a/test/test_artifacts/v0/glue-sessions.test.Dockerfile b/test/test_artifacts/v0/glue-sessions.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v0/keras.test.Dockerfile b/test/test_artifacts/v0/keras.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v0/matplotlib.test.Dockerfile b/test/test_artifacts/v0/matplotlib.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v0/numpy.test.Dockerfile b/test/test_artifacts/v0/numpy.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v0/pandas.test.Dockerfile b/test/test_artifacts/v0/pandas.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN pip install pandas[test]

diff --git a/test/test_artifacts/v0/pytorch.examples.Dockerfile b/test/test_artifacts/v0/pytorch.examples.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN sudo apt-get update && \

diff --git a/test/test_artifacts/v0/sagemaker-headless-execution-driver.test.Dockerfile b/test/test_artifacts/v0/sagemaker-headless-execution-driver.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v0/sagemaker-studio-analytics-extension.test.Dockerfile b/test/test_artifacts/v0/sagemaker-studio-analytics-extension.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v0/scipy.test.Dockerfile b/test/test_artifacts/v0/scipy.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v0/sm-python-sdk.test.Dockerfile b/test/test_artifacts/v0/sm-python-sdk.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN sudo apt-get update && sudo apt-get install -y git && \

diff --git a/test/test_artifacts/v0/tensorflow.examples.Dockerfile b/test/test_artifacts/v0/tensorflow.examples.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN sudo apt-get update && \

diff --git a/test/test_artifacts/v1/altair.test.Dockerfile b/test/test_artifacts/v1/altair.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/amazon-codewhisperer-jupyterlab-ext.test.Dockerfile b/test/test_artifacts/v1/amazon-codewhisperer-jupyterlab-ext.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/amazon-sagemaker-sql-magic.test.Dockerfile b/test/test_artifacts/v1/amazon-sagemaker-sql-magic.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/amazon_sagemaker_sql_editor.test.Dockerfile b/test/test_artifacts/v1/amazon_sagemaker_sql_editor.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/autogluon.test.Dockerfile b/test/test_artifacts/v1/autogluon.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE as base
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE as base
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/boto3.test.Dockerfile b/test/test_artifacts/v1/boto3.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN sudo apt-get update && sudo apt-get install -y git && \

diff --git a/test/test_artifacts/v1/glue-sessions.test.Dockerfile b/test/test_artifacts/v1/glue-sessions.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/jupyter-ai.test.Dockerfile b/test/test_artifacts/v1/jupyter-ai.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/jupyter-dash.test.Dockerfile b/test/test_artifacts/v1/jupyter-dash.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/jupyterlab-git.test.Dockerfile b/test/test_artifacts/v1/jupyterlab-git.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/jupyterlab-lsp.test.Dockerfile b/test/test_artifacts/v1/jupyterlab-lsp.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/keras.test.Dockerfile b/test/test_artifacts/v1/keras.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/matplotlib.test.Dockerfile b/test/test_artifacts/v1/matplotlib.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/notebook.test.Dockerfile b/test/test_artifacts/v1/notebook.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/numpy.test.Dockerfile b/test/test_artifacts/v1/numpy.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/pandas.test.Dockerfile b/test/test_artifacts/v1/pandas.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN micromamba install -y --freeze-installed -c conda-forge pytest hypothesis pytest-asyncio lxml

diff --git a/test/test_artifacts/v1/python-lsp-server.test.Dockerfile b/test/test_artifacts/v1/python-lsp-server.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/pytorch.examples.Dockerfile b/test/test_artifacts/v1/pytorch.examples.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN sudo apt-get update && \

diff --git a/test/test_artifacts/v1/sagemaker-code-editor.test.Dockerfile b/test/test_artifacts/v1/sagemaker-code-editor.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/sagemaker-headless-execution-driver.test.Dockerfile b/test/test_artifacts/v1/sagemaker-headless-execution-driver.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/sagemaker-studio-analytics-extension.test.Dockerfile b/test/test_artifacts/v1/sagemaker-studio-analytics-extension.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/scipy.test.Dockerfile b/test/test_artifacts/v1/scipy.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/serve.test.Dockerfile b/test/test_artifacts/v1/serve.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 

diff --git a/test/test_artifacts/v1/sm-python-sdk.test.Dockerfile b/test/test_artifacts/v1/sm-python-sdk.test.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN sudo apt-get update && sudo apt-get install -y git && \

diff --git a/test/test_artifacts/v1/tensorflow.examples.Dockerfile b/test/test_artifacts/v1/tensorflow.examples.Dockerfile
@@ -1,5 +1,5 @@
-ARG COSMOS_IMAGE
-FROM $COSMOS_IMAGE
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN sudo apt-get update && \

diff --git a/test/test_dockerfile_based_harness.py b/test/test_dockerfile_based_harness.py
@@ -132,7 +132,7 @@ def _validate_docker_images(dockerfile_path: str, required_packages: List[str],
         image, _ = _docker_client.images.build(path=test_artifacts_path,
                                                dockerfile=dockerfile_path,
                                                tag=dockerfile_path.lower().replace('.', '-'),
-                                               rm=True, buildargs={'COSMOS_IMAGE': docker_image_identifier})
+                                               rm=True, buildargs={'SAGEMAKER_DISTRIBUTION_IMAGE': docker_image_identifier})
     except BuildError as e:
         for line in e.build_log:
             if 'stream' in line: