Skip to content

Commit 38caa1b

Browse files
model integration with DT
1 parent c349397 commit 38caa1b

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-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: 8 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
@@ -267,3 +269,9 @@ def ml_model_share(cmd, name, version, share_with_name, share_with_version,
267269
share_with_name=share_with_name,
268270
share_with_version=share_with_version)
269271
return _dump_entity_with_warnings(model)
272+
273+
c.argument(
274+
"default_deployment_template",
275+
options_list=["--default-deployment-template"],
276+
help="Default deployment template for the model.",
277+
)

0 commit comments

Comments
 (0)