Fix: codestyles (#1606)

Author: benieric

src/sagemaker/image_uris.py

@@ -17,7 +17,7 @@
 import logging
 import os
 import re
-from typing import Optional, Tuple
+from typing import Optional
 from packaging.version import Version
 
 from sagemaker import utils
@@ -463,6 +463,7 @@ def _get_latest_versions(list_of_versions):
     print("SORT")
     return sorted(list_of_versions, reverse=True)[0]
 
+
 def _get_latest_version(framework, version, image_scope):
     """Get the latest version from the input framework"""
     if version:
@@ -479,6 +480,7 @@ def _get_latest_version(framework, version, image_scope):
         version = _fetch_latest_version_from_config(framework_config, image_scope)
     return version
 
+
 def _validate_accelerator_type(accelerator_type):
     """Raises a ``ValueError`` if ``accelerator_type`` is invalid."""
     if not accelerator_type.startswith("ml.eia") and accelerator_type != "local_sagemaker_notebook":
@@ -748,7 +750,7 @@ def get_base_python_image_uri(region, py_version="310") -> str:
     return ECR_URI_TEMPLATE.format(registry=registry, hostname=hostname, repository=repo_and_tag)
 
 
-def _fetch_latest_version_from_config(
+def _fetch_latest_version_from_config(  # pylint: disable=R0911
     framework_config: dict, image_scope: Optional[str] = None
 ) -> Optional[str]:
     """Helper function to fetch the latest version as a string from a framework's config

src/sagemaker/serve/builder/model_builder.py

@@ -477,10 +477,9 @@ def _get_client_translators(self):
 
         return serializer, deserializer
 
-    def _get_predictor(self,
-                       endpoint_name: str,
-                       sagemaker_session: Session,
-                       component_name: Optional[str] = None) -> Predictor:
+    def _get_predictor(
+        self, endpoint_name: str, sagemaker_session: Session, component_name: Optional[str] = None
+    ) -> Predictor:
         """Placeholder docstring"""
         serializer, deserializer = self._get_client_translators()
 
@@ -489,7 +488,7 @@ def _get_predictor(self,
             sagemaker_session=sagemaker_session,
             serializer=serializer,
             deserializer=deserializer,
-            component_name=component_name
+            component_name=component_name,
         )
 
     def _create_model(self):

tests/unit/sagemaker/image_uris/test_retrieve.py

@@ -16,7 +16,6 @@
 import logging
 
 import pytest
-from contextlib import nullcontext
 from mock import patch
 
 from sagemaker import image_uris
@@ -715,8 +714,8 @@ def test_retrieve_huggingface(config_for_framework):
         container_version="cu110-ubuntu18.04",
     )
     assert (
-            "564829616587.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-training:"
-            "1.6.0-transformers4.3.1-gpu-py37-cu110-ubuntu18.04" == pt_new_version
+        "564829616587.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-training:"
+        "1.6.0-transformers4.3.1-gpu-py37-cu110-ubuntu18.04" == pt_new_version
     )
 
 
@@ -769,6 +768,7 @@ def test_retrieve_with_pipeline_variable():
         image_scope="training",
     )
 
+
 @patch("sagemaker.image_uris.config_for_framework")
 def test_get_latest_version_function_with_invalid_framework(config_for_framework):
     config_for_framework.side_effect = FileNotFoundError
@@ -777,6 +777,7 @@ def test_get_latest_version_function_with_invalid_framework(config_for_framework
         image_uris.retrieve("xgboost", "inference")
         assert "No framework config for framework" in str(e.exception)
 
+
 @patch("sagemaker.image_uris.config_for_framework")
 def test_get_latest_version_function_with_no_framework(config_for_framework):
     config_for_framework.side_effect = {}
@@ -785,6 +786,7 @@ def test_get_latest_version_function_with_no_framework(config_for_framework):
         image_uris.retrieve("xgboost", "inference")
         assert "No framework config for framework" in str(e.exception)
 
+
 @pytest.mark.parametrize(
     "framework",
     [
@@ -854,7 +856,6 @@ def test_get_latest_version_function_with_no_framework(config_for_framework):
         "sagemaker-base-python",
     ],
 )
-
 @patch("sagemaker.image_uris.config_for_framework")
 @patch("sagemaker.image_uris.retrieve")
 def test_retrieve_with_parameterized(mock_image_retrieve, mock_config_for_framework, framework):
@@ -867,4 +868,4 @@ def test_retrieve_with_parameterized(mock_image_retrieve, mock_config_for_framew
             image_scope="inference",
         )
     except ValueError as e:
-        pytest.fail(e.value)
+        pytest.fail(e.value)