cuvs/.pre-commit-config.yaml at main · jamxia155/cuvs · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
# SPDX-FileCopyrightText: Copyright (c) 2022-2026, NVIDIA CORPORATION.
# SPDX-License-Identifier: Apache-2.0

repos:
      - repo: https://github.com/pre-commit/pre-commit-hooks
        rev: v5.0.0
        hooks:
              - id: check-json
              - id: trailing-whitespace
              - id: end-of-file-fixer
              - id: check-symlinks
      - repo: https://github.com/PyCQA/isort
        rev: 5.12.0
        hooks:
              - id: isort
                # Use the config file specific to each subproject so that each
                # project can specify its own first/third-party packages.
                args: ["--config-root=python/", "--resolve-all-configs"]
                files: python/.*
                types: [cython]
      - repo: https://github.com/astral-sh/ruff-pre-commit
        rev: v0.14.3
        hooks:
              - id: ruff-check
                args: [--fix]
              - id: ruff-format
      - repo: https://github.com/pre-commit/mirrors-mypy
        rev: 'v0.971'
        hooks:
              - id: mypy
                additional_dependencies: [types-cachetools]
                args: ["--config-file=pyproject.toml",
                       "python/cuvs/cuvs"]
                pass_filenames: false
      - repo: https://github.com/PyCQA/pydocstyle
        rev: 6.1.1
        hooks:
              - id: pydocstyle
                # https://github.com/PyCQA/pydocstyle/issues/603
                additional_dependencies: [toml]
                args: ["--config=pyproject.toml"]
      - repo: https://github.com/pre-commit/mirrors-clang-format
        rev: v20.1.4
        hooks:
              - id: clang-format
                types_or: [c, c++, cuda]
                args: ["-fallback-style=none", "-style=file", "-i"]
      - repo: local
        hooks:
              - id: no-deprecationwarning
                name: no-deprecationwarning
                description: 'Enforce that DeprecationWarning is not introduced (use FutureWarning instead)'
                entry: '(category=|\s)DeprecationWarning[,)]'
                language: pygrep
                types_or: [python, cython]
              - id: cmake-format
                name: cmake-format
                entry: ./cpp/scripts/run-cmake-format.sh cmake-format
                language: python
                types: [cmake]
                exclude: .*/thirdparty/.*|.*FindAVX.cmake.*
                # Note that pre-commit autoupdate does not update the versions
                # of dependencies, so we'll have to update this manually.
                additional_dependencies:
                  - cmakelang==0.6.13
                verbose: true
                require_serial: true
              - id: cmake-lint
                name: cmake-lint
                entry: ./cpp/scripts/run-cmake-format.sh cmake-lint
                language: python
                types: [cmake]
                # Note that pre-commit autoupdate does not update the versions
                # of dependencies, so we'll have to update this manually.
                additional_dependencies:
                  - cmakelang==0.6.13
                verbose: true
                require_serial: true
                exclude: .*/thirdparty/.*
              - id: include-check
                name: include-check
                entry: python ./cpp/scripts/include_checker.py cpp/bench cpp/include cpp/tests
                pass_filenames: false
                language: python
                additional_dependencies: [gitpython]
              - id: cargo-fmt
                name: cargo-fmt
                entry: cargo fmt --manifest-path rust/Cargo.toml --all
                pass_filenames: false
                files: rust/.*
                language: rust
      - repo: https://github.com/codespell-project/codespell
        rev: v2.2.2
        hooks:
              - id: codespell
                additional_dependencies: [tomli]
                args: ["--toml", "pyproject.toml"]
                exclude: |
                  (?x)
                    ^CHANGELOG[.]md$|
                    ^cpp/cmake/patches/cutlass/build-export[.]patch$
      - repo: https://github.com/rapidsai/pre-commit-hooks
        rev: v1.3.3
        hooks:
          - id: verify-copyright
            name: verify-copyright-cuvs
            args: [--fix, --spdx]
            files: |
              (?x)
                [.](cmake|c|cpp|cu|cuh|h|hpp|sh|pxd|py|pyx|rs|java)$|
                CMakeLists[.]txt$|
                CMakeLists_standalone[.]txt$|
                meta[.]yaml$|
                pyproject[.]toml$|
                ^python/cuvs_bench/cuvs_bench/split_groundtruth/split_groundtruth[.]pl$|
                Dockerfile$|
                pom[.]xml$|
                ^java/cuvs-java/src/assembly/native-with-deps[.]xml$|
                ^java/docker-build/build-in-docker$|
                ^java/docker-build/run-in-docker$|
                ^[.]flake8$|
                recipe[.]yaml$|
                ^[.]pre-commit-config[.]yaml$
            exclude: |
              (?x)^(
                docs/source/sphinxext/github_link[.]py$|
                cpp/cmake/modules/FindAVX[.]cmake$|
                cpp/src/neighbors/detail/faiss_distance_utils[.]h$|
                cpp/src/distance/detail/fused_distance_nn/gemm[.]h$|
                cpp/src/distance/detail/fused_distance_nn/epilogue[.]cuh$|
                cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_normvec_smem[.]h$|
                cpp/src/distance/detail/fused_distance_nn/persistent_gemm[.]h$|
                cpp/src/distance/detail/fused_distance_nn/epilogue_elementwise[.]cuh$|
                cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_reduced_vec[.]h$|
                cpp/src/distance/detail/fused_distance_nn/custom_epilogue_with_broadcast[.]h$
              )
          - id: verify-copyright
            name: verify-copyright-scikit-learn
            args: [--fix, --spdx, "--spdx-license-identifier=Apache-2.0 AND BSD-3-Clause"]
            files: |
              (?x)^(
                docs/source/sphinxext/github_link[.]py$
              )
          - id: verify-copyright
            name: verify-copyright-cutlass
            args: [--fix, --spdx, "--spdx-license-identifier=Apache-2.0 AND BSD-3-Clause"]
            files: |
              (?x)^(
                cpp/src/distance/detail/fused_distance_nn/gemm[.]h$|
                cpp/src/distance/detail/fused_distance_nn/epilogue[.]cuh$|
                cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_normvec_smem[.]h$|
                cpp/src/distance/detail/fused_distance_nn/persistent_gemm[.]h$|
                cpp/src/distance/detail/fused_distance_nn/epilogue_elementwise[.]cuh$|
                cpp/src/distance/detail/fused_distance_nn/predicated_tile_iterator_reduced_vec[.]h$|
                cpp/src/distance/detail/fused_distance_nn/custom_epilogue_with_broadcast[.]h$
              )
          - id: verify-copyright
            name: verify-copyright-faiss
            args: [--fix, --spdx, "--spdx-license-identifier=Apache-2.0 AND MIT"]
            files: |
              (?x)^(
                cpp/src/neighbors/detail/faiss_distance_utils[.]h$
              )
          - id: verify-alpha-spec
          - id: verify-codeowners
            args: [--fix, --project-prefix=cuvs]
          - id: verify-pyproject-license
            # ignore the top-level pyproject.toml, which doesn't
            # have or need a [project] table
            exclude: |
              (?x)
                ^pyproject[.]toml$
      - repo: https://github.com/rapidsai/dependency-file-generator
        rev: v1.20.0
        hooks:
            - id: rapids-dependency-file-generator
              args: ["--clean", "--warn-all", "--strict"]
      - repo: https://github.com/shellcheck-py/shellcheck-py
        rev: v0.10.0.1
        hooks:
          - id: shellcheck

default_language_version:
      python: python3