update linter

Author: bencrabtree

src/sagemaker/jumpstart/cache.py

@@ -473,7 +473,9 @@ def get_hub_model(self, hub_model_arn: str) -> JumpStartModelSpecs:
             hub_model_arn (str): Arn for the Hub model to get specs for
         """
 
-        details, _ = self._content_cache.get(JumpStartCachedContentKey(HubDataType.MODEL, hub_model_arn))
+        details, _ = self._content_cache.get(
+            JumpStartCachedContentKey(HubDataType.MODEL, hub_model_arn)
+        )
         return details.formatted_content
 
     def get_hub(self, hub_arn: str) -> Dict[str, Any]:

tests/unit/sagemaker/jumpstart/test_cache.py

@@ -860,16 +860,12 @@ def test_jumpstart_local_metadata_override_specs_not_exist_both_directories(
 def test_jumpstart_cache_get_hub_model():
     cache = JumpStartModelsCache(s3_bucket_name="some_bucket")
 
-    model_arn = (
-        "arn:aws:sagemaker:us-west-2:000000000000:hub-content/HubName/Model/huggingface-mock-model-123/1.2.3"
-    )
+    model_arn = "arn:aws:sagemaker:us-west-2:000000000000:hub-content/HubName/Model/huggingface-mock-model-123/1.2.3"
     assert get_spec_from_base_spec(
         model_id="huggingface-mock-model-123", version="1.2.3"
     ) == cache.get_hub_model(hub_model_arn=model_arn)
 
-    model_arn = (
-        "arn:aws:sagemaker:us-west-2:000000000000:hub-content/HubName/Model/pytorch-mock-model-123/*"
-    )
-    assert get_spec_from_base_spec(model_id="pytorch-mock-model-123", version="*") == cache.get_hub_model(
-        hub_model_arn=model_arn
-    )
+    model_arn = "arn:aws:sagemaker:us-west-2:000000000000:hub-content/HubName/Model/pytorch-mock-model-123/*"
+    assert get_spec_from_base_spec(
+        model_id="pytorch-mock-model-123", version="*"
+    ) == cache.get_hub_model(hub_model_arn=model_arn)