Skip to content

Commit 9adf71d

Browse files
Remove Model deployment policy test in common policies section and remove bucket option form MD
1 parent 11f164a commit 9adf71d

File tree

1 file changed

+5
-11
lines changed

1 file changed

+5
-11
lines changed

ads/aqua/verify_policies/verify.py

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -252,13 +252,7 @@ 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-
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]
255+
return result
262256

263257
def model_register(self, **kwargs):
264258
"""Verifies policies required to register a model, including object storage access.
@@ -283,11 +277,11 @@ def model_deployment(self, **kwargs):
283277
List of result dicts for registration, deployment, and cleanup.
284278
"""
285279
logger.info("[magenta]Verifying Model Deployment")
286-
logger.info("Object, Model, Model deployment will be created.")
280+
logger.info("Model, Model deployment will be created.")
287281
kwargs.pop("consent", None) == True or self._consent()
288-
model_save_bucket = kwargs.pop("bucket", None) or self._prompt(
289-
"Provide bucket name where model artifacts will be saved")
290-
model_register = self._test_model_register(bucket=model_save_bucket)
282+
283+
self.model_id, test_model_register = self._execute(self._util.register_model)
284+
model_register = [test_model_register.to_dict()]
291285

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

0 commit comments

Comments
 (0)