.pre-commit-config.yaml

repos:
    -   repo: https://github.com/asottile/pyupgrade
        rev: v2.37.3
        hooks:
        -   id: pyupgrade
            args: [--py37-plus]

    -   repo: https://github.com/pycqa/isort
        rev: 5.10.1
        hooks:
        -    id: isort
             additional_dependencies: ["colorama>=0.4.3"]

    -   repo: https://github.com/psf/black
        rev: 22.6.0
        hooks:
        -   id: black
            args: [--skip-magic-trailing-comma]

    -   repo: https://gitlab.com/pycqa/flake8.git
        rev: 3.9.2
        hooks:
        -   id: flake8

    -   repo: https://github.com/pycqa/pydocstyle
        rev: 6.1.1
        hooks:
        -   id: pydocstyle
            exclude: (tests|docs|experiments)
            additional_dependencies: ["toml"]

    -   repo: https://github.com/pre-commit/mirrors-mypy
        rev: v0.971
        hooks:
        -   id: mypy
            args: [--strict, --ignore-missing-imports, --show-error-codes]
            additional_dependencies: [torch>=1.11, torchmetrics>=0.7, pytorch-lightning>=1.7, pytest>=6, omegaconf>=2.1, kornia>=0.6, numpy>=1.22.0]
            exclude: (build|data|dist|logo|logs|output)/