fix bug in envmapper

tapis-project · Jan 5, 2024 · d70af64 · d70af64
1 parent c01dfc5
commit d70af64
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/src/engine/src/core/mappers/EnvMapper.py b/src/engine/src/core/mappers/EnvMapper.py
@@ -6,9 +6,6 @@ def __init__(self, dao: WorkflowExecutorStateDAO):
         self._dao = dao
 
     def get_value_by_key(self, key):
-        print("ENVMAPPER")
-        print("KEY", key)
-        print("KEY ON ENV", self._dao.get_state().ctx.env.get(key))
-        env = self._dao.get_state().ctx.env.get(key).value
+        env = self._dao.get_state().ctx.env.get(key, None)
         if env == None: return None
         return env.value
diff --git a/src/engine/src/core/workflows/WorkflowExecutor.py b/src/engine/src/core/workflows/WorkflowExecutor.py
@@ -627,7 +627,6 @@ def _prepare_pipeline_fs(self):
         env_var_value_file_repo = self.container.load("EnvVarValueFileRepository")
         env_repo = self.container.load("EnvRepository")
         for key in self.state.ctx.env:
-            print("KEY", key)
             env_var_value_file_repo.save(
                 self.state.ctx.pipeline.env_dir + key,
                 env_repo.get_value_by_key(key)