try utils.DATA_PATH in verify_params

gbrammer · Oct 25, 2024 · 292302a · 292302a
1 parent c2ce7db
commit 292302a
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/eazy/param.py b/eazy/param.py
@@ -2,6 +2,8 @@
 import collections
 import numpy as np
 
+from . import utils
+
 __all__ = ["EazyParam", "TranslateFile", "read_param_file"]
 
 def read_param_file(param_file=None, verbose=True):
@@ -157,8 +159,14 @@ def verify_params(self):
         for k in ['TEMPLATES_FILE', 'TEMP_ERR_FILE', 'CATALOG_FILE', 
                   'FILTERS_RES']:
             if isinstance(self[k], str):
-                if not os.path.exists(self[k]):
-                    raise FileNotFoundError(f'{k} ({self[k]}) not found')        
+                file_path = self[k]
+                if not os.path.exists(file_path):
+                    file_path = os.path.join(utils.DATA_PATH, file_path)
+
+                if not os.path.exists(file_path):
+                    raise FileNotFoundError(
+                        f'{k} ({self[k]}) not found in ./ or {utils.DATA_PATH}'
+                    )
 
         assert(int(self['ARRAY_NBITS']) in [32,64])