File tree Expand file tree Collapse file tree 2 files changed +11
-0
lines changed Expand file tree Collapse file tree 2 files changed +11
-0
lines changed Original file line number Diff line number Diff 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 :
Original file line number Diff line number Diff 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>" ,
You can’t perform that action at this time.
0 commit comments