fix env_vars init

molmod · Jul 23, 2024 · fa6e8d8 · fa6e8d8
1 parent 7e1e8d9
commit fa6e8d8
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/psiflow/execution.py b/psiflow/execution.py
@@ -186,9 +186,10 @@ def __init__(
             'PYTHONUNBUFFERED': 'TRUE',
         }
         if env_vars is None:
-            self.env_vars = dict(default_env_vars)
+            env_vars = dict(default_env_vars)
         else:
-            self.env_vars.update(default_env_vars)
+            env_vars.update(default_env_vars)
+        self.env_vars = env_vars
 
     def server_command(self):
         command_list = ["psiflow-server"]
@@ -265,9 +266,10 @@ def __init__(
             'PYTHONUNBUFFERED': 'TRUE',
         }
         if env_vars is None:
-            self.env_vars = dict(default_env_vars)
+            env_vars = dict(default_env_vars)
         else:
-            self.env_vars.update(default_env_vars)
+            env_vars.update(default_env_vars)
+        self.env_vars = env_vars
 
     def train_command(self, initialize: bool = False):
         # script = "$(python -c 'import psiflow.models.mace_utils; print(psiflow.models.mace_utils.__file__)')"