Update according to black and pylint

Author: nikhil-sk

src/sagemaker/image_uris.py

@@ -339,7 +339,7 @@ def _get_image_tag(
     # Triton images don't have a trailing -gpu tag. Only -cpu images do.
     if framework == SAGEMAKER_TRITONSERVER_FRAMEWORK:
         if processor == "gpu":
-            tag = tag.rstrip(f"-gpu")
+            tag = tag.rstrip("-gpu")
 
     return tag
 

tests/unit/sagemaker/image_uris/expected_uris.py

@@ -83,6 +83,7 @@ def djl_framework_uri(repo, account, tag, region=REGION):
     domain = ALTERNATE_DOMAINS.get(region, DOMAIN)
     return IMAGE_URI_FORMAT.format(account, region, domain, repo, tag)
 
+
 def sagemaker_triton_framework_uri(repo, account, tag, processor="gpu", region=REGION):
     domain = ALTERNATE_DOMAINS.get(region, DOMAIN)
     if processor == "cpu":

tests/unit/sagemaker/image_uris/test_sagemaker_tritonserver.py

@@ -17,11 +17,10 @@
 
 INSTANCE_TYPES = {"cpu": "ml.c4.xlarge", "gpu": "ml.p2.xlarge"}
 
+
 @pytest.mark.parametrize(
     "load_config_and_file_name",
-    [
-        "sagemaker-tritonserver.json"
-    ],
+    ["sagemaker-tritonserver.json"],
     indirect=True,
 )
 def test_sagemaker_tritonserver_uris(load_config_and_file_name):
@@ -35,11 +34,17 @@ def test_sagemaker_tritonserver_uris(load_config_and_file_name):
         for processor in processors:
             instance_type = INSTANCE_TYPES[processor]
             for region in ACCOUNTS.keys():
-                _test_sagemaker_tritonserver_uris(ACCOUNTS[region], region, version, tag, framework, instance_type, processor)
+                _test_sagemaker_tritonserver_uris(
+                    ACCOUNTS[region], region, version, tag, framework, instance_type, processor
+                )
 
 
-def _test_sagemaker_tritonserver_uris(account, region, version, tag, triton_framework, instance_type, processor):
-    uri = image_uris.retrieve(framework=triton_framework, region=region, version=version, instance_type=instance_type)
+def _test_sagemaker_tritonserver_uris(
+    account, region, version, tag, triton_framework, instance_type, processor
+):
+    uri = image_uris.retrieve(
+        framework=triton_framework, region=region, version=version, instance_type=instance_type
+    )
     expected = expected_uris.sagemaker_triton_framework_uri(
         "sagemaker-tritonserver",
         account,