Skip to content
Open
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 8 additions & 6 deletions ads/aqua/verify_policies/verify.py
Original file line number Diff line number Diff line change
Expand Up @@ -277,13 +277,15 @@ def model_deployment(self, **kwargs):
List of result dicts for registration, deployment, and cleanup.
"""
logger.info("[magenta]Verifying Model Deployment")
logger.info("Object, Model, Model deployment will be created.")
logger.info("Model, Model deployment will be created.")
kwargs.pop("consent", None) == True or self._consent()
model_save_bucket = kwargs.pop("bucket", None) or self._prompt(
"Provide bucket name where model artifacts will be saved")
model_register = self._test_model_register(bucket=model_save_bucket)
model_deployment = self._test_model_deployment()
delete_model_result = self._test_delete_model(**kwargs)

self.model_id, test_model_register = self._execute(self._util.register_model)
model_register = [test_model_register.to_dict()]

model_deployment = self._test_model_deployment() if self.model_id else []
delete_model_result = self._test_delete_model(**kwargs) if self.model_id else []


return [*model_register, *model_deployment, *delete_model_result]

Expand Down
Loading