Skip to content

Commit 2f4e858

Browse files
Bug Fix for MD failure for deployment of Aqua models registered from OSS (#1016)
2 parents affeae4 + 100a88f commit 2f4e858

File tree

1 file changed

+15
-9
lines changed

1 file changed

+15
-9
lines changed

ads/aqua/modeldeployment/deployment.py

Lines changed: 15 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,7 @@ def create(
185185
tags[tag] = aqua_model.freeform_tags[tag]
186186

187187
tags.update({Tags.AQUA_MODEL_NAME_TAG: aqua_model.display_name})
188-
tags.update({Tags.TASK: aqua_model.freeform_tags.get(Tags.TASK, None)})
188+
tags.update({Tags.TASK: aqua_model.freeform_tags.get(Tags.TASK, UNKNOWN)})
189189

190190
# Set up info to get deployment config
191191
config_source_id = model_id
@@ -533,16 +533,22 @@ def list(self, **kwargs) -> List["AquaDeployment"]:
533533
return results
534534

535535
@telemetry(entry_point="plugin=deployment&action=delete", name="aqua")
536-
def delete(self,model_deployment_id:str):
537-
return self.ds_client.delete_model_deployment(model_deployment_id=model_deployment_id).data
536+
def delete(self, model_deployment_id: str):
537+
return self.ds_client.delete_model_deployment(
538+
model_deployment_id=model_deployment_id
539+
).data
538540

539-
@telemetry(entry_point="plugin=deployment&action=deactivate",name="aqua")
540-
def deactivate(self,model_deployment_id:str):
541-
return self.ds_client.deactivate_model_deployment(model_deployment_id=model_deployment_id).data
541+
@telemetry(entry_point="plugin=deployment&action=deactivate", name="aqua")
542+
def deactivate(self, model_deployment_id: str):
543+
return self.ds_client.deactivate_model_deployment(
544+
model_deployment_id=model_deployment_id
545+
).data
542546

543-
@telemetry(entry_point="plugin=deployment&action=activate",name="aqua")
544-
def activate(self,model_deployment_id:str):
545-
return self.ds_client.activate_model_deployment(model_deployment_id=model_deployment_id).data
547+
@telemetry(entry_point="plugin=deployment&action=activate", name="aqua")
548+
def activate(self, model_deployment_id: str):
549+
return self.ds_client.activate_model_deployment(
550+
model_deployment_id=model_deployment_id
551+
).data
546552

547553
@telemetry(entry_point="plugin=deployment&action=get", name="aqua")
548554
def get(self, model_deployment_id: str, **kwargs) -> "AquaDeploymentDetail":

0 commit comments

Comments
 (0)