fxing test path

C2SM-RCM · Nov 15, 2023 · 899250c · 899250c
1 parent de57ba5
commit 899250c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/emiproc/tests_utils/__init__.py b/emiproc/tests_utils/__init__.py
@@ -2,14 +2,16 @@
 import emiproc
 from pathlib import Path
 
+TEST_DIR = Path(*emiproc.__path__) / ".." / "tests"
+TEST_DIR.mkdir(exist_ok=True)
 
-WEIGHTS_DIR = Path(*emiproc.__path__) / ".." / "tests" / ".weights"
+WEIGHTS_DIR = TEST_DIR / ".weights"
 WEIGHTS_DIR.mkdir(exist_ok=True)
 
-TEST_OUTPUTS_DIR = Path(*emiproc.__path__) / ".." / "tests" / ".outputs"
+TEST_OUTPUTS_DIR = TEST_DIR / ".outputs"
 TEST_OUTPUTS_DIR.mkdir(exist_ok=True)
 
 
 if __name__ == "__main__":
     print(WEIGHTS_DIR)
-    print(WEIGHTS_DIR.is_dir())
+    print(WEIGHTS_DIR.is_dir())