Skip to content

Commit

Permalink
SDK - Components - Fixed the output types for outputs with converted …
Browse files Browse the repository at this point in the history
…names

Fixes #2130
  • Loading branch information
Ark-kun committed Sep 18, 2019
1 parent 0971006 commit 3dea33d
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions sdk/python/kfp/components/_dsl_bridge.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,6 +153,12 @@ def _create_container_op_from_resolved_task(name:str, container_image:str, comma
file_outputs=output_paths_for_container_op,
artifact_argument_paths=[dsl.InputArgumentPath(argument=artifact_arguments[input_name], input=input_name, path=path) for input_name, path in input_paths.items()],
)
# Fixing ContainerOp output types
if component_spec.outputs:
for output in component_spec.outputs:
pythonic_name = output_name_to_kubernetes[output.name]
if pythonic_name in task.outputs:
task.outputs[task.outputs].param_type = output.type

component_meta = copy.copy(component_spec)
component_meta.implementation = None
Expand Down

0 comments on commit 3dea33d

Please sign in to comment.