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.",