File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
tests/unit/sagemaker/serve/builder Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -1343,8 +1343,10 @@ def test_build_for_transformers_happy_case_with_valid_gpu_fallback(
1343
1343
self .assertEqual (model_builder ._can_fit_on_single_gpu (), True )
1344
1344
1345
1345
@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
+ )
1348
1350
@patch ("sagemaker.serve.builder.model_builder.accelerate.commands.estimate.gather_data" )
1349
1351
@patch ("sagemaker.image_uris.retrieve" )
1350
1352
@patch ("sagemaker.djl_inference.model.urllib" )
You can’t perform that action at this time.
0 commit comments