From a72f17f80fea8da60abc444d8de7af2f5c29216e Mon Sep 17 00:00:00 2001 From: David Farrington Date: Thu, 19 Dec 2024 18:40:00 +0000 Subject: [PATCH] Fix issue where tasks preceeding parallel for loops that recieve pipeline parameters, are wrongly expected to have task attributes, Signed-off-by: David Farrington --- sdk/python/kfp/compiler/compiler_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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