diff --git a/sdk/python/kfp/compiler/compiler_utils.py b/sdk/python/kfp/compiler/compiler_utils.py index dc10665944f..59567a5851e 100644 --- a/sdk/python/kfp/compiler/compiler_utils.py +++ b/sdk/python/kfp/compiler/compiler_utils.py @@ -762,7 +762,7 @@ def get_dependencies( # then make this validation dsl.Collected-aware elif isinstance(upstream_parent_group, tasks_group.ParallelFor): upstream_tasks_that_downstream_consumers_from = [ - channel.task.name for channel in task._channel_inputs + channel.task.name for channel in task._channel_inputs if channel.task ] has_data_exchange = upstream_task.name in upstream_tasks_that_downstream_consumers_from # don't raise for .after