sagemaker-core>=2.5.0
sagemaker-train>=1.5.0
boto3<2.0,>=1.42.2
botocore<2.0,>=1.35.75
deepdiff
mlflow
sagemaker_schema_inference_artifacts
pytest
tqdm
psutil
tritonclient[http]
onnx
onnxruntime
torch>=2.0.0

[dev]
pytest
pytest-cov
mock
black
flake8

[test]
pytest
pytest-cov
mock
