wip: inject step env into step execution env

equinor · Nov 18, 2024 · 11961ab · 11961ab
1 parent c166243
commit 11961ab
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/_ert/forward_model_runner/forward_model_step.py b/src/_ert/forward_model_runner/forward_model_step.py
@@ -145,9 +145,14 @@ def _open_file_handles(
         return (stdin, stdout, stderr)
 
     def _create_environment(self) -> Optional[Dict[str, str]]:
-        combined_environment = None
+        combined_environment: Optional[Dict[str, str]] = None
         if environment := self.job_data.get("environment"):
             combined_environment = {**os.environ, **environment}
+        if step_environment := self.job_data.get("env_pr_fm_step"):
+            if combined_environment is None:
+                combined_environment = {**os.environ, **step_environment}
+            else:
+                combined_environment = {**combined_environment, **step_environment}
         return combined_environment
 
     def _run(self) -> Generator[Start | Exited | Running | None]: