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 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 }}}'