@@ -1012,7 +1012,9 @@ def get_jumpstart_configs(
1012
1012
)
1013
1013
1014
1014
1015
- def get_jumpstart_user_agent_extra_suffix (model_id : Optional [str ], model_version : Optional [str ], is_hub_content : Optional [bool ]) -> str :
1015
+ def get_jumpstart_user_agent_extra_suffix (
1016
+ model_id : Optional [str ], model_version : Optional [str ], is_hub_content : Optional [bool ]
1017
+ ) -> str :
1016
1018
"""Returns the model-specific user agent string to be added to requests."""
1017
1019
sagemaker_python_sdk_headers = get_user_agent_extra_suffix ()
1018
1020
jumpstart_specific_suffix = f"md/js_model_id#{ model_id } md/js_model_ver#{ model_version } "
@@ -1025,12 +1027,16 @@ def get_jumpstart_user_agent_extra_suffix(model_id: Optional[str], model_version
1025
1027
1026
1028
1027
1029
def get_default_jumpstart_session_with_user_agent_suffix (
1028
- model_id : Optional [str ] = None , model_version : Optional [str ] = None , is_hub_content : Optional [bool ] = False
1030
+ model_id : Optional [str ] = None ,
1031
+ model_version : Optional [str ] = None ,
1032
+ is_hub_content : Optional [bool ] = False ,
1029
1033
) -> Session :
1030
1034
"""Returns default JumpStart SageMaker Session with model-specific user agent suffix."""
1031
1035
botocore_session = botocore .session .get_session ()
1032
1036
botocore_config = botocore .config .Config (
1033
- user_agent_extra = get_jumpstart_user_agent_extra_suffix (model_id , model_version , is_hub_content ),
1037
+ user_agent_extra = get_jumpstart_user_agent_extra_suffix (
1038
+ model_id , model_version , is_hub_content
1039
+ ),
1034
1040
)
1035
1041
botocore_session .set_default_client_config (botocore_config )
1036
1042
# shallow copy to not affect default session constant
0 commit comments