rocm_jax/.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/pre-commit/pre-commit-hooks
  rev: 2c9f875913ee60ca25ce70243dc24d5b6415598c  # frozen: v4.6.0
  hooks:
  - id: check-ast
  - id: check-merge-conflict
  - id: check-toml
  - id: check-yaml
  - id: end-of-file-fixer
    # only include python files
    files: \.py$
  - id: debug-statements
    # only include python files
    files: \.py$
  - id: trailing-whitespace
    # only include python files
    files: \.py$

- repo: https://github.com/astral-sh/ruff-pre-commit
  rev: 8983acb92ee4b01924893632cf90af926fa608f0  # frozen: v0.7.0
  hooks:
  - id: ruff

- repo: https://github.com/pre-commit/mirrors-mypy
  rev: 'bbc3dc1f890007061f18f17e2334f216ea9e5df7'  # frozen: v1.14.1
  hooks:
  - id: mypy
    files: (jax/|tests/typing_test\.py)
    exclude: jax/_src/basearray.py|jax/numpy/__init__.py  # Use pyi instead
    additional_dependencies: [types-requests==2.31.0, jaxlib, numpy>=2.2.0]
    args: [--config=pyproject.toml]

- repo: https://github.com/mwouts/jupytext
  rev: 8ed836db64ad5d304f2315e6bfd9049c9142e190  # frozen: v1.16.4
  hooks:
  - id: jupytext
    files: docs/
    args: [--sync]
DOC: use jupytext to maintain synced markdown versions of notebooks. 2021-02-10 06:39:35 -08:00			`# 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:`
add trailing-whitespace pre-commit hook 2023-03-23 21:17:54 +00:00			`- repo: https://github.com/pre-commit/pre-commit-hooks`
pre-commit: update hooks & pin using hashes 2024-08-27 15:23:13 -07:00			`rev: 2c9f875913ee60ca25ce70243dc24d5b6415598c # frozen: v4.6.0`
add trailing-whitespace pre-commit hook 2023-03-23 21:17:54 +00:00			`hooks:`
CI: add more pre-commit checks 2023-11-29 16:33:55 -08:00			`- id: check-ast`
			`- id: check-merge-conflict`
			`- id: check-toml`
			`- id: check-yaml`
CI: add EOF and debug precommit hooks 2023-03-30 13:29:50 -07:00			`- id: end-of-file-fixer`
			`# only include python files`
			`files: \.py$`
			`- id: debug-statements`
			`# only include python files`
			`files: \.py$`
add trailing-whitespace pre-commit hook 2023-03-23 21:17:54 +00:00			`- id: trailing-whitespace`
			`# only include python files`
			`files: \.py$`

Switch from flake8 to Ruff 2023-11-14 23:34:30 -05:00			`- repo: https://github.com/astral-sh/ruff-pre-commit`
Bumped mypy and ruff versions used by pre-commit 2024-10-21 13:17:16 +01:00			`rev: 8983acb92ee4b01924893632cf90af926fa608f0 # frozen: v0.7.0`
Move flake8 & mypy checks to pre-commit 2021-02-10 13:44:40 -08:00			`hooks:`
Switch from flake8 to Ruff 2023-11-14 23:34:30 -05:00			`- id: ruff`
Move flake8 & mypy checks to pre-commit 2021-02-10 13:44:40 -08:00
			`- repo: https://github.com/pre-commit/mirrors-mypy`
Migrated to mypy 1.14.1 with --allow_redefinition I initially wanted to upgrade to 1.15, but it seems to have a bug in how ternary expressions are type checked. For example, def f(x: int) -> str: ... def g(x: int) -> str: ... callback = f if ... else g # has type object! 2025-02-12 13:27:29 +00:00			`rev: 'bbc3dc1f890007061f18f17e2334f216ea9e5df7' # frozen: v1.14.1`
Move flake8 & mypy checks to pre-commit 2021-02-10 13:44:40 -08:00			`hooks:`
			`- id: mypy`
Add typing_test.py 2022-09-13 12:43:51 -07:00			`files: (jax/\|tests/typing_test\.py)`
Add a type stub for jax.numpy. This type stub is intended to match what pytype currently infers for jax.numpy, which is not particularly accurate in many cases. Future changes will add more accurate types to this stub. Fix a number of new type errors this reveals to mypy. PiperOrigin-RevId: 559179804 2023-08-22 11:50:09 -07:00			`exclude: jax/_src/basearray.py\|jax/numpy/__init__.py # Use pyi instead`
Fix type annotations for NumPy 2.2 2024-12-10 13:38:51 -08:00			`additional_dependencies: [types-requests==2.31.0, jaxlib, numpy>=2.2.0]`
feat: move configurations to pyproject 2023-04-15 02:39:39 +01:00			`args: [--config=pyproject.toml]`
Move flake8 & mypy checks to pre-commit 2021-02-10 13:44:40 -08:00
			`- repo: https://github.com/mwouts/jupytext`
pre-commit: update hooks & pin using hashes 2024-08-27 15:23:13 -07:00			`rev: 8ed836db64ad5d304f2315e6bfd9049c9142e190 # frozen: v1.16.4`
Move flake8 & mypy checks to pre-commit 2021-02-10 13:44:40 -08:00			`hooks:`
			`- id: jupytext`
Speed up pre-commit by filtering jupytext to only run on files in docs. I noticed that the `jupytext` pre-commit was running slowly because it reads a lot of files that it doesn't need to. I think that the only synced notebooks are in `docs`, and restricting this pre-commit to only run in this directory decreases the runtime of `pre-commit run --all-files` on my machine from 37s to 15s. 2024-07-18 11:27:31 -04:00			`files: docs/`
Move flake8 & mypy checks to pre-commit 2021-02-10 13:44:40 -08:00			`args: [--sync]`