Skip to content

Commit 26767b6

Browse files
author
Chia-Eng
committed
Reformat files to resolve black-check failures
1 parent eaeda5d commit 26767b6

File tree

9 files changed

+24
-18
lines changed

9 files changed

+24
-18
lines changed

‎src/sagemaker/session.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -456,7 +456,7 @@ def train( # noqa: C901
456456
enable_sagemaker_metrics=None,
457457
profiler_rule_configs=None,
458458
profiler_config=None,
459-
environment=None
459+
environment=None,
460460
):
461461
"""Create an Amazon SageMaker training job.
462462
@@ -560,7 +560,7 @@ def train( # noqa: C901
560560
enable_sagemaker_metrics=enable_sagemaker_metrics,
561561
profiler_rule_configs=profiler_rule_configs,
562562
profiler_config=profiler_config,
563-
environment=environment
563+
environment=environment,
564564
)
565565
LOGGER.info("Creating training-job with name: %s", job_name)
566566
LOGGER.debug("train request: %s", json.dumps(train_request, indent=4))

‎tests/integ/test_tf.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
PARAMETER_SERVER_DISTRIBUTION = {"parameter_server": {"enabled": True}}
3737
MPI_DISTRIBUTION = {"mpi": {"enabled": True}}
3838
TAGS = [{"Key": "some-key", "Value": "some-value"}]
39-
ENV_INPUT = {'env_key1': 'env_val1', 'env_key2': 'env_val2', 'env_key3': 'env_val3'}
39+
ENV_INPUT = {"env_key1": "env_val1", "env_key2": "env_val2", "env_key3": "env_val3"}
4040

4141

4242
def test_mnist_with_checkpoint_config(
@@ -84,9 +84,11 @@ def test_mnist_with_checkpoint_config(
8484
actual_training_checkpoint_config = sagemaker_session.sagemaker_client.describe_training_job(
8585
TrainingJobName=training_job_name
8686
)["CheckpointConfig"]
87-
actual_training_environment_variable_config = sagemaker_session.sagemaker_client.describe_training_job(
88-
TrainingJobName=training_job_name
89-
)["Environment"]
87+
actual_training_environment_variable_config = (
88+
sagemaker_session.sagemaker_client.describe_training_job(TrainingJobName=training_job_name)[
89+
"Environment"
90+
]
91+
)
9092
assert actual_training_checkpoint_config == expected_training_checkpoint_config
9193
assert actual_training_environment_variable_config == ENV_INPUT
9294

‎tests/unit/sagemaker/tensorflow/test_estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def _create_train_job(tf_version, horovod=False, ps=False, py_version="py2", smd
130130
"tags": None,
131131
"vpc_config": None,
132132
"metric_definitions": None,
133-
'environment': None,
133+
"environment": None,
134134
"experiment_config": None,
135135
"profiler_rule_configs": [
136136
{

‎tests/unit/sagemaker/tensorflow/test_estimator_init.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121
REGION = "us-west-2"
2222

23-
ENV_INPUT = {'env_key1': 'env_val1', 'env_key2': 'env_val2', 'env_key3': 'env_val3'}
23+
ENV_INPUT = {"env_key1": "env_val1", "env_key2": "env_val2", "env_key3": "env_val3"}
2424

2525

2626
@pytest.fixture()

‎tests/unit/test_chainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ def _create_train_job(version, py_version):
143143
"tags": None,
144144
"vpc_config": None,
145145
"metric_definitions": None,
146-
'environment': None,
146+
"environment": None,
147147
"experiment_config": None,
148148
"debugger_hook_config": {
149149
"CollectionConfigurations": [],

‎tests/unit/test_estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@
7171
CODECOMMIT_REPO_SSH = "ssh://git-codecommit.us-west-2.amazonaws.com/v1/repos/test-repo/"
7272
CODECOMMIT_BRANCH = "master"
7373
REPO_DIR = "/tmp/repo_dir"
74-
ENV_INPUT = {'env_key1': 'env_val1', 'env_key2': 'env_val2', 'env_key3': 'env_val3'}
74+
ENV_INPUT = {"env_key1": "env_val1", "env_key2": "env_val2", "env_key3": "env_val3"}
7575

7676
DESCRIBE_TRAINING_JOB_RESULT = {"ModelArtifacts": {"S3ModelArtifacts": MODEL_DATA}}
7777

‎tests/unit/test_mxnet.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@
6565

6666
MODEL_PKG_RESPONSE = {"ModelPackageArn": "arn:model-pkg-arn"}
6767

68-
ENV_INPUT = {'env_key1': 'env_val1', 'env_key2': 'env_val2', 'env_key3': 'env_val3'}
68+
ENV_INPUT = {"env_key1": "env_val1", "env_key2": "env_val2", "env_key3": "env_val3"}
6969

7070

7171
@pytest.fixture()
@@ -962,7 +962,9 @@ def test_create_model_with_custom_hosting_image(sagemaker_session):
962962
assert model.image_uri == custom_hosting_image
963963

964964

965-
def test_mx_add_environment_variables(sagemaker_session, mxnet_training_version, mxnet_training_py_version):
965+
def test_mx_add_environment_variables(
966+
sagemaker_session, mxnet_training_version, mxnet_training_py_version
967+
):
966968
mx = MXNet(
967969
entry_point=SCRIPT_PATH,
968970
framework_version=mxnet_training_version,
@@ -976,7 +978,9 @@ def test_mx_add_environment_variables(sagemaker_session, mxnet_training_version,
976978
assert mx.environment == ENV_INPUT
977979

978980

979-
def test_mx_missing_environment_variables(sagemaker_session, mxnet_training_version, mxnet_training_py_version):
981+
def test_mx_missing_environment_variables(
982+
sagemaker_session, mxnet_training_version, mxnet_training_py_version
983+
):
980984
mx = MXNet(
981985
entry_point=SCRIPT_PATH,
982986
framework_version=mxnet_training_version,

‎tests/unit/test_pytorch.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@
4646

4747
ENDPOINT_CONFIG_DESC = {"ProductionVariants": [{"ModelName": "model-1"}, {"ModelName": "model-2"}]}
4848

49-
ENV_INPUT = {'env_key1': 'env_val1', 'env_key2': 'env_val2', 'env_key3': 'env_val3'}
49+
ENV_INPUT = {"env_key1": "env_val1", "env_key2": "env_val2", "env_key3": "env_val3"}
5050

5151
LIST_TAGS_RESULT = {"Tags": [{"Key": "TagtestKey", "Value": "TagtestValue"}]}
5252

@@ -629,7 +629,7 @@ def test_pt_enable_sm_metrics(
629629

630630

631631
def test_pt_disable_sm_metrics(
632-
sagemaker_session, pytorch_training_version, pytorch_training_py_version
632+
sagemaker_session, pytorch_training_version, pytorch_training_py_version
633633
):
634634
pytorch = _pytorch_estimator(
635635
sagemaker_session,
@@ -641,7 +641,7 @@ def test_pt_disable_sm_metrics(
641641

642642

643643
def test_pt_add_environment_variables(
644-
sagemaker_session, pytorch_training_version, pytorch_training_py_version
644+
sagemaker_session, pytorch_training_version, pytorch_training_py_version
645645
):
646646
pytorch = _pytorch_estimator(
647647
sagemaker_session,
@@ -653,7 +653,7 @@ def test_pt_add_environment_variables(
653653

654654

655655
def test_pt_miss_environment_variables(
656-
sagemaker_session, pytorch_training_version, pytorch_training_py_version
656+
sagemaker_session, pytorch_training_version, pytorch_training_py_version
657657
):
658658
pytorch = _pytorch_estimator(
659659
sagemaker_session,

‎tests/unit/test_session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636

3737
SAMPLE_PARAM_RANGES = [{"Name": "mini_batch_size", "MinValue": "10", "MaxValue": "100"}]
3838

39-
ENV_INPUT = {'env_key1': 'env_val1', 'env_key2': 'env_val2', 'env_key3': 'env_val3'}
39+
ENV_INPUT = {"env_key1": "env_val1", "env_key2": "env_val2", "env_key3": "env_val3"}
4040

4141
REGION = "us-west-2"
4242
STS_ENDPOINT = "sts.us-west-2.amazonaws.com"

0 commit comments

Comments
 (0)