.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: v4.5.0
  hooks:
  - 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: v0.1.5
  hooks:
  - id: ruff

- repo: https://github.com/pre-commit/mirrors-mypy
  rev: 'v1.6.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==0.4.16, ml_dtypes==0.2.0, numpy==1.26.0, scipy==1.11.2]
    args: [--config=pyproject.toml]

- repo: https://github.com/mwouts/jupytext
  rev: v1.15.2
  hooks:
  - id: jupytext
    args: [--sync]