From b48ae1cde2cd36b53ff2c5cd56e2c97980105fd3 Mon Sep 17 00:00:00 2001 From: Ruinong Tian Date: Tue, 4 Feb 2025 23:40:06 +0000 Subject: [PATCH] Fixing sagemaker-python-sdk tests 1. specify aws region in .sh file instead of Dockerfile 2. upgrade pytest version specifically --- test/test_artifacts/v2/scripts/run_pysdk_tests.sh | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test/test_artifacts/v2/scripts/run_pysdk_tests.sh b/test/test_artifacts/v2/scripts/run_pysdk_tests.sh index 2500585a..ab969556 100644 --- a/test/test_artifacts/v2/scripts/run_pysdk_tests.sh +++ b/test/test_artifacts/v2/scripts/run_pysdk_tests.sh @@ -1,8 +1,11 @@ #!/bin/bash -# We need to checkout the version of sagemaker-python-sdk that is installed in the mamba environment. +# Sagemaker Python SDK's unit tests requires AWS_DEFAULT_REGION to be set. So, using an arbitrary value of us-east-1 +aws configure set default.region us-east-1 +# We need to checkout the version of sagemaker-python-sdk that is installed in the mamba environment. pysdk_version=$(micromamba list | grep sagemaker-python-sdk | tr -s ' ' | cut -d ' ' -f 3) + # Checkout the corresponding sagemaker-python-sdk version git checkout tags/v$pysdk_version