append PYTHONPATH for not installed modules

dcfidalgo · Nov 14, 2023 · 4e83a88 · 4e83a88
1 parent 8ab1746
commit 4e83a88
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/slurm_sweeps/experiment.py b/src/slurm_sweeps/experiment.py
@@ -85,6 +85,11 @@ def __init__(
         os.environ[EXPERIMENT_NAME] = name
         os.environ[STORAGE_PATH] = str(self._storage.path)
         os.environ[DB_PATH] = str(self._database.path)
+        # in case we import modules that are not installed
+        try:
+            os.environ["PYTHONPATH"] = f"{os.environ['PYTHONPATH']}:{os.getcwd()}"
+        except KeyError:
+            os.environ["PYTHONPATH"] = os.getcwd()
 
     @staticmethod
     def _create_experiment_dir(