fixed system variables

iprafols · Oct 3, 2023 · 1d83f81 · 1d83f81
1 parent 0598ede
commit 1d83f81
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/py/squeze/config.py b/py/squeze/config.py
@@ -20,6 +20,12 @@
 QSO_COLS = ("ra dec thing_id plate mjd fiberid z_vi class_person z_conf_person "
             "boss_target1 ancillary_target1 ancillary_target2 eboss_target0")
 
+# add system variables required to load the default configuration
+THIS_DIR = os.path.dirname(os.path.abspath(__file__))
+SQUEZE = THIS_DIR.split("py/squeze")[0]
+os.environ["SQUEZE"] = SQUEZE
+
+
 default_config = {
     "general": {
         "mode": "training",