add pipeline component sample with primitive output (#2904)

* add pipeline component sample with primitive output * add pipeline component sample with primitive output * add pipeline component sample with primitive output * add pipeline component sample with primitive output * upgrade mldesigner version to fix bug with not support string output * modify notebook * modify notebook * modify notebook * modify notebook * update component * update component * update component
Azure · Dec 25, 2023 · e5f7809 · e5f7809
1 parent 342e602
commit e5f7809
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 6 deletions.
diff --git a/...b_pipeline_with_python_function_components/pipeline_with_python_function_components.ipynb b/...b_pipeline_with_python_function_components/pipeline_with_python_function_components.ipynb
@@ -196,15 +196,18 @@
     "        training_data=input_data, max_epochs=5, learning_rate=learning_rate\n",
     "    )\n",
     "    score_with_sample_data = score_data(\n",
-    "        model_input=train_with_sample_data.outputs.model_output, test_data=test_data\n",
+    "        model_input=train_with_sample_data.outputs.model_output,\n",
+    "        test_data=test_data,\n",
+    "        model_file=train_with_sample_data.outputs.output,\n",
     "    )\n",
     "    # example how to change path of output on step level,\n",
     "    # please note if the output is promoted to pipeline level you need to change path in pipeline job level\n",
     "    score_with_sample_data.outputs.score_output = Output(\n",
     "        type=\"uri_folder\", mode=\"rw_mount\", path=custom_path\n",
     "    )\n",
     "    eval_with_sample_data = eval_model(\n",
-    "        scoring_result=score_with_sample_data.outputs.score_output\n",
+    "        scoring_result=score_with_sample_data.outputs.score_output,\n",
+    "        scoring_file=score_with_sample_data.outputs.output,\n",
     "    )\n",
     "\n",
     "    # Return: pipeline outputs\n",

diff --git a/sdk/python/jobs/pipelines/1b_pipeline_with_python_function_components/src/components.py b/sdk/python/jobs/pipelines/1b_pipeline_with_python_function_components/src/components.py
@@ -12,7 +12,7 @@ def train_model(
     max_epochs: int,
     model_output: Output(type="uri_folder"),
     learning_rate=0.02,
-):
+) -> str:
     """A dummy train component.
 
     Args:
@@ -37,6 +37,8 @@ def train_model(
     model = str(uuid4())
     (Path(model_output) / "model").write_text(model)
 
+    return str((Path(model_output) / "model"))
+
 
 @command_component(
     display_name="Score",
@@ -50,11 +52,13 @@ def score_data(
     model_input: Input(type="uri_folder"),
     test_data: Input(type="uri_file"),
     score_output: Output(type="uri_folder"),
-):
+    model_file: str = None,
+) -> str:
     """A dummy score component."""
 
     lines = [
         f"Model path: {model_input}",
+        f"Model file: {model_file}",
         f"Test data path: {test_data}",
         f"Scoring output path: {score_output}",
     ]
@@ -71,15 +75,20 @@ def score_data(
     # Here only print text to output file as demo
     (Path(score_output) / "score").write_text("scored with {}".format(model))
 
+    return str(Path(score_output) / "score")
+
 
 @command_component(display_name="Evaluate", environment="./env.yaml")
 def eval_model(
-    scoring_result: Input(type="uri_folder"), eval_output: Output(type="uri_folder")
+    scoring_result: Input(type="uri_folder"),
+    eval_output: Output(type="uri_folder"),
+    scoring_file: str = None,
 ):
     """A dummy evaluate component."""
 
     lines = [
         f"Scoring result path: {scoring_result}",
+        f"Scoring file: {scoring_file}",
         f"Evaluation output path: {eval_output}",
     ]
 

diff --git a/sdk/python/jobs/pipelines/1b_pipeline_with_python_function_components/src/env.yaml b/sdk/python/jobs/pipelines/1b_pipeline_with_python_function_components/src/env.yaml
@@ -11,4 +11,4 @@ conda_file:
     - python=3.8.12
     - pip=21.2.2
     - pip:
-      - mldesigner==0.1.0b4
+      - mldesigner==0.1.0b17