Merge pull request graphnet-team#370 from MortenHolmRep/root_dir

Implemented root dir variable for graphnet
RasmusOrsoe · Dec 8, 2022 · da612e4 · da612e4
2 parents 8d27c35 + dc64713
commit da612e4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/graphnet/constants.py b/src/graphnet/constants.py
@@ -2,6 +2,8 @@
 
 import os.path
 
-TEST_DATA_DIR = os.path.abspath(
-    os.path.join(os.path.dirname(__file__), "..", "..", "test_data")
+GRAPHNET_ROOT_DIR = os.path.abspath(
+    os.path.join(os.path.dirname(__file__), "..", "..")
 )
+
+TEST_DATA_DIR = os.path.abspath(os.path.join(GRAPHNET_ROOT_DIR, "test_data"))