Skip to content

Commit 197cca3

Browse files
committed
updated pr.
1 parent 7248881 commit 197cca3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ads/aqua/modeldeployment/deployment.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1294,8 +1294,6 @@ def update(
12941294
f"Invalid parameters for updating a model group deployment. Error details: {custom_errors}."
12951295
) from ex
12961296

1297-
self._validate_input_models(update_model_deployment_details)
1298-
12991297
model_deployment = ModelDeployment.from_id(model_deployment_id)
13001298

13011299
infrastructure = model_deployment.infrastructure
@@ -1421,6 +1419,8 @@ def _update_model_group(
14211419
raise AquaValueError(
14221420
"Invalid 'models' provided. Only one base model is required for updating model stack deployment."
14231421
)
1422+
# validates input base and fine tune models
1423+
self._validate_input_models(update_model_deployment_details)
14241424
target_stacked_model = update_model_deployment_details.models[0]
14251425
target_base_model_id = target_stacked_model.model_id
14261426
if model_group.base_model_id != target_base_model_id:

0 commit comments

Comments
 (0)