lightning/requirements/pytorch/test.txt

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

19 lines
532 B
Plaintext
Raw Normal View History

coverage==6.5.0
pytest==7.2.2
pytest-cov==4.0.0
pytest-forked==1.4.0
pytest-rerunfailures==10.3
2021-10-26 11:58:20 +00:00
# needed in tests
cloudpickle>=1.3, <2.3.0
scikit-learn>0.22.1, <1.2.1
onnx<1.14.0
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