Skip to content

Commit

Permalink
Fix issue where tasks preceeding parallel for loops that recieve pipe…
Browse files Browse the repository at this point in the history
…line parameters, are wrongly expected to have task attributes, Signed-off-by: David Farrington <[email protected]>
  • Loading branch information
farridav committed Dec 19, 2024
1 parent 399a597 commit a72f17f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sdk/python/kfp/compiler/compiler_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit a72f17f

Please sign in to comment.