mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00
cca3961cde
Include _src/distributed.py and _src/clusters/*.py in the same target because they are in a strongly-connected component. [XLA:Python] Set type of ArrayImpl to Any, since the JAX change now allows pytype to see that some values are ArrayImpls but ArrayImpls are not instances of jax.Array to Pytype. Fix type of buffer_from_pyval. PiperOrigin-RevId: 515687258
29 lines
641 B
YAML
29 lines
641 B
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: '6.0.0'
|
|
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.4.6]
|
|
|
|
- repo: https://github.com/mwouts/jupytext
|
|
rev: v1.14.4
|
|
hooks:
|
|
- id: jupytext
|
|
args: [--sync]
|