sagemaker>=2.125.0
psutil

[cdk]
aws-cdk-lib==2.64.0
constructs<11.0.0,>=10.0.0

[dev]
sagemaker-pytorch-training
sagemaker-pytorch-inference
torch-model-archiver
tox
wheel
build
twine
pydevd-pycharm~=222.4459.20
scikit-learn

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

[test]
coverage
flake8
mock
pydocstyle
pytest
pytest-cov
pytest-html
bandit
aws-cdk-lib==2.64.0
constructs<11.0.0,>=10.0.0
sagemaker-studio-image-build
sagemaker-training
