fixed black-check

aws · Apr 17, 2024 · 01f6272 · 01f6272
1 parent 0ba1744
commit 01f6272
Showing 1 changed file with 1 addition and 1 deletion.
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,