Skip to content

Commit a0b7de4

Browse files
committed
fix black + PR comments.
1 parent d2cb361 commit a0b7de4

File tree

3 files changed

+8
-4
lines changed

3 files changed

+8
-4
lines changed

src/sagemaker/user_agent.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
)
3030

3131

32-
def determine_prefix(user_agent):
32+
def determine_prefix(user_agent=""):
3333
"""Placeholder docstring"""
3434
prefix = "AWS-SageMaker-Python-SDK/{}".format(SDK_VERSION)
3535

@@ -56,7 +56,7 @@ def prepend_user_agent(client):
5656
Args:
5757
client:
5858
"""
59-
prefix = determine_prefix(client._client_config.user_agent or "")
59+
prefix = determine_prefix(client._client_config.user_agent)
6060

6161
if client._client_config.user_agent is None:
6262
client._client_config.user_agent = prefix

tests/unit/test_endpoint_from_job.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,9 @@
4343
@pytest.fixture()
4444
def sagemaker_session():
4545
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+
)
4749
ims.sagemaker_client.describe_training_job = Mock(
4850
name="describe_training_job", return_value=TRAINING_JOB_RESPONSE
4951
)

tests/unit/test_endpoint_from_model_data.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,9 @@
3636
@pytest.fixture()
3737
def sagemaker_session():
3838
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+
)
4042
ims.sagemaker_client.describe_model = Mock(
4143
name="describe_model", side_effect=_raise_does_not_exist_client_error
4244
)

0 commit comments

Comments
 (0)