Skip to content

Commit c331b0c

Browse files
author
Malav Shastri
committed
change default session object in hub class to one with user agent string
1 parent 90006f6 commit c331b0c

File tree

1 file changed

+3
-1
lines changed
  • src/sagemaker/jumpstart/hub

1 file changed

+3
-1
lines changed

src/sagemaker/jumpstart/hub/hub.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,9 @@ def __init__(
6868
self,
6969
hub_name: str,
7070
bucket_name: Optional[str] = None,
71-
sagemaker_session: Optional[Session] = DEFAULT_JUMPSTART_SAGEMAKER_SESSION,
71+
sagemaker_session: Optional[
72+
Session
73+
] = utils.get_default_jumpstart_session_with_user_agent_suffix(is_hub_content=True),
7274
) -> None:
7375
"""Instantiates a SageMaker ``Hub``.
7476

0 commit comments

Comments
 (0)