From 374250076d57f23ee3040be03e3869b90419cde8 Mon Sep 17 00:00:00 2001 From: Ning Gao Date: Fri, 12 Jul 2019 11:47:40 -0700 Subject: [PATCH] fix dependency bug in the recursion support --- sdk/python/kfp/compiler/compiler.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/sdk/python/kfp/compiler/compiler.py b/sdk/python/kfp/compiler/compiler.py index 1a239087f6f..92413a4c081 100644 --- a/sdk/python/kfp/compiler/compiler.py +++ b/sdk/python/kfp/compiler/compiler.py @@ -298,13 +298,11 @@ def _get_dependencies(self, pipeline, root_group, op_groups, opsgroups_groups, o # Generate dependencies based on the recursive opsgroups #TODO: refactor the following codes with the above def _get_dependency_opsgroup(group, dependencies): - upstream_op_names = set() + upstream_op_names = set([dependency.name for dependency in group.dependencies]) if group.recursive_ref: for param in group.inputs + list(condition_params[group.name]): if param.op_name: upstream_op_names.add(param.op_name) - else: - upstream_op_names = set([dependency.name for dependency in group.dependencies]) for op_name in upstream_op_names: if op_name in pipeline.ops: