retrying
sagemaker-training>=4.3.0
six>=1.12.0

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