Only load algos configuration files with .yaml suffix

Fixing the following warning during benchmark run ./cuvs_benchmarks/lib/python3.12/site-packages/cuvs_bench/run/run.py:186: UserWarning: Could not load YAML config ./lib/python3.12/site-packages/cuvs_bench/run/../config/algos/__pycache__ due to [Errno 21] Is a directory: ' warnings.warn(f"Could not load YAML config {algo_f} due to {e}") Signed-off-by: Lizhen You <[email protected]>
rapidsai · Dec 28, 2024 · 1af6427 · 1af6427
1 parent b3ce774
commit 1af6427
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cuvs_bench/cuvs_bench/run/run.py b/python/cuvs_bench/cuvs_bench/run/run.py
@@ -140,7 +140,7 @@ def gather_algorithm_configs(
     algos_conf_fs = [
         os.path.join(scripts_path, "../config", "algos", f)
         for f in algos_conf_fs
-        if ".json" not in f and "constraint" not in f and ".py" not in f
+        if f.endswith(".yaml")
     ]
 
     if configuration: