Skip to content

Commit c336dad

Browse files
committed
fix black-check
1 parent d2d0253 commit c336dad

File tree

1 file changed

+6
-12
lines changed

1 file changed

+6
-12
lines changed

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

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1722,10 +1722,8 @@ def test_build_mlflow_model_local_input_happy(
17221722
mock_path_exists.side_effect = lambda path: True if path == "test_path" else False
17231723

17241724
mock_mode = Mock()
1725-
mock_sageMakerEndpointMode.side_effect = (
1726-
lambda inference_spec, model_server: mock_mode
1727-
if inference_spec is None and model_server == ModelServer.TORCHSERVE
1728-
else None
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
17291727
)
17301728
mock_mode.prepare.return_value = (
17311729
model_data,
@@ -1799,10 +1797,8 @@ def test_build_mlflow_model_s3_input_happy(
17991797
mock_path_exists.side_effect = lambda path: True if path == "test_path" else False
18001798

18011799
mock_mode = Mock()
1802-
mock_sageMakerEndpointMode.side_effect = (
1803-
lambda inference_spec, model_server: mock_mode
1804-
if inference_spec is None and model_server == ModelServer.TORCHSERVE
1805-
else None
1800+
mock_sageMakerEndpointMode.side_effect = lambda inference_spec, model_server: (
1801+
mock_mode if inference_spec is None and model_server == ModelServer.TORCHSERVE else None
18061802
)
18071803
mock_mode.prepare.return_value = (
18081804
model_data,
@@ -1875,10 +1871,8 @@ def test_build_mlflow_model_s3_input_non_mlflow_case(
18751871
mock_path_exists.side_effect = lambda path: True if path == "test_path" else False
18761872

18771873
mock_mode = Mock()
1878-
mock_sageMakerEndpointMode.side_effect = (
1879-
lambda inference_spec, model_server: mock_mode
1880-
if inference_spec is None and model_server == ModelServer.TORCHSERVE
1881-
else None
1874+
mock_sageMakerEndpointMode.side_effect = lambda inference_spec, model_server: (
1875+
mock_mode if inference_spec is None and model_server == ModelServer.TORCHSERVE else None
18821876
)
18831877
mock_mode.prepare.return_value = (
18841878
model_data,

0 commit comments

Comments
 (0)