fix path

rapidsai · Oct 18, 2023 · 07385df · 07385df
1 parent f7835fa
commit 07385df
Showing 1 changed file with 3 additions 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
@@ -19,7 +19,9 @@
 
 
 def split_groundtruth(groundtruth_filepath):
-    ann_bench_scripts_path = "split_groundtruth.pl"
+    ann_bench_scripts_path = \
+        os.path.join(os.path.dirname(os.path.realpath(__file__)),
+                     "split_groundtruth.pl")
     pwd = os.getcwd()
     os.chdir("/".join(groundtruth_filepath.split("/")[:-1]))
     groundtruth_filename = groundtruth_filepath.split("/")[-1]