From 1ef4517dd81f1d49d708123aa7cee0f160d0d59b Mon Sep 17 00:00:00 2001 From: Lucas Jia Date: Thu, 28 May 2026 16:02:17 -0700 Subject: [PATCH 1/2] test: add a job to invoke code build project in another account --- .github/workflows/pr-checks-master.yml | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/.github/workflows/pr-checks-master.yml b/.github/workflows/pr-checks-master.yml index 4f63ad0b9a..3c3ceea5c7 100644 --- a/.github/workflows/pr-checks-master.yml +++ b/.github/workflows/pr-checks-master.yml @@ -215,3 +215,21 @@ jobs: with: project-name: ${{ github.event.repository.name }}-ci-${{ matrix.submodule }}-integ-tests source-version-override: 'refs/pull/${{ github.event.pull_request.number }}/head^{${{ github.event.pull_request.head.sha }}}' + + integ-tests-us-east-1: + runs-on: ubuntu-latest + needs: [detect-changes] + if: contains(fromJson(needs.detect-changes.outputs.submodules), 'sagemaker-train') + steps: + - name: Configure AWS Credentials (us-east-1) + uses: aws-actions/configure-aws-credentials@v4 + with: + role-to-assume: ${{ secrets.CI_AWS_ROLE_US_EAST_1_ARN }} + aws-region: us-east-1 + role-duration-seconds: 10800 + + - name: Run us-east-1 Integ Tests for sagemaker-train + uses: aws-actions/aws-codebuild-run-build@v1 + with: + project-name: ${{ github.event.repository.name }}-ci-sagemaker-train-integ-tests + source-version-override: 'refs/pull/${{ github.event.pull_request.number }}/head^{${{ github.event.pull_request.head.sha }}}' From 9c85865d388a0de23b68b0352ff468ed5ffbd9af Mon Sep 17 00:00:00 2001 From: Lucas Jia Date: Thu, 28 May 2026 16:39:55 -0700 Subject: [PATCH 2/2] test: add job for gpu_intensive workflow to run nova tests in us east 1 --- .github/workflows/gpu-integ-tests.yml | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/.github/workflows/gpu-integ-tests.yml b/.github/workflows/gpu-integ-tests.yml index a53d76d8e3..9c9fb3e01b 100644 --- a/.github/workflows/gpu-integ-tests.yml +++ b/.github/workflows/gpu-integ-tests.yml @@ -22,3 +22,18 @@ jobs: with: project-name: sagemaker-python-sdk-ci-health-gpu-integ-tests source-version: refs/heads/master + + gpu-integ-tests-us-east-1: + runs-on: ubuntu-latest + steps: + - name: Configure AWS Credentials (us-east-1) + uses: aws-actions/configure-aws-credentials@v4 + with: + role-to-assume: ${{ secrets.CI_AWS_ROLE_US_EAST_1_ARN }} + aws-region: us-east-1 + role-duration-seconds: 10800 + - name: Run GPU Integ Tests (us-east-1) + uses: aws-actions/aws-codebuild-run-build@v1 + with: + project-name: sagemaker-python-sdk-ci-health-gpu-integ-tests + source-version: refs/heads/master