retrying
sagemaker-training>=3.5.1
six>=1.12.0

[test]
boto3
coverage
flake8
future
mock
pytest
pytest-cov
pytest-xdist
sagemaker[local]
torch
torchvision
tox
