missing an @

Bryannah Hernandez · Bryannah Hernandez · commit cd3dbaa62c07 · 2024-07-16T12:42:03.000-07:00
diff --git a/tests/unit/sagemaker/serve/builder/test_model_builder.py b/tests/unit/sagemaker/serve/builder/test_model_builder.py
@@ -84,8 +84,7 @@ def test_validation_in_progress_mode_not_supported(self, mock_serveSettings):
             mock_session,
         )
 
-    patch("sagemaker.serve.builder.model_builder._ServeSettings")
-
+    @patch("sagemaker.serve.builder.model_builder._ServeSettings")
     def test_validation_in_progress_mode_supported(self, mock_serveSettings):
         builder = ModelBuilder(model_server=ModelServer.MMS)
         self.assertRaisesRegex(

Original file line number	Diff line number	Diff line change
`@@ -84,8 +84,7 @@ def test_validation_in_progress_mode_not_supported(self, mock_serveSettings):`
`84`	`84`	`mock_session,`
`85`	`85`	`)`
`86`	`86`
`87`		`- patch("sagemaker.serve.builder.model_builder._ServeSettings")`
`88`		`-`
	`87`	`+ @patch("sagemaker.serve.builder.model_builder._ServeSettings")`
`89`	`88`	`def test_validation_in_progress_mode_supported(self, mock_serveSettings):`
`90`	`89`	`builder = ModelBuilder(model_server=ModelServer.MMS)`
`91`	`90`	`self.assertRaisesRegex(`