Fix formatting

Author: Bryannah Hernandez

src/sagemaker/serve/builder/transformers_builder.py

@@ -29,7 +29,7 @@
 from sagemaker.huggingface import HuggingFaceModel
 from sagemaker.serve.model_server.multi_model_server.prepare import (
     _create_dir_structure,
-    prepare_for_mms
+    prepare_for_mms,
 )
 from sagemaker.serve.detector.image_detector import (
     auto_detect_container,
@@ -280,7 +280,7 @@ def _build_transformers_env(self):
             self._prepare_for_mode()
 
         logger.info("Model configuration %s", self.pysdk_model)
-        
+
         return self.pysdk_model
 
     def _set_instance(self, **kwargs):
@@ -343,7 +343,6 @@ def _auto_detect_container(self):
         else:
             raise ValueError("Cannot detect required model or inference spec")
 
-
     def _build_for_transformers(self):
         """Method that triggers model build
 

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

@@ -33,7 +33,7 @@ def model_fn(model_dir):
     with open(str(serve_path), mode="rb") as file:
         global inference_spec, schema_builder
         obj = cloudpickle.load(file)
-        if isinstance(obj[0], InferenceSpec): 
+        if isinstance(obj[0], InferenceSpec):
             inference_spec, schema_builder = obj
 
     logger.info("in model_fn")
@@ -100,4 +100,4 @@ def _pickle_file_integrity_check():
 
 
 # on import, execute
-_run_preflight_diagnostics()
+_run_preflight_diagnostics()

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

@@ -75,13 +75,14 @@ def prepare_mms_js_resources(
 
     return _copy_jumpstart_artifacts(model_data, js_id, code_dir)
 
+
 def prepare_for_mms(
-        model_path: str,
-        shared_libs: List[str],
-        dependencies: dict,
-        session: Session,
-        image_uri: str,
-        inference_spec: InferenceSpec = None,
+    model_path: str,
+    shared_libs: List[str],
+    dependencies: dict,
+    session: Session,
+    image_uri: str,
+    inference_spec: InferenceSpec = None,
 ) -> str:
     """This is a one-line summary of the function.
     Args:to
@@ -124,4 +125,4 @@ def prepare_for_mms(
     with open(str(code_dir.joinpath("metadata.json")), "wb") as metadata:
         metadata.write(_MetaData(hash_value).to_json())
 
-    return secret_key
+    return secret_key