Minor update to description

aws · Mar 14, 2024 · ac1d1e8 · ac1d1e8
1 parent 30b6c88
commit ac1d1e8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/sagemaker/serve/builder/model_builder.py b/src/sagemaker/serve/builder/model_builder.py
@@ -73,11 +73,11 @@
 MEMORY_BUFFER_MULTIPLIER = 1.2  # 20% buffer
 VERSION_DETECTION_ERROR = (
     "Please install accelerate and transformers for HuggingFace (HF) model "
-    "size calculations pip install 'sagemaker[huggingface]'"
+    "size calculations e.g. pip install 'sagemaker[huggingface]'"
 )
 
 
-# pylint: disable=attribute-defined-outside-init
+# pylint: disable=attribute-defined-outside-init, disable=E1101
 @dataclass
 class ModelBuilder(Triton, DJL, JumpStart, TGI, Transformers):
     """Class that builds a deployable model.

diff --git a/tests/unit/sagemaker/serve/builder/test_model_builder.py b/tests/unit/sagemaker/serve/builder/test_model_builder.py
@@ -1343,8 +1343,8 @@ def test_build_for_transformers_happy_case_with_valid_gpu_fallback(
         self.assertEqual(model_builder._can_fit_on_single_gpu(), True)
 
     @patch("sagemaker.serve.builder.model_builder.ModelBuilder._build_for_transformers", Mock())
-    @patch("sagemaker.serve.builder.model_builder.estimate_command_parser")
-    @patch("sagemaker.serve.builder.model_builder.gather_data")
+    @patch("accelerate.commands.estimate.estimate_command_parser")
+    @patch("accelerate.commands.estimate.gather_data")
     @patch("sagemaker.image_uris.retrieve")
     @patch("sagemaker.djl_inference.model.urllib")
     @patch("sagemaker.djl_inference.model.json")