Skip to content

Commit 1e26760

Browse files
committed
feat: add hub and hubcontent support in retrieval function for jumpstart model cache (aws#4438)
1 parent 5492474 commit 1e26760

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/jumpstart/curated_hub/curated_hub.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,11 @@
2222
class CuratedHub:
2323
"""Class for creating and managing a curated JumpStart hub"""
2424

25-
def __init__(self, hub_name: str, region: str, session: Optional[Session] = None):
25+
def __init__(self, hub_name: str, region: str, session: Optional[Session] = DEFAULT_JUMPSTART_SAGEMAKER_SESSION):
2626
self.hub_name = hub_name
2727
self.region = region
2828
self.session = session
29-
self._sm_session = session or DEFAULT_JUMPSTART_SAGEMAKER_SESSION
29+
self._sm_session = session
3030

3131
def describe_model(self, model_name: str, model_version: str = "*") -> Dict[str, Any]:
3232
"""Returns descriptive information about the Hub Model"""

0 commit comments

Comments
 (0)