Skip to content

Commit

Permalink
Fixing sagemaker-python-sdk tests
Browse files Browse the repository at this point in the history
  1. specify aws region in .sh file instead of Dockerfile
  2. upgrade pytest version specifically
  • Loading branch information
Ruinong Tian committed Feb 4, 2025
1 parent 398d20c commit b48ae1c
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion test/test_artifacts/v2/scripts/run_pysdk_tests.sh
Original file line number Diff line number Diff line change
@@ -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

Expand Down

0 comments on commit b48ae1c

Please sign in to comment.