@@ -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