File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
src/sagemaker/jumpstart/hub Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -67,9 +67,7 @@ def __init__(
67
67
self ,
68
68
hub_name : str ,
69
69
bucket_name : Optional [str ] = None ,
70
- sagemaker_session : Optional [
71
- Session
72
- ] = utils .get_default_jumpstart_session_with_user_agent_suffix (is_hub_content = True ),
70
+ sagemaker_session : Optional [Session ] = None ,
73
71
) -> None :
74
72
"""Instantiates a SageMaker ``Hub``.
75
73
@@ -80,7 +78,7 @@ def __init__(
80
78
"""
81
79
self .hub_name = hub_name
82
80
self .region = sagemaker_session .boto_region_name
83
- self ._sagemaker_session = sagemaker_session
81
+ self ._sagemaker_session = sagemaker_session or utils . get_default_jumpstart_session_with_user_agent_suffix ( is_hub_content = True )
84
82
self .hub_storage_location = self ._generate_hub_storage_location (bucket_name )
85
83
86
84
def _fetch_hub_bucket_name (self ) -> str :
You can’t perform that action at this time.
0 commit comments