Skip to content

Commit 4870c0b

Browse files
committed
update types and var names
1 parent 8df4478 commit 4870c0b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/sagemaker/jumpstart/cache.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -475,8 +475,8 @@ def get_hub_model(self, hub_model_arn: str) -> JumpStartModelSpecs:
475475
hub_model_arn (str): Arn for the Hub model to get specs for
476476
"""
477477

478-
specs, _ = self._content_cache.get(JumpStartCachedContentKey(HubDataType.MODEL, hub_model_arn))
479-
return specs.formatted_content
478+
details, _ = self._content_cache.get(JumpStartCachedContentKey(HubDataType.MODEL, hub_model_arn))
479+
return details.formatted_content
480480

481481
def get_hub(self, hub_arn: str) -> Dict[str, Any]:
482482
"""Return descriptive info for a given Hub
@@ -485,8 +485,8 @@ def get_hub(self, hub_arn: str) -> Dict[str, Any]:
485485
hub_arn (str): Arn for the Hub to get info for
486486
"""
487487

488-
manifest, _ = self._content_cache.get(JumpStartCachedContentKey(HubDataType.HUB, hub_arn))
489-
return manifest.formatted_content
488+
details, _ = self._content_cache.get(JumpStartCachedContentKey(HubDataType.HUB, hub_arn))
489+
return details.formatted_content
490490

491491
def get_hub_model(self, hub_model_arn: str) -> JumpStartModelSpecs:
492492
"""Return JumpStart-compatible specs for a given Hub model

0 commit comments

Comments
 (0)