Update executable directory to look in current directory first, then …

…package directory
MintonGroup · Nov 15, 2023 · 3fa3464 · 3fa3464
1 parent e5431c4
commit 3fa3464
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ctem/ctem/driver.py b/ctem/ctem/driver.py
@@ -52,10 +52,10 @@ def __init__(self, param_file="ctem.in", isnew=True):
         }
 
         # Get the location of the CTEM executable
-        self.ctem_executable = Path(_pyfile).parent.parent.parent / "bin" / "CTEM"
+        self.ctem_executable = Path(currentdir) / "CTEM"
         if not self.ctem_executable.exists():
-            print(f"CTEM driver not found at {self.ctem_executable}. Trying current directory.")
-            self.ctem_executable = Path(currentdir) / "CTEM"
+            print(f"CTEM driver not found at {self.ctem_executable}. Trying package directory..")
+            self.ctem_executable = Path(_pyfile).parent.parent.parent / "bin" / "CTEM"
             if not self.ctem_executable.exists():
                 warnings.warn(f"Cannot find the CTEM driver {str(self.ctem_executable)}", stacklevel=2)
                 self.ctem_executable = None