From 43810269958abde6f1aacb137a8c51a06d2a8506 Mon Sep 17 00:00:00 2001 From: Lu Peng Date: Mon, 22 Sep 2025 18:03:35 -0400 Subject: [PATCH] added blocker for updating multi deployment --- ads/aqua/modeldeployment/deployment.py | 7 +++++++ tests/unitary/with_extras/aqua/test_deployment.py | 4 ++++ 2 files changed, 11 insertions(+) diff --git a/ads/aqua/modeldeployment/deployment.py b/ads/aqua/modeldeployment/deployment.py index 08e2f1c18..fa7eebdf0 100644 --- a/ads/aqua/modeldeployment/deployment.py +++ b/ads/aqua/modeldeployment/deployment.py @@ -1412,6 +1412,13 @@ def _update_model_group( The instance of DataScienceModelGroup. """ model_group = DataScienceModelGroup.from_id(model_group_id) + if ( + model_group.dsc_model_group.model_group_details.type + != DeploymentType.STACKED + ): + raise AquaValueError( + "Invalid 'model_deployment_id'. Only stacked deployment is supported to update." + ) # create a new model group if fine tune weights changed as member models in ds model group is inmutable if update_model_deployment_details.models: if len(update_model_deployment_details.models) != 1: diff --git a/tests/unitary/with_extras/aqua/test_deployment.py b/tests/unitary/with_extras/aqua/test_deployment.py index 65a0ffa8b..c6f49d6ae 100644 --- a/tests/unitary/with_extras/aqua/test_deployment.py +++ b/tests/unitary/with_extras/aqua/test_deployment.py @@ -2249,6 +2249,10 @@ def test_update_model_group_deployment( aqua_model_group = DataScienceModelGroup( spec=TestDataset.aqua_deployment_stack_model ) + model_group_details = MagicMock(type="STACKED") + aqua_model_group.dsc_model_group = MagicMock( + model_group_details=model_group_details + ) aqua_model_group.set_spec( aqua_model_group.CONST_BASE_MODEL_ID, "ocid1.datasciencemodel.oc1.iad.",