torchgeo/.pre-commit-config.yaml

repos:
  - repo: https://github.com/astral-sh/ruff-pre-commit
    rev: v0.3.7
    hooks:
      - id: ruff
        types_or:
          - python
          - jupyter
        args:
          - --fix
      - id: ruff-format
        types_or:
          - python
          - jupyter
  - repo: https://github.com/pre-commit/mirrors-mypy
    rev: v1.9.0
    hooks:
      - id: mypy
        args:
          - --strict
          - --ignore-missing-imports
          - --show-error-codes
        additional_dependencies:
          - einops>=0.6.0
          - kornia>=0.6.9
          - lightning>=2.0.9
          - matplotlib>=3.8.1
          - numpy>=1.22
          - pillow>=10.3.0
          - pytest>=6.1.2
          - pyvista>=0.34.2
          - scikit-image>=0.22.0
          - torch>=2.2
          - torchmetrics>=0.10
        exclude: (build|data|dist|logo|logs|output)/