sagemaker>=2.125.0

[test]
black
coverage
flake8
mock
pydocstyle
pytest
pytest-cov
sagemaker-pytorch-training
sagemaker-pytorch-inference
torch-model-archiver
tox
wheel
build
twine
pydevd-pycharm~=213.6777.50
scikit-learn

[test-macos]
tensorflow-macos==2.9.2
numpy==1.22.4
