Skip to content

Commit af52a23

Browse files
model integration with DT
1 parent c349397 commit af52a23

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

src/machinelearningservices/azext_mlv2/manual/_params/_model_params.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,11 @@ def load_model_params(self):
6161
"The default type is custom_model.",
6262
)
6363
add_description_param(c, help_message="Description of the model.")
64+
c.argument(
65+
"default_deployment_template",
66+
options_list=["--default-deployment-template"],
67+
help="Default deployment template for the model.",
68+
)
6469

6570
with self.argument_context("ml model show") as c:
6671
add_common_params(c)

src/machinelearningservices/azext_mlv2/manual/custom/model.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ def ml_model_create(
3535
registry_name=None,
3636
params_override=None,
3737
datastore=None,
38+
default_deployment_template=None,
3839
no_wait=None, # pylint: disable=unused-argument
3940
):
4041
ml_client, debug = get_ml_client(
@@ -81,6 +82,7 @@ def ml_model_create(
8182
stage=stage,
8283
type=type,
8384
datastore=datastore,
85+
default_deployment_template=default_deployment_template,
8486
)
8587
if model and model.type:
8688
model_info.model_type = model.type

0 commit comments

Comments
 (0)