sagemaker-training>=3.5.2
numpy
scipy
sklearn
pandas
Pillow
h5py

[benchmark]
click

[test]
tox
flake8
pytest
pytest-cov
pytest-xdist
mock
sagemaker==1.50.1
tensorflow<2.0
docker-compose
boto3==1.10.50
six==1.13.0
python-dateutil<2.8.1,>=2.1
botocore==1.13.50
requests-mock
awscli>=1.16.314
sagemaker-experiments==0.1.7
