add current_project_folder env variable to __init_

datajoint · Oct 2, 2023 · 7980338 · 7980338
1 parent 7ce6007
commit 7980338
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/element_deeplabcut/__init__.py b/element_deeplabcut/__init__.py
@@ -14,4 +14,7 @@
     "DLC_ROOT_DATA_DIR", dj.config["custom"].get("dlc_root_data_dir", "")
 )
 
+dj.config["custom"]["current_project_folder"] = os.getenv(
+    "CURRENT_PROJECT_FOLDER", dj.config["custom"].get("current_project_folder", "")
+)
 db_prefix = dj.config["custom"].get("database.prefix", "")