coverage ==7.10.6
pytest ==8.4.2
pytest-cov ==6.3.0
pytest-timeout ==2.4.0
pytest-rerunfailures ==16.0.1
pytest-random-order ==1.2.0

# needed in tests
cloudpickle >=1.3, <3.2.0
scikit-learn >0.22.1, <1.8.0
numpy >1.20.0, <1.27.0
onnx >1.12.0, <1.20.0
onnxruntime >=1.12.0, <1.23.0
onnxscript >= 0.1.0, < 0.5.0
psutil <7.0.1 # 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 == "Linux" and python_version >= "3.12"
