Skip to content

Commit b89cac8

Browse files
author
Joseph Zhang
committed
Fix ModelBuilder UTs after removing unique_name_from_base import.
1 parent 5e9cb95 commit b89cac8

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/unit/sagemaker/serve/builder/test_model_builder.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4241,9 +4241,7 @@ def test_neuron_configurations_rule_set(self):
42414241
"Batch",
42424242
],
42434243
)
4244-
@patch("sagemaker.serve.builder.model_builder.unique_name_from_base")
4245-
def test_deploy(mock_unique_name_from_base, test_case):
4246-
mock_unique_name_from_base.return_value = "test"
4244+
def test_deploy(test_case):
42474245
model: Model = MagicMock()
42484246
model_builder = ModelBuilder(
42494247
model="meta-llama/Meta-Llama-3-8B-Instruct",

0 commit comments

Comments
 (0)