Skip to main content

Browse the archive

Revision 8d4b50e397ece89be229afa762d8466a03bd443c authored by Skye Wanderman-Milne on 09 December 2022, 22:11:18 UTC, committed by Skye Wanderman-Milne on 09 December 2022, 22:32:09 UTC 
[TPU CI] Run build matrix on v3-8 as well as v4-8
 
We're seeing failures on v3-8 that don't appear on the current v4-8
testing. v3-8 also exposes 8 devices (vs. v4-8 exposes 4), and some
tests needs 8 devices to run.

I just added a v3-8 runner VM.

Also adds a missing pip install command (I only caught this with a
fresh runner since it only needs to be installed once).
1 parent f2c5d28
Raw File
.pre-commit-config.yaml 
# Install the pre-commit hooks below with
# 'pre-commit install'

# Auto-update the version of the hooks with
# 'pre-commit autoupdate'

# Run the hooks on all files with
# 'pre-commit run --all'

repos:
- repo: https://github.com/pycqa/flake8
  rev: '5.0.4'
  hooks:
  - id: flake8

- repo: https://github.com/pre-commit/mirrors-mypy
  rev: 'v0.982'
  hooks:
  - id: mypy
    files: (jax/|tests/typing_test\.py)
    exclude: jax/_src/basearray.py  # Use pyi instead
    additional_dependencies: [types-requests==2.28.11, jaxlib==0.3.24]

- repo: https://github.com/mwouts/jupytext
  rev: v1.14.1
  hooks:
  - id: jupytext
    args: [--sync]
back to top