From 1af6427ea6a7c6bcccbd10a47881b7f01c95ef88 Mon Sep 17 00:00:00 2001 From: Lizhen You Date: Sat, 28 Dec 2024 03:29:34 -0800 Subject: [PATCH] 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 --- python/cuvs_bench/cuvs_bench/run/run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cuvs_bench/cuvs_bench/run/run.py b/python/cuvs_bench/cuvs_bench/run/run.py index 0159d2c19..b7813f1c6 100644 --- 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: