Update

Author: amzn-choeric

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

@@ -24,7 +24,7 @@
     "0.9.3": "2.0.1-tgi0.9.3-gpu-py39-cu118-ubuntu20.04",
     "1.0.3": "2.0.1-tgi1.0.3-gpu-py39-cu118-ubuntu20.04",
     "1.1.0": "2.0.1-tgi1.1.0-gpu-py39-cu118-ubuntu20.04",
-    "1.2.0": "2.1.1-tgi1.2.0-gpu-py310-cu121-ubuntu20.04"
+    "1.2.0": "2.1.1-tgi1.2.0-gpu-py310-cu121-ubuntu20.04",
 }
 
 
@@ -34,7 +34,9 @@ def test_huggingface_uris(load_config):
     for version in VERSIONS:
         ACCOUNTS = load_config["inference"]["versions"][version]["registries"]
         for region in ACCOUNTS.keys():
-            uri = get_huggingface_llm_image_uri("huggingface", region=region, version=version)
+            uri = get_huggingface_llm_image_uri(
+                "huggingface", region=region, version=version
+            )
             expected = expected_uris.huggingface_llm_framework_uri(
                 "huggingface-pytorch-tgi-inference",
                 ACCOUNTS[region],
@@ -59,7 +61,9 @@ def test_lmi_uris(load_config):
                 if region not in DJL_DEEPSPEED_REGIONS:
                     continue
 
-                uri = get_huggingface_llm_image_uri("lmi", region=region, version=lmi_version)
+                uri = get_huggingface_llm_image_uri(
+                    "lmi", region=region, version=lmi_version
+                )
                 tag = djl_deepspeed_config["versions"][lmi_version]["tag_prefix"]
 
                 expected = expected_uris.djl_framework_uri(