handle edge case

Signed-off-by: zazulam <[email protected]> Co-authored-by: droctothorpe <[email protected]>
kubeflow · Sep 17, 2024 · 1cb4db8 · 1cb4db8
1 parent a0a7b7b
commit 1cb4db8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/backend/src/v2/driver/driver.go b/backend/src/v2/driver/driver.go
@@ -1267,6 +1267,7 @@ func resolveInputs(ctx context.Context, dag *metadata.DAG, iterationIndex *int,
 				} else {
 					// The producer subtask is not a DAG, so we exit the loop.
 					producerSubTaskMaybeDAG = false
+					inputs.ParameterValues[name] = producerOutputs[taskOutput.GetOutputParameterKey()]
 				}
 			}