retrying
sagemaker-training>=3.5.0
six>=1.12.0

[:python_version >= "3.6"]
sagemaker-experiments==0.1.7

[test]
boto3==1.10.32
coverage==4.5.3
docker-compose==1.23.2
flake8==3.7.7
Flask==1.1.1
mock==2.0.0
pytest==4.4.0
pytest-cov==2.7.1
pytest-xdist==1.28.0
PyYAML==3.10
sagemaker==1.28.1
torch==1.4.0
torchvision==0.5.0
tox==3.7.0
requests_mock==1.6.0
sagemaker-experiments==0.1.7
