feat(component): use resource name for input artifact (#6698)

kubeflow · Oct 7, 2021 · 89e4c1c · 89e4c1c
1 parent 2b2e5fd
commit 89e4c1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/components/google-cloud/google_cloud_pipeline_components/aiplatform/utils.py b/components/google-cloud/google_cloud_pipeline_components/aiplatform/utils.py
@@ -605,7 +605,7 @@ def component_yaml_generator(**kwargs):
                 )
                 input_args.append(f'--{prefix_key}.{component_param_name}')
                 if is_mb_sdk_resource_noun_type(param_type):
-                    input_args.append(structures.InputUriPlaceholder(input_name=key))
+                    input_args.append(f'{{{{$.inputs.artifacts[\'{key}\'].metadata[\'resourceName\']}}}}')
                 else:
                     input_args.append(structures.InputValuePlaceholder(input_name=key))