Skip to content

Commit 36fcba3

Browse files
committed
Fix imports
1 parent 8f114d1 commit 36fcba3

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1343,8 +1343,9 @@ 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.estimate_parser")
1347-
@patch("sagemaker.serve.builder.model_builder.estimate_gather")
1346+
@patch("sagemaker.serve.builder.model_builder.accelerate.commands.estimate"
1347+
".estimate_command_parser")
1348+
@patch("sagemaker.serve.builder.model_builder.accelerate.commands.estimate.gather_data")
13481349
@patch("sagemaker.image_uris.retrieve")
13491350
@patch("sagemaker.djl_inference.model.urllib")
13501351
@patch("sagemaker.djl_inference.model.json")

0 commit comments

Comments
 (0)