From ffaf1e21b3ee30db063f078c90e49d701a2d45fe Mon Sep 17 00:00:00 2001 From: Alexey Volkov Date: Fri, 15 May 2020 01:02:58 -0700 Subject: [PATCH] SDK - Moved some data from the component_ref annotation to the component_spec annotation (#3751) Removing the component spec from component_ref (since it would be a duplicate), but making sure the whole spec if available in component_spec. --- sdk/python/kfp/dsl/_component_bridge.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/sdk/python/kfp/dsl/_component_bridge.py b/sdk/python/kfp/dsl/_component_bridge.py index a4de562536a..055a5de154d 100644 --- a/sdk/python/kfp/dsl/_component_bridge.py +++ b/sdk/python/kfp/dsl/_component_bridge.py @@ -59,9 +59,10 @@ def _create_container_op_from_component_and_arguments( ) component_meta = copy.copy(component_spec) - component_meta.implementation = None task._set_metadata(component_meta) - task._component_ref = component_ref + component_ref_without_spec = copy.copy(component_ref) + component_ref_without_spec.spec = None + task._component_ref = component_ref_without_spec # Previously, ContainerOp had strict requirements for the output names, so we had to # convert all the names before passing them to the ContainerOp constructor.