Skip to content

Commit 11f164a

Browse files
Add Model deployment policy test in common policies section.
1 parent 5ef6e9c commit 11f164a

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

ads/aqua/verify_policies/verify.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -252,7 +252,13 @@ def common_policies(self, **kwargs):
252252
_, get_resource_availability_status = self._execute(self._util.get_resource_availability,
253253
limit_name=TEST_LIMIT_NAME)
254254
result.append(get_resource_availability_status.to_dict())
255-
return result
255+
256+
self.model_id, test_model_register = self._execute(self._util.register_model)
257+
258+
model_deployment = self._test_model_deployment() if self.model_id else []
259+
delete_model_result = self._test_delete_model(**kwargs) if self.model_id else []
260+
261+
return [*result, *model_deployment, *delete_model_result]
256262

257263
def model_register(self, **kwargs):
258264
"""Verifies policies required to register a model, including object storage access.
@@ -282,8 +288,10 @@ def model_deployment(self, **kwargs):
282288
model_save_bucket = kwargs.pop("bucket", None) or self._prompt(
283289
"Provide bucket name where model artifacts will be saved")
284290
model_register = self._test_model_register(bucket=model_save_bucket)
285-
model_deployment = self._test_model_deployment()
286-
delete_model_result = self._test_delete_model(**kwargs)
291+
292+
model_deployment = self._test_model_deployment() if self.model_id else []
293+
delete_model_result = self._test_delete_model(**kwargs) if self.model_id else []
294+
287295

288296
return [*model_register, *model_deployment, *delete_model_result]
289297

0 commit comments

Comments
 (0)