Skip to content

Commit 71eb54c

Browse files
committed
handling Model Deployment list for model group deployment type
1 parent 845ec6d commit 71eb54c

File tree

1 file changed

+21
-5
lines changed

1 file changed

+21
-5
lines changed

ads/aqua/modeldeployment/deployment.py

Lines changed: 21 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -878,12 +878,28 @@ def list(self, **kwargs) -> List["AquaDeployment"]:
878878
)
879879

880880
if oci_aqua:
881-
results.append(
882-
AquaDeployment.from_oci_model_deployment(
883-
model_deployment, self.region
884-
)
881+
# skipping the AQUA model deployments that are created from model group
882+
# TODO: remove this checker after AQUA deployment is integrated with model group
883+
aqua_model_id = model_deployment.freeform_tags.get(
884+
Tags.AQUA_MODEL_ID_TAG, UNKNOWN
885885
)
886-
886+
if (
887+
"datasciencemodelgroup" in aqua_model_id
888+
or model_deployment.model_deployment_configuration_details.deployment_type
889+
in ["UNKNOWN_ENUM_VALUE", "MODEL_GROUP", "SINGLE_MODEL_FLEX"]
890+
):
891+
continue
892+
try:
893+
results.append(
894+
AquaDeployment.from_oci_model_deployment(
895+
model_deployment, self.region
896+
)
897+
)
898+
except Exception as e:
899+
logger.error(
900+
f"There was an issue processing the list of model deployments . Error: {str(e)}",
901+
exc_info=True,
902+
)
887903
# log telemetry if MD is in active or failed state
888904
deployment_id = model_deployment.id
889905
state = model_deployment.lifecycle_state.upper()

0 commit comments

Comments
 (0)