sagemaker-core>=2.27.0
sagemaker-train>=1.27.0
boto3<2.0,>=1.42.2
botocore<2.0,>=1.35.75
deepdiff
sagemaker_schema_inference_artifacts
tqdm
psutil

[all]
mlflow
onnx
onnxruntime
torch>=2.0.0
tritonclient[http]

[dev]
pytest
pytest-cov
mock
black
flake8

[mlflow]
mlflow

[onnx]
onnx
onnxruntime

[test]
pytest
pytest-cov
mock
mlflow
onnx
onnxruntime
torch>=2.0.0
tritonclient[http]

[torch]
torch>=2.0.0

[triton]
tritonclient[http]
onnx
onnxruntime
torch>=2.0.0
