Skip to content

Commit 4a909d2

Browse files
authored
fix(tests): fix TestLoraAcceptEula missing dataclass attrs and patches
1 parent 2b98c5d commit 4a909d2

File tree

1 file changed

+23
-2
lines changed

1 file changed

+23
-2
lines changed

sagemaker-serve/tests/unit/test_model_builder.py

Lines changed: 23 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)