Skip to content

Commit e847e69

Browse files
author
Deng
committed
fix black erroe
1 parent acb6c5d commit e847e69

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tests/integ/test_pytorch_train.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -119,10 +119,7 @@ def test_deploy_model(pytorch_training_job, sagemaker_session, cpu_instance_type
119119
assert output.shape == (batch_size, 10)
120120

121121

122-
@pytest.mark.skipif(
123-
PYTHON_VERSION == "py2",
124-
reason="PyTorch EIA does not support Python 2.",
125-
)
122+
@pytest.mark.skipif(PYTHON_VERSION == "py2", reason="PyTorch EIA does not support Python 2.")
126123
def test_deploy_model_with_accelerator(sagemaker_session, cpu_instance_type):
127124
endpoint_name = "test-pytorch-deploy-eia-{}".format(sagemaker_timestamp())
128125
model_data = sagemaker_session.upload_data(path=EIA_MODEL)

0 commit comments

Comments
 (0)