Fixing codestyle issues

Author: aviruthen

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

@@ -30,9 +30,7 @@
 class LocalDJLServing:
     """Placeholder docstring"""
 
-    def _start_djl_serving(
-        self, client: object, image: str, model_path: str, env_vars: dict
-    ):
+    def _start_djl_serving(self, client: object, image: str, model_path: str, env_vars: dict):
         """Placeholder docstring"""
         updated_env_vars = _update_env_vars(env_vars)
 

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

@@ -140,14 +140,16 @@ def _upload_server_artifacts(
 
         if env_vars is None:
             env_vars = {}
-        
-        env_vars.update({
-            "SAGEMAKER_SUBMIT_DIRECTORY": "/opt/ml/model/code",
-            "SAGEMAKER_PROGRAM": "inference.py",
-            "SAGEMAKER_REGION": sagemaker_session.boto_region_name,
-            "SAGEMAKER_CONTAINER_LOG_LEVEL": "10",
-            "LOCAL_PYTHON": platform.python_version(),
-        })
+
+        env_vars.update(
+            {
+                "SAGEMAKER_SUBMIT_DIRECTORY": "/opt/ml/model/code",
+                "SAGEMAKER_PROGRAM": "inference.py",
+                "SAGEMAKER_REGION": sagemaker_session.boto_region_name,
+                "SAGEMAKER_CONTAINER_LOG_LEVEL": "10",
+                "LOCAL_PYTHON": platform.python_version(),
+            }
+        )
 
         return model_data, _update_env_vars(env_vars)
 

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

@@ -27,9 +27,7 @@
 class LocalTeiServing:
     """LocalTeiServing class"""
 
-    def _start_tei_serving(
-        self, client: object, image: str, model_path: str, env_vars: dict
-    ):
+    def _start_tei_serving(self, client: object, image: str, model_path: str, env_vars: dict):
         """Starts a local tei serving container.
 
         Args:

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

@@ -21,9 +21,7 @@
 class LocalTorchServe:
     """Placeholder docstring"""
 
-    def _start_torch_serve(
-        self, client: object, image: str, model_path: str, env_vars: dict
-    ):
+    def _start_torch_serve(self, client: object, image: str, model_path: str, env_vars: dict):
         """Placeholder docstring"""
         self.container = client.containers.run(
             image,

src/sagemaker/serve/save_retrive/version_1_0_0/save/utils.py

@@ -129,10 +129,10 @@ def capture_optimization_metadata(model: Model, framework) -> dict:
 
 def compute_hash(buffer: bytes) -> str:
     """Compute SHA256 hash value of buffer.
-    
+
     Args:
         buffer: Bytes to hash
-        
+
     Returns:
         Hexadecimal SHA256 hash string
     """

src/sagemaker/serve/validations/check_integrity.py

@@ -9,10 +9,10 @@
 
 def compute_hash(buffer: bytes) -> str:
     """Compute SHA256 hash value of buffer.
-    
+
     Args:
         buffer: Bytes to hash
-        
+
     Returns:
         Hexadecimal SHA256 hash string
     """
@@ -21,11 +21,11 @@ def compute_hash(buffer: bytes) -> str:
 
 def perform_integrity_check(buffer: bytes, metadata_path: Path):
     """Validates the integrity of bytes by comparing SHA256 hash.
-    
+
     Args:
         buffer: Bytes to verify
         metadata_path: Path to metadata.json file
-        
+
     Raises:
         ValueError: If metadata file doesn't exist or hash doesn't match
     """

tests/unit/sagemaker/serve/builder/test_model_builder.py

@@ -1123,10 +1123,7 @@ def test_build_happy_path_with_sagemaker_endpoint_mode_overwritten_with_local_co
         mock_lc_mode.prepare.side_effect = lambda: None
         mock_lc_mode.create_server.side_effect = (
             lambda image_uri, container_timeout_seconds, predictor, **kwargs: (
-                None
-                if image_uri == mock_image_uri
-                and container_timeout_seconds == 60
-                else None
+                None if image_uri == mock_image_uri and container_timeout_seconds == 60 else None
             )
         )