[Bug Fix] Fix a nit issue for Jax rollback #38
Workflow file for this run
This file contains 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: pre-commit | |
| on: pull_request | |
| jobs: | |
| pre-commit: | |
| runs-on: ubuntu-latest | |
| # resource_class: large | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-python@v5 | |
| with: | |
| python-version: '3.10' | |
| cache: 'pip' | |
| - run: pip install --upgrade pip | |
| # TODO(markblee): Remove gcp,vertexai_tensorboard from CI. (needed by pytype) | |
| - run: pip install '.[core,dev,grain,gcp,vertexai_tensorboard]' | |
| # pylint uses approx 12GB of memory during this run, look into split to decrease? | |
| - run: pre-commit run --all-files | |
| - run: pytype -j auto . |