[Python] How to define Output in AML pipelien?

Here is my AML pipeline using python SDK v2.

input_path = Input(type="uri_folder", path="azureml://datastores/test/paths/input")
output_path = Output(type="uri_folder", path="azureml://datastores/test/paths/output")


@command_component(
name="test_com",
version="1",
display_name="test com",
description="test",
environment=custom_env,
)
def prepare_data_component(
input_data: Input(type="uri_folder"),
output_data: Output(type="uri_folder"),
):
print("input_data: ", input_data)
print("output_data: ", output_data)


@pipeline(
default_compute=cpu_compute_target,
)
def pipeline_test(pipeline_input_data):
prepare_data_node = prepare_data_component(
input_data=pipeline_input_data,
output_data=output_data,
)

return {
"output_data": prepare_data_node.outputs.output_data,
}

# create a pipeline
pipeline_job = pipeline_test(
pipeline_input_data=input_data,
)

print(pipeline_job)


But it shows me this error:

[component] test com() got an unexpected keyword argument 'output_data', valid keywords: 'input_data'.


Can output be passed as a parameter to component? Or can you give me some references on how to use output?

I don't want the official documentation because it is missing a lot of content, such as "How to use output in python sdk"

Continue reading...