fix(backend): Fix broken dependantTasks in ParallelFor

Signed-off-by: Giulio Frasca <[email protected]>
kubeflow · Oct 3, 2024 · 02e27f0 · 02e27f0
1 parent 0ea9187
commit 02e27f0
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/backend/src/v2/compiler/argocompiler/dag.go b/backend/src/v2/compiler/argocompiler/dag.go
@@ -274,9 +274,10 @@ func (c *workflowCompiler) iteratorTask(name string, task *pipelinespec.Pipeline
 	// Set up Loop Control Template
 	loopDriverArgoName := name + "-loop-driver"
 	loopDriverInputs := dagDriverInputs{
-		component:   componentSpecPlaceholder,
-		parentDagID: parentDagID,
-		task:        taskJson, // TODO(Bobgy): avoid duplicating task JSON twice in the template.
+		component:      componentSpecPlaceholder,
+		parentDagID:    parentDagID,
+		task:           taskJson, // TODO(Bobgy): avoid duplicating task JSON twice in the template.
+		iterationIndex: "0",
 	}
 	loopDriver, loopDriverOutputs, err := c.dagDriverTask(loopDriverArgoName, loopDriverInputs)
 	if err != nil {
@@ -356,7 +357,7 @@ func (c *workflowCompiler) iterationItemTask(name string, task *pipelinespec.Pip
 	if err != nil {
 		return nil, err
 	}
-	driver.Depends = depends(task.GetDependentTasks())
+	//driver.Depends = depends(task.GetDependentTasks())  # TODO(gfrasca): Handled already by root task
 
 	iterationCount := intstr.FromString(driverOutputs.iterationCount)
 	iterationTasks, err := c.task(