coverage ==7.13.1; python_version >= "3.10"
coverage ==7.10.7; python_version < "3.10"
pytest ==9.0.2
pytest-cov ==7.0.0
pytest-timeout ==2.4.0
pytest-rerunfailures ==16.0.1; python_version < "3.10"
pytest-rerunfailures ==16.1; python_version >= "3.10"
pytest-random-order ==1.2.0

# needed in tests
cloudpickle >=1.3, <3.2.0
scikit-learn >0.22.1, <1.9.0
numpy >1.21.0, <1.27.0; python_version < "3.12"
numpy >2.1.0, <2.4.2; python_version >= "3.12"
onnx >1.12.0, <1.21.0
onnxruntime >=1.12.0, <1.24.0
onnxscript >= 0.1.0, < 0.6.0
onnx-ir < 0.1.16
psutil <7.3.0 # for `DeviceStatsMonitor`
pandas >2.0, <2.4.0  # needed in benchmarks
fastapi  # for `ServableModuleValidator`  # not setting version as re-defined in App
uvicorn  # for `ServableModuleValidator`  # not setting version as re-defined in App

tensorboard >=2.11, <2.21.0  # for `TensorBoardLogger`

torch-tensorrt; platform_system != "Darwin" and python_version >= "3.12"
huggingface-hub
