.pre-commit-config.yaml

---
files: ^(.*\.(py|md|sh|yaml|yml|in|cfg|txt|rst|toml|precommit-toml|wordlist))$
exclude: ^(\.[^/]*(cache|assets|uv|venv)/.*)$
repos:
  - repo: https://github.com/codespell-project/codespell
    rev: v2.4.1
    hooks:
      - id: codespell
        args:
          - '-w'
          - '--skip="*.txt,pylintrc,.*,src/maxtext/assets/*,src/maxtext/input_pipeline/protos/*"'
          - '-L ND,nd,sems,TE,ROUGE,rouge,astroid,ags,dout'
          - '.'
        additional_dependencies:
          - tomli

  #   - repo: https://github.com/abravalheri/validate-pyproject
  #     rev: v0.24.1
  #     hooks:
  #       - id: validate-pyproject
  #         name: Validate pyproject.toml
  #         language: python
  #         files: '^pyproject\.toml$'
  #         entry: python

  - repo: https://github.com/pylint-dev/pylint
    rev: v3.3.8
    hooks:
      - id: pylint
        args:
          - '--disable=R0401,R0917,W0201,W0613'
          - "--ignore-patterns='.pytype,.*pyi$'"
          - '--ignore-paths=src/maxtext/input_pipeline/protos'
          - 'benchmarks'
          - 'src'
          - 'tests'

  #  - repo: https://github.com/google/pytype
  #    rev: 2024.10.11
  #    hooks:
  #      - id: pytype
  #        args:
  #          - '--jobs=auto'
  #          - '--keep-going'
  #          - 'src/maxtext/'

  - repo: https://github.com/google/pyink
    rev: 24.10.1
    hooks:
      - id: pyink
        exclude: src/maxtext/input_pipeline/protos/
        args:
          - '--pyink-indentation=2'
          - '--line-length=122'

  - repo: https://github.com/executablebooks/mdformat
    rev: 0.7.22
    hooks:
      - id: mdformat
        args: ['--number']
        additional_dependencies: [mdformat-myst, mdformat-ruff]
        files: (docs/.)
        exclude: docs/guides/checkpointing_solutions.md