fixed black-check

knikure · knikure · commit 01f627255232 · 2024-04-17T13:47:09.000Z
diff --git a/tests/unit/sagemaker/serve/builder/test_model_builder.py b/tests/unit/sagemaker/serve/builder/test_model_builder.py
@@ -1723,7 +1723,7 @@ def test_build_mlflow_model_local_input_happy(
 
         mock_mode = Mock()
         mock_sageMakerEndpointMode.side_effect = lambda inference_spec, model_server: (
-	        mock_mode if inference_spec is None and model_server == ModelServer.TORCHSERVE else None
+            mock_mode if inference_spec is None and model_server == ModelServer.TORCHSERVE else None
         )
         mock_mode.prepare.return_value = (
             model_data,

Original file line number	Diff line number	Diff line change
`@@ -1723,7 +1723,7 @@ def test_build_mlflow_model_local_input_happy(`
`1723`	`1723`
`1724`	`1724`	`mock_mode = Mock()`
`1725`	`1725`	`mock_sageMakerEndpointMode.side_effect = lambda inference_spec, model_server: (`
`1726`		`- mock_mode if inference_spec is None and model_server == ModelServer.TORCHSERVE else None`
	`1726`	`+ mock_mode if inference_spec is None and model_server == ModelServer.TORCHSERVE else None`
`1727`	`1727`	`)`
`1728`	`1728`	`mock_mode.prepare.return_value = (`
`1729`	`1729`	`model_data,`