Skip to content

Commit ed05700

Browse files
committed
Minor update to description
1 parent 30b6c88 commit ed05700

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,11 +73,11 @@
7373
MEMORY_BUFFER_MULTIPLIER = 1.2 # 20% buffer
7474
VERSION_DETECTION_ERROR = (
7575
"Please install accelerate and transformers for HuggingFace (HF) model "
76-
"size calculations pip install 'sagemaker[huggingface]'"
76+
"size calculations e.g. pip install 'sagemaker[huggingface]'"
7777
)
7878

7979

80-
# pylint: disable=attribute-defined-outside-init
80+
# pylint: disable=attribute-defined-outside-init, disable=E1101
8181
@dataclass
8282
class ModelBuilder(Triton, DJL, JumpStart, TGI, Transformers):
8383
"""Class that builds a deployable model.

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_command_parser")
1347-
@patch("sagemaker.serve.builder.model_builder.gather_data")
1346+
@patch("sagemaker.serve.builder.model_builder.accelerate.commands.estimate"
1347+
".estimate_command_parser.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)