@@ -730,16 +730,37 @@ def _make_mb(self, accept_eula=None):
730730 mb .role_arn = "arn:aws:iam::123456789012:role/role"
731731 mb .model = MagicMock ()
732732 mb ._adapter_s3_uri = None
733+ mb .shared_libs = []
734+ mb .dependencies = {"auto" : True }
735+ mb .image_config = None
736+ mb .inference_spec = None
737+ mb .schema_builder = None
738+ mb .modelbuilder_list = None
739+ mb .sagemaker_session = None
740+ mb .s3_model_data_url = None
741+ mb .source_code = None
742+ mb .model_server = None
743+ mb .model_metadata = None
744+ mb .log_level = None
745+ mb .content_type = None
746+ mb .accept_type = None
747+ mb .compute = None
748+ mb .network = None
749+ mb .instance_type = None
750+ mb .mode = None
733751 return mb
734752
735753 def _patch_lora_deps (self , mb , hosting_uri = "s3://bucket/hosting/" ):
736754 """Patch all dependencies needed to reach the LoRA ContainerDefinition block."""
737755 patches = [
756+ patch .object (mb , "_get_serve_setting" , return_value = MagicMock ()),
757+ patch .object (mb , "_is_model_customization" , return_value = True ),
758+ patch .object (mb , "_fetch_model_package" , return_value = MagicMock ()),
759+ patch .object (mb , "_fetch_and_cache_recipe_config" ),
760+ patch .object (mb , "_is_nova_model" , return_value = False ),
738761 patch .object (mb , "_fetch_peft" , return_value = "LORA" ),
739762 patch .object (mb , "_fetch_hub_document_for_custom_model" ,
740763 return_value = {"HostingArtifactUri" : hosting_uri }),
741- patch .object (mb , "_get_model_package_for_training_job" ,
742- return_value = MagicMock ()),
743764 ]
744765 return patches
745766
0 commit comments