Skip to content

Commit 9a75879

Browse files
committed
Fix formatting
1 parent 73bc6c4 commit 9a75879

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

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

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1343,8 +1343,10 @@ def test_build_for_transformers_happy_case_with_valid_gpu_fallback(
13431343
self.assertEqual(model_builder._can_fit_on_single_gpu(), True)
13441344

13451345
@patch("sagemaker.serve.builder.model_builder.ModelBuilder._build_for_transformers", Mock())
1346-
@patch("sagemaker.serve.builder.model_builder.accelerate.commands.estimate"
1347-
".estimate_command_parser")
1346+
@patch(
1347+
"sagemaker.serve.builder.model_builder.accelerate.commands.estimate"
1348+
".estimate_command_parser"
1349+
)
13481350
@patch("sagemaker.serve.builder.model_builder.accelerate.commands.estimate.gather_data")
13491351
@patch("sagemaker.image_uris.retrieve")
13501352
@patch("sagemaker.djl_inference.model.urllib")

0 commit comments

Comments
 (0)