Format files

Author: samruds

src/sagemaker/serve/builder/transformers_builder.py

@@ -38,6 +38,7 @@
 DEFAULT_TIMEOUT = 1800
 
 
+# pylint: disable=W0108
 class Transformers(ABC):
     """Transformers build logic with ModelBuilder()"""
 
@@ -87,7 +88,7 @@ def _create_transformers_model(self) -> Type[Model]:
         )
         hf_config = image_uris.config_for_framework("huggingface").get("inference")
         config = hf_config["versions"]
-        base_hf_version = sorted(config.keys(), key=lambda v: Version(v))[0]  # pylint: disable=W0108
+        base_hf_version = sorted(config.keys(), key=lambda v: Version(v))[0]
 
         if hf_model_md is None:
             raise ValueError("Could not fetch HF metadata")
@@ -256,7 +257,7 @@ def _get_supported_version(self, hf_config, hugging_face_version, base_fw):
         versions_to_return = list()
         for key in list(version_config.keys()):
             if key.startswith(base_fw):
-                base_fw_version = key[len(base_fw):]
+                base_fw_version = key[len(base_fw) :]
                 if len(hugging_face_version.split(".")) == 2:
                     base_fw_version = ".".join(base_fw_version.split(".")[:-1])
                 versions_to_return.append(base_fw_version)

src/sagemaker/serve/model_server/multi_model_server/server.py

@@ -1,7 +1,4 @@
-"""Module for the MultiModel Local and Remote servers
-   This module has: 1. LocalMultiModelServer 2. SageMakerMultiModelServer
-   Link: https://github.com/awslabs/multi-model-server
-"""
+"""Module for the MultiModel Local and Remote servers"""
 from __future__ import absolute_import
 
 import requests

src/sagemaker/serve/utils/predictors.py

@@ -166,9 +166,7 @@ def delete_predictor(self):
 
 
 class TransformersLocalModePredictor(PredictorBase):
-    """Lightweight Transformers predictor for local deployment in IN_PROCESS
-    and LOCAL_CONTAINER modes
-    """
+    """Lightweight Transformers predictor for local deployment"""
 
     def __init__(
         self,