Skip to content

Commit b8b01a2

Browse files
authored
change: fix notebook tests (#900)
1 parent ce82bfd commit b8b01a2

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

buildspec-notebooktests.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ phases:
77
- echo "running notebook test"
88
- start_time=`date +%s`
99
- |
10-
if has-matching-changes "src/*.py" "setup.py" "setup.cfg"; then
10+
if has-matching-changes "src/*.py" "setup.py" "setup.cfg" "tests/scripts/run-notebook-test.sh"; then
1111
./tests/scripts/run-notebook-test.sh
1212
fi
1313
- ./ci-scripts/displaytime.sh 'notebook test' $start_time

tests/scripts/run-notebook-test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ tar -xzf mead-nb-test.tar.gz
1212
git clone --depth 1 https://github.com/awslabs/amazon-sagemaker-examples.git
1313
export JAVA_HOME=$(get-java-home)
1414
echo "set JAVA_HOME=$JAVA_HOME"
15-
export SAGEMAKER_ROLE_ARN=$(get-sagemaker-role-arn)
15+
export SAGEMAKER_ROLE_ARN=$(aws iam list-roles --output text --query "Roles[?RoleName == 'SageMakerRole'].Arn")
1616
echo "set SAGEMAKER_ROLE_ARN=$SAGEMAKER_ROLE_ARN"
1717
./runtime/bin/mead-run-nb-test \
1818
--instance-type ml.c4.8xlarge \

0 commit comments

Comments
 (0)