GPU Test Cleanup #10
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: GPU Test Cleanup | |
| # Remove the "GPU CI" label after GPU tests complete | |
| # This requires maintainers to explicitly re-add the label for each test run | |
| on: | |
| workflow_run: | |
| workflows: ["GPU Test"] | |
| types: [completed] | |
| jobs: | |
| remove-label: | |
| name: Remove GPU CI Label | |
| runs-on: ubuntu-latest | |
| # Only run if there's an associated PR | |
| if: github.event.workflow_run.event == 'pull_request' | |
| steps: | |
| - name: Remove GPU CI label | |
| uses: actions/github-script@v7 | |
| with: | |
| script: | | |
| // Get the PR number from the workflow run | |
| const prNumber = context.payload.workflow_run.pull_requests[0]?.number; | |
| if (!prNumber) { | |
| console.log('No PR number found, skipping label removal'); | |
| return; | |
| } | |
| try { | |
| await github.rest.issues.removeLabel({ | |
| owner: context.repo.owner, | |
| repo: context.repo.repo, | |
| issue_number: prNumber, | |
| name: 'GPU CI' | |
| }); | |
| console.log(`Removed "GPU CI" label from PR #${prNumber}`); | |
| } catch (error) { | |
| // Label might already be removed or not exist | |
| if (error.status === 404) { | |
| console.log('Label not found, may have already been removed'); | |
| } else { | |
| throw error; | |
| } | |
| } |