File tree 2 files changed +11
-12
lines changed
sdk/python/jobs/pipelines/1b_pipeline_with_python_function_components/src
2 files changed +11
-12
lines changed Original file line number Diff line number Diff line change 6
6
from mldesigner import command_component , Input , Output
7
7
8
8
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 ()
19
10
def train_model (
20
11
training_data : Input (type = "uri_file" ),
21
12
max_epochs : int ,
@@ -47,6 +38,15 @@ def train_model(
47
38
(Path (model_output ) / "model" ).write_text (model )
48
39
49
40
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
+
50
50
@command_component (
51
51
display_name = "Score" ,
52
52
environment = conda_env ,
Original file line number Diff line number Diff line change @@ -8,5 +8,4 @@ dependencies:
8
8
- python=3.8.12
9
9
- pip=21.2.2
10
10
- pip :
11
- - mldesigner==0.1.0b4
12
- - typing-extensions==4.3.0
11
+ - mldesigner==0.1.0b4
You can’t perform that action at this time.
0 commit comments