From 32454486949575984e95087a95028237481aebed Mon Sep 17 00:00:00 2001 From: Nathan Freeman Date: Thu, 26 Oct 2023 14:16:17 -0500 Subject: [PATCH] fix task_output fetching --- src/engine/src/utils/k8s.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/engine/src/utils/k8s.py b/src/engine/src/utils/k8s.py index 7bc2e0d3..449bc081 100644 --- a/src/engine/src/utils/k8s.py +++ b/src/engine/src/utils/k8s.py @@ -30,7 +30,6 @@ def input_to_k8s_env_vars(_inputs, pipeline_work_dir, env={}, args={}, prefix="" for input_id, _input in _inputs.items(): # Use input[input_id].value if provided value = _input.value - print(input_id, _input, "\n\n") if value != None: k8senvvars.append( V1EnvVar( @@ -61,10 +60,10 @@ def input_to_k8s_env_vars(_inputs, pipeline_work_dir, env={}, args={}, prefix="" key = value_from.get("args") k8senvvar_value = get_value_from_args(args, key) k8senvvar_value_source_key = key - elif hasattr(value_from, "task_output"): + elif value_from.get("task_output", None) != None: k8senvvar_value_source = "task_output" - task_id = value_from.task_output.task_id - output_id = value_from.task_output.output_id + task_id = value_from.get("task_output").task_id + output_id = value_from.get("task_output").output_id # Get the value from the output of a previous task previous_task_output_path = os.path.join(