Skip to content

Commit 60c13b3

Browse files
author
Malav Shastri
committed
Address codestyledoc issues
1 parent 7e307bf commit 60c13b3

File tree

2 files changed

+11
-4
lines changed

2 files changed

+11
-4
lines changed

src/sagemaker/jumpstart/accessors.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,8 @@ def get_model_specs(
301301

302302
except Exception as ex:
303303
logging.info(
304-
"Recieved exeption while calling APIs for ContentType Model, retrying with ContentType ModelReference: "
304+
"Recieved exeption while calling APIs for ContentType Model, \
305+
retrying with ContentType ModelReference: "
305306
+ str(ex)
306307
)
307308
hub_model_arn = construct_hub_model_arn_from_inputs(

src/sagemaker/jumpstart/utils.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1012,7 +1012,9 @@ def get_jumpstart_configs(
10121012
)
10131013

10141014

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:
10161018
"""Returns the model-specific user agent string to be added to requests."""
10171019
sagemaker_python_sdk_headers = get_user_agent_extra_suffix()
10181020
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
10251027

10261028

10271029
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,
10291033
) -> Session:
10301034
"""Returns default JumpStart SageMaker Session with model-specific user agent suffix."""
10311035
botocore_session = botocore.session.get_session()
10321036
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+
),
10341040
)
10351041
botocore_session.set_default_client_config(botocore_config)
10361042
# shallow copy to not affect default session constant

0 commit comments

Comments
 (0)