Skip to content

Commit d379da7

Browse files
authored
[mldesigner] Fix mldesigner version in sample (Azure#1797)
* update * update * update * update * update * update
1 parent 795bc34 commit d379da7

File tree

2 files changed

+11
-12
lines changed

2 files changed

+11
-12
lines changed

sdk/python/jobs/pipelines/1b_pipeline_with_python_function_components/src/components.py

+10-10
Original file line numberDiff line numberDiff line change
@@ -6,16 +6,7 @@
66
from mldesigner import command_component, Input, Output
77

88

9-
# init customer environment with conda YAML
10-
# the YAML file shall be put under your code folder.
11-
conda_env = dict(
12-
# note that mldesigner package must be included.
13-
conda_file=Path(__file__).parent / "conda.yaml",
14-
image="mcr.microsoft.com/azureml/openmpi3.1.2-ubuntu18.04",
15-
)
16-
17-
18-
@command_component(environment=conda_env)
9+
@command_component()
1910
def train_model(
2011
training_data: Input(type="uri_file"),
2112
max_epochs: int,
@@ -47,6 +38,15 @@ def train_model(
4738
(Path(model_output) / "model").write_text(model)
4839

4940

41+
# init customer environment with conda YAML
42+
# the YAML file shall be put under your code folder.
43+
conda_env = dict(
44+
# note that mldesigner package must be included.
45+
conda_file=Path(__file__).parent / "conda.yaml",
46+
image="mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu20.04",
47+
)
48+
49+
5050
@command_component(
5151
display_name="Score",
5252
environment=conda_env,

sdk/python/jobs/pipelines/1b_pipeline_with_python_function_components/src/conda.yaml

+1-2
Original file line numberDiff line numberDiff line change
@@ -8,5 +8,4 @@ dependencies:
88
- python=3.8.12
99
- pip=21.2.2
1010
- pip:
11-
- mldesigner==0.1.0b4
12-
- typing-extensions==4.3.0
11+
- mldesigner==0.1.0b4

0 commit comments

Comments
 (0)