Fixed __has_dependent_tasks

Signed-off-by: hbelmiro <[email protected]>
kubeflow · Jul 18, 2024 · 2ede4cb · 2ede4cb
1 parent ba14f74
commit 2ede4cb
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/sdk/python/kfp/dsl/tasks_group.py b/sdk/python/kfp/dsl/tasks_group.py
@@ -150,9 +150,12 @@ def __has_dependent_tasks(self) -> bool:
         if not self.exit_task.inputs:
             return False
 
-        return any(
-            isinstance(task_input, PipelineParameterChannel)
-            for task_input in self.exit_task.inputs.values())
+        for task_input in self.exit_task.inputs.values():
+            if isinstance(
+                    task_input,
+                    PipelineParameterChannel) and task_input.task is not None:
+                return True
+        return False
 
 
 class ConditionBranches(TasksGroup):