File tree Expand file tree Collapse file tree 3 files changed +8
-4
lines changed Expand file tree Collapse file tree 3 files changed +8
-4
lines changed Original file line number Diff line number Diff line change 29
29
)
30
30
31
31
32
- def determine_prefix (user_agent ):
32
+ def determine_prefix (user_agent = "" ):
33
33
"""Placeholder docstring"""
34
34
prefix = "AWS-SageMaker-Python-SDK/{}" .format (SDK_VERSION )
35
35
@@ -56,7 +56,7 @@ def prepend_user_agent(client):
56
56
Args:
57
57
client:
58
58
"""
59
- prefix = determine_prefix (client ._client_config .user_agent or "" )
59
+ prefix = determine_prefix (client ._client_config .user_agent )
60
60
61
61
if client ._client_config .user_agent is None :
62
62
client ._client_config .user_agent = prefix
Original file line number Diff line number Diff line change 43
43
@pytest .fixture ()
44
44
def sagemaker_session ():
45
45
boto_mock = MagicMock (name = "boto_session" , region_name = REGION )
46
- ims = sagemaker .Session (sagemaker_client = MagicMock (name = "sagemaker_client" ), boto_session = boto_mock )
46
+ ims = sagemaker .Session (
47
+ sagemaker_client = MagicMock (name = "sagemaker_client" ), boto_session = boto_mock
48
+ )
47
49
ims .sagemaker_client .describe_training_job = Mock (
48
50
name = "describe_training_job" , return_value = TRAINING_JOB_RESPONSE
49
51
)
Original file line number Diff line number Diff line change 36
36
@pytest .fixture ()
37
37
def sagemaker_session ():
38
38
boto_mock = MagicMock (name = "boto_session" , region_name = REGION )
39
- ims = sagemaker .Session (sagemaker_client = MagicMock (name = "sagemaker_client" ), boto_session = boto_mock )
39
+ ims = sagemaker .Session (
40
+ sagemaker_client = MagicMock (name = "sagemaker_client" ), boto_session = boto_mock
41
+ )
40
42
ims .sagemaker_client .describe_model = Mock (
41
43
name = "describe_model" , side_effect = _raise_does_not_exist_client_error
42
44
)
You can’t perform that action at this time.
0 commit comments