From 656b9d4f049afd894dc4d6f8a8a7d8ea51be7953 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 --- 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 dc10665944f7..59567a5851e1 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