test: set path

vra · Jul 8, 2023 · e5e0806 · e5e0806
1 parent 846fbab
commit e5e0806
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_image_retrieval.py b/tests/test_image_retrieval.py
@@ -1,11 +1,13 @@
 from collections import namedtuple
+import os
 from pathlib import Path
 import sys
 
 import pytest
 
-sys.path.insert(0, "..")
-from src.dinov2_retrieval.image_retriever import ImageRetriver
+src_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "src")
+sys.path.insert(0, src_dir)
+from dinov2_retrieval.image_retriever import ImageRetriver
 
 
 def test_init():