linter

Author: bencrabtree

src/sagemaker/jumpstart/cache.py

@@ -339,10 +339,10 @@ def _retrieval_function(
                 formatted_content=model_specs
             )
         if data_type == HubDataType.MODEL:
-            hub_name, hub_region, model_name, model_version = utils.extract_info_from_hub_content_arn(
+            hub_name, region, model_name, model_version = utils.extract_info_from_hub_content_arn(
                 id_info
             )
-            hub = CuratedHub(hub_name=hub_name, region=hub_region)
+            hub = CuratedHub(hub_name=hub_name, region=region)
             hub_content = hub.describe_model(model_name=model_name, model_version=model_version)
             utils.emit_logs_based_on_model_specs(
                 hub_content.content_document,
@@ -354,8 +354,8 @@ def _retrieval_function(
                 formatted_content=model_specs
             )
         if data_type == HubDataType.HUB:
-            hub_name, hub_region, _, _ = utils.extract_info_from_hub_content_arn(id_info)
-            hub = CuratedHub(hub_name=hub_name, region=hub_region)
+            hub_name, region, _, _ = utils.extract_info_from_hub_content_arn(id_info)
+            hub = CuratedHub(hub_name=hub_name, region=region)
             hub_info = hub.describe()
             return JumpStartCachedContentValue(formatted_content=hub_info)
         raise ValueError(
@@ -465,10 +465,10 @@ def get_specs(self, model_id: str, semantic_version_str: str) -> JumpStartModelS
                 )
             )
         return specs.formatted_content
-    
+
     def get_hub_model(self, hub_model_arn: str) -> JumpStartModelSpecs:
         """Return JumpStart-compatible specs for a given Hub model
-        
+
         Args:
             hub_model_arn (str): Arn for the Hub model to get specs for
         """
@@ -477,14 +477,14 @@ def get_hub_model(self, hub_model_arn: str) -> JumpStartModelSpecs:
             JumpStartCachedContentKey(HubDataType.MODEL, hub_model_arn)
         )
         return details.formatted_content
-    
+
     def get_hub(self, hub_arn: str) -> Dict[str, Any]:
         """Return descriptive info for a given Hub
-        
+
         Args:
             hub_arn (str): Arn for the Hub to get info for
         """
-        
+
         details, _ = self._content_cache.get(JumpStartCachedContentKey(HubDataType.HUB, hub_arn))
         return details.formatted_content