Skip to content

Commit c87d17c

Browse files
author
Christian Osendorfer
committed
Resolve black formating issue.
1 parent e7d7085 commit c87d17c

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/unit/test_estimator.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -1245,6 +1245,7 @@ def test_custom_code_bucket(time, sagemaker_session):
12451245
expected_submit_dir
12461246
)
12471247

1248+
12481249
@patch("time.strftime", return_value=TIMESTAMP)
12491250
def test_custom_code_bucket_without_prefix(time, sagemaker_session):
12501251
code_bucket = "codebucket"
@@ -1267,7 +1268,7 @@ def test_custom_code_bucket_without_prefix(time, sagemaker_session):
12671268
_, _, train_kwargs = sagemaker_session.train.mock_calls[0]
12681269
assert train_kwargs["hyperparameters"]["sagemaker_submit_directory"] == json.dumps(
12691270
expected_submit_dir
1270-
)
1271+
)
12711272

12721273

12731274
def test_invalid_custom_code_bucket(sagemaker_session):
@@ -1341,7 +1342,7 @@ def test_shuffle_config(sagemaker_session):
13411342
BASE_HP = {
13421343
"sagemaker_program": json.dumps(SCRIPT_NAME),
13431344
"sagemaker_submit_directory": json.dumps(
1344-
"s3://mybucket/{}/source/sourcedir.tar.gz".format(JOB_NAME)
1345+
"s3://mybucket/{}/source/sourcedir.tar.gz".format(JOB_NAME)
13451346
),
13461347
"sagemaker_job_name": json.dumps(JOB_NAME),
13471348
}

0 commit comments

Comments
 (0)