UT

Jonathan Makunga · Jonathan Makunga · commit 59edbfb24141 · 2024-07-09T14:43:47.000-07:00
diff --git a/tests/unit/sagemaker/serve/utils/test_optimize_utils.py b/tests/unit/sagemaker/serve/utils/test_optimize_utils.py
@@ -399,5 +399,5 @@ def test_with_non_s3(self, mock_model):
         )
 
         mock_model.add_tags.assert_called_once_with(
-            {"Key": Tag.SPECULATIVE_DRAFT_MODEL_PROVIDER, "Value": "customer"}
+            {"Key": Tag.SPECULATIVE_DRAFT_MODEL_PROVIDER, "Value": "custom"}
         )

Original file line number	Diff line number	Diff line change
`@@ -399,5 +399,5 @@ def test_with_non_s3(self, mock_model):`
`399`	`399`	`)`
`400`	`400`
`401`	`401`	`mock_model.add_tags.assert_called_once_with(`
`402`		`- {"Key": Tag.SPECULATIVE_DRAFT_MODEL_PROVIDER, "Value": "customer"}`
	`402`	`+ {"Key": Tag.SPECULATIVE_DRAFT_MODEL_PROVIDER, "Value": "custom"}`
`403`	`403`	`)`