add MODULE_DIR

naik-aakash · Jan 9, 2025 · b918db8 · b918db8
1 parent 58e6b2a
commit b918db8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/autoplex/fitting/common/jobs.py b/src/autoplex/fitting/common/jobs.py
@@ -1,5 +1,6 @@
 """General fitting jobs using several MLIPs available."""
 
+import os
 from pathlib import Path
 
 from jobflow import job
@@ -13,8 +14,8 @@
     nequip_fitting,
 )
 
-current_dir = Path(__file__).absolute().parent
-GAP_DEFAULTS_FILE_PATH = current_dir / "mlip-phonon-defaults.json"
+MODULE_DIR = Path(os.path.dirname(__file__))
+GAP_DEFAULTS_FILE_PATH = MODULE_DIR / "mlip-phonon-defaults.json"
 
 
 @job