Skip to content

Commit 5cfa052

Browse files
authored
Add blocker for multi deployment (#1275)
2 parents 6829706 + ef0096c commit 5cfa052

File tree

2 files changed

+11
-0
lines changed

2 files changed

+11
-0
lines changed

ads/aqua/modeldeployment/deployment.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1412,6 +1412,13 @@ def _update_model_group(
14121412
The instance of DataScienceModelGroup.
14131413
"""
14141414
model_group = DataScienceModelGroup.from_id(model_group_id)
1415+
if (
1416+
model_group.dsc_model_group.model_group_details.type
1417+
!= DeploymentType.STACKED
1418+
):
1419+
raise AquaValueError(
1420+
"Invalid 'model_deployment_id'. Only stacked deployment is supported to update."
1421+
)
14151422
# create a new model group if fine tune weights changed as member models in ds model group is inmutable
14161423
if update_model_deployment_details.models:
14171424
if len(update_model_deployment_details.models) != 1:

tests/unitary/with_extras/aqua/test_deployment.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2249,6 +2249,10 @@ def test_update_model_group_deployment(
22492249
aqua_model_group = DataScienceModelGroup(
22502250
spec=TestDataset.aqua_deployment_stack_model
22512251
)
2252+
model_group_details = MagicMock(type="STACKED")
2253+
aqua_model_group.dsc_model_group = MagicMock(
2254+
model_group_details=model_group_details
2255+
)
22522256
aqua_model_group.set_spec(
22532257
aqua_model_group.CONST_BASE_MODEL_ID,
22542258
"ocid1.datasciencemodel.oc1.iad.<OCID>",

0 commit comments

Comments
 (0)