use normpath

rapidsai · Oct 18, 2023 · d9a62e9 · d9a62e9
1 parent cf8eced
commit d9a62e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/raft-ann-bench/src/raft-ann-bench/split_groundtruth/__main__.py b/python/raft-ann-bench/src/raft-ann-bench/split_groundtruth/__main__.py
@@ -26,7 +26,7 @@ def split_groundtruth(groundtruth_filepath):
     path_to_groundtruth = os.path.normpath(groundtruth_filepath).split(os.sep)
     if len(path_to_groundtruth) > 1:
         os.chdir(os.path.join(*path_to_groundtruth[:-1]))
-    groundtruth_filename = groundtruth_filepath.split("/")[-1]
+    groundtruth_filename = path_to_groundtruth[-1]
     subprocess.run(
         [ann_bench_scripts_path, groundtruth_filename, "groundtruth"],
         check=True,