fix merge

rapidsai · Sep 21, 2023 · 6811e61 · 6811e61
1 parent a733ffe
commit 6811e61
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -8,21 +8,21 @@ repos:
                 # 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|bench/ann)/.*
+                files: python/.*
                 types_or: [python, cython, pyi]
       - repo: https://github.com/psf/black
         rev: 22.3.0
         hooks:
               - id: black
-                files: (python|bench/ann)/.*
+                files: python/.*
                 # Explicitly specify the pyproject.toml at the repo root, not per-project.
                 args: ["--config", "pyproject.toml"]
       - repo: https://github.com/PyCQA/flake8
         rev: 5.0.4
         hooks:
               - id: flake8
                 args: ["--config=.flake8"]
-                files: (python|bench/ann)/.*
+                files: python/.*$
                 types: [file]
                 types_or: [python, cython]
                 additional_dependencies: ["flake8-force"]

diff --git a/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py b/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py
@@ -20,10 +20,6 @@
 
 import yaml
 
-RAFT_HOME = os.getenv("RAFT_HOME")
-if RAFT_HOME is None:
-    RAFT_HOME = os.path.realpath(os.path.join(__file__, "..", "..", ".."))
-
 
 def positive_int(input_str: str) -> int:
     try:
@@ -190,7 +186,6 @@ def main():
         help="dataset whose configuration file will be used",
         default="glove-100-inner",
     )
-
     parser.add_argument(
         "--dataset-path",
         help="path to dataset folder, by default will look in "