diff --git a/.github/workflows/ci-health.yml b/.github/workflows/ci-health.yml index db94ce084a..f90968ad06 100644 --- a/.github/workflows/ci-health.yml +++ b/.github/workflows/ci-health.yml @@ -36,3 +36,75 @@ jobs: with: project-name: sagemaker-python-sdk-ci-health-canaries-v2 source-version: refs/heads/master-v2 + unit-tests-v2: + runs-on: ubuntu-latest + strategy: + fail-fast: false + matrix: + python-version: ["py39", "py310", "py311", "py312"] + steps: + - name: Configure AWS Credentials + uses: aws-actions/configure-aws-credentials@v4 + with: + role-to-assume: ${{ secrets.CI_AWS_ROLE_ARN }} + aws-region: us-west-2 + role-duration-seconds: 10800 + - name: Run Unit Tests (${{ matrix.python-version }}) + uses: aws-actions/aws-codebuild-run-build@v1 + with: + project-name: sagemaker-python-sdk-ci-health-unit-tests + source-version: refs/heads/master-v2 + env-vars-for-codebuild: | + PY_VERSION + env: + PY_VERSION: ${{ matrix.python-version }} + slow-tests-v2: + runs-on: ubuntu-latest + steps: + - name: Configure AWS Credentials + uses: aws-actions/configure-aws-credentials@v4 + with: + role-to-assume: ${{ secrets.CI_AWS_ROLE_ARN }} + aws-region: us-west-2 + role-duration-seconds: 10800 + - name: Run Slow Tests + uses: aws-actions/aws-codebuild-run-build@v1 + with: + project-name: sagemaker-python-sdk-ci-health-slow-tests + source-version: refs/heads/master-v2 + localmode-tests-v2: + runs-on: ubuntu-latest + steps: + - name: Configure AWS Credentials + uses: aws-actions/configure-aws-credentials@v4 + with: + role-to-assume: ${{ secrets.CI_AWS_ROLE_ARN }} + aws-region: us-west-2 + role-duration-seconds: 10800 + - name: Run Local Mode Tests + uses: aws-actions/aws-codebuild-run-build@v1 + with: + project-name: sagemaker-python-sdk-ci-health-localmode-tests + source-version: refs/heads/master-v2 + unit-test-v3: + runs-on: ubuntu-latest + strategy: + fail-fast: false + matrix: + submodule: [sagemaker-core, sagemaker-train, sagemaker-serve, sagemaker-mlops] + steps: + - name: Configure AWS Credentials + uses: aws-actions/configure-aws-credentials@v4 + with: + role-to-assume: ${{ secrets.CI_AWS_ROLE_ARN }} + aws-region: us-west-2 + role-duration-seconds: 10800 + - name: Run Unit Tests V3 for ${{ matrix.submodule }} + uses: aws-actions/aws-codebuild-run-build@v1 + with: + project-name: sagemaker-python-sdk-ci-health-unit-test-v3 + source-version: refs/heads/master + env-vars-for-codebuild: | + SUBMODULE + env: + SUBMODULE: ${{ matrix.submodule }}