coverage ==6.5.0
codecov ==2.1.12
pytest ==7.2.0
pytest-cov ==4.0.0
pytest-forked ==1.4.0
pytest-rerunfailures ==10.3
pre-commit ==2.20.0

# needed in tests
cloudpickle >=1.3, <2.3.0
scikit-learn >0.22.1, <1.2.1
onnx <1.14.0  # newer versions of PT does not have it build it any more
onnxruntime <1.14.0
psutil <5.9.5 # for `DeviceStatsMonitor`
pandas >1.0, <1.5.4  # needed in benchmarks
fastapi <0.87.0  # for `ServableModuleValidator`
uvicorn <0.19.1  # for `ServableModuleValidator`

tensorboard >=2.9.1, <2.12.0  # for `TensorBoardLogger`
protobuf <=3.20.1 # strict  # an extra is updating protobuf, this pin prevents TensorBoard failure
