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