Skip to content

Commit dc5d407

Browse files
author
Deng
committed
update pytorch_train skip message
1 parent 21f0a6a commit dc5d407

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/integ/test_pytorch_train.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,10 @@ def test_fit_deploy(sagemaker_local_session, pytorch_full_version):
8888
predictor.delete_endpoint()
8989

9090

91-
@pytest.mark.skipif(PYTHON_VERSION == "py2", reason="PyTorch Inference not supporting Python2.")
91+
@pytest.mark.skipif(
92+
PYTHON_VERSION == "py2",
93+
reason="Python 2 is supported by PyTorch {} and lower versions.".format(LATEST_PY2_VERSION),
94+
)
9295
def test_deploy_model(pytorch_training_job, sagemaker_session, cpu_instance_type):
9396
endpoint_name = "test-pytorch-deploy-model-{}".format(sagemaker_timestamp())
9497

0 commit comments

Comments
 (0)