diff --git a/.github/workflows/labeler.yaml b/.github/workflows/labeler.yaml index 350fd262c16..a52950ba8d1 100644 --- a/.github/workflows/labeler.yaml +++ b/.github/workflows/labeler.yaml @@ -31,7 +31,7 @@ permissions: pull-requests: write jobs: - welcome: + add_labels: runs-on: ubuntu-latest steps: - uses: actions/checkout@v5 diff --git a/.github/workflows/tpu_tests.yml b/.github/workflows/tpu_tests.yml index ca4ff3b9701..3cb318bb4eb 100644 --- a/.github/workflows/tpu_tests.yml +++ b/.github/workflows/tpu_tests.yml @@ -1,15 +1,10 @@ -name: Keras Tests - -# TODO: Consider enabling all tests (pytest, applications, etc.) with NNX in the future -# Currently only basic flow tests run with NNX enabled +name: Keras TPU Tests on: push: branches: [ master ] pull_request: types: [labeled] - pull_request_review: - types: [submitted] release: types: [created] @@ -21,12 +16,11 @@ jobs: test-in-container: name: Run tests on TPU runs-on: linux-x86-ct6e-44-1tpu - # Only run on approved PRs, pushes to master, releases or "run_tpu_tests" labels + # Only run on pushes to master, releases or "run_tpu_tests" labels if: | github.event_name == 'push' || github.event_name == 'release' || - (github.event_name == 'pull_request' && github.event.action == 'labeled' && github.event.label.name == 'run_tpu_tests') || - (github.event_name == 'pull_request_review' && github.event.review.state == 'approved') + (github.event_name == 'pull_request' && github.event.action == 'labeled' && github.event.label.name == 'run_tpu_tests') strategy: fail-fast: false diff --git a/.github/workflows/tpu_tests_labeler.yml b/.github/workflows/tpu_tests_labeler.yml new file mode 100644 index 00000000000..0f1eeaffa52 --- /dev/null +++ b/.github/workflows/tpu_tests_labeler.yml @@ -0,0 +1,31 @@ +name: Labeler for TPU Tests + +on: + pull_request_review: + types: [submitted] + +permissions: + contents: read + issues: write + pull-requests: write + +jobs: + add_run_tpu_tests_label: + runs-on: ubuntu-latest + + # Only run on "approved" reviews for the "pull_request_review" event. + if: | + github.event_name == 'pull_request_review' && + github.event.review.state == 'approved' + + steps: + - uses: actions/github-script@v8 + with: + github-token: ${{ secrets.GITHUB_TOKEN }} + script: | + github.rest.issues.addLabels({ + owner: context.repo.owner, + repo: context.repo.repo, + issue_number: context.payload.pull_request.number, + labels: ['run_tpu_tests'] + })