Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

SDK - Added support for raw input artifact argument values to ContainerOp #791

Prev Previous commit
Next Next commit
Renamed input_artifact_arguments to artifact_arguments to reduce conf…
…usion
  • Loading branch information
Ark-kun committed Aug 28, 2019
commit 6bd95bd48757fdc2121294efa08e5a46685c7c44
2 changes: 1 addition & 1 deletion sdk/python/kfp/compiler/compiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -516,7 +516,7 @@ def _group_to_dag_template(self, group, inputs, outputs, dependencies):
arguments.sort(key=lambda x: x['name'])
task['arguments'] = {'parameters': arguments}

artifact_arguments = getattr(sub_group, 'input_artifact_arguments', None)
artifact_arguments = getattr(sub_group, 'artifact_arguments', None)
if artifact_arguments:
artifact_argument_structs = []
for input_name, argument in artifact_arguments.items():
Expand Down
2 changes: 1 addition & 1 deletion sdk/python/kfp/dsl/_container_op.py
Original file line number Diff line number Diff line change
Expand Up @@ -1077,7 +1077,7 @@ def _decorated(*args, **kwargs):

# attributes specific to `ContainerOp`
self.input_artifact_paths = input_artifact_paths
self.input_artifact_arguments = artifact_arguments
self.artifact_arguments = artifact_arguments
self.file_outputs = file_outputs
self.output_artifact_paths = output_artifact_paths or {}
self.artifact_location = artifact_location
Expand Down