fix: add branch name to remote gpu test run command (#195)

Author: icywang86rui

buildspec-release.yml

@@ -31,24 +31,25 @@ phases:
       # Build all images
       - python3 scripts/build_all.py --account $ACCOUNT --region $AWS_DEFAULT_REGION
 
+      # Publish all images
+      - python3 scripts/publish_all.py --account $ACCOUNT --region $AWS_DEFAULT_REGION
+
+      - base_name="$ACCOUNT.dkr.ecr.$AWS_DEFAULT_REGION.amazonaws.com/sagemaker-tensorflow-scriptmode"
       # run local cpu integ tests
       - $(aws ecr get-login --registry-ids $ACCOUNT --no-include-email --region $AWS_DEFAULT_REGION)
-      - IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --docker-base-name sagemaker-tensorflow-scriptmode --framework-version $FRAMEWORK_VERSION --processor cpu
-      - IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --docker-base-name sagemaker-tensorflow-scriptmode --py-version 2 --framework-version $FRAMEWORK_VERSION --processor cpu
+      - IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --docker-base-name $base_name --framework-version $FRAMEWORK_VERSION --processor cpu
+      - IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --docker-base-name $base_name --py-version 2 --framework-version $FRAMEWORK_VERSION --processor cpu
 
       # launch remote gpu instance
       - prefix='ml.'
       - instance_type=${GPU_INSTANCE_TYPE#"$prefix"}
       - create-key-pair
       - launch-ec2-instance --instance-type $instance_type --ami-name dlami-ubuntu
 
-      - py2_cmd="IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --docker-base-name sagemaker-tensorflow-scriptmode --py-version 2 --framework-version $FRAMEWORK_VERSION --processor gpu"
-      - remote-test --github-repo $GITHUB_REPO --test-cmd "$py2_cmd"
-      - py3_cmd="IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --docker-base-name sagemaker-tensorflow-scriptmode --framework-version $FRAMEWORK_VERSION --processor gpu"
-      - remote-test --github-repo $GITHUB_REPO --test-cmd "$py3_cmd"
-
-      # Publish all images
-      - python3 scripts/publish_all.py --account $ACCOUNT --region $AWS_DEFAULT_REGION
+      - py2_cmd="IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --docker-base-name $base_name --py-version 2 --framework-version $FRAMEWORK_VERSION --processor gpu"
+      - remote-test --github-repo sagemaker-tensorflow-container --branch script-mode --test-cmd "$py2_cmd"
+      - py3_cmd="IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --docker-base-name $base_name --framework-version $FRAMEWORK_VERSION --processor gpu"
+      - remote-test --github-repo sagemaker-tensorflow-container --branch script-mode --test-cmd "$py3_cmd"
 
       - |
         echo '[{