From ac1d1e813654418a7b764fdb3242f379e2b0e232 Mon Sep 17 00:00:00 2001 From: Samrudhi Sharma Date: Thu, 14 Mar 2024 16:14:35 -0700 Subject: [PATCH] Minor update to description --- src/sagemaker/serve/builder/model_builder.py | 4 ++-- tests/unit/sagemaker/serve/builder/test_model_builder.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/sagemaker/serve/builder/model_builder.py b/src/sagemaker/serve/builder/model_builder.py index 15c0279b63..2547dd586f 100644 --- 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 index 3b60d13dfb..a38c60b60e 100644 --- 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")