lightning/requirements/pytorch/test.txt

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

20 lines
575 B
Plaintext
Raw Normal View History

coverage ==6.5.0
pytest ==7.3.1
pytest-cov ==4.0.0
pytest-forked ==1.4.0
pytest-rerunfailures ==10.3
pytest-random-order ==1.1.0
2021-10-26 11:58:20 +00:00
# needed in tests
2023-04-24 08:19:20 +00:00
cloudpickle >=1.3, <2.3.0
scikit-learn >0.22.1, <1.2.1
onnx <1.15.0
onnxruntime <1.14.0
psutil <5.9.5 # for `DeviceStatsMonitor`
pandas >1.0, <2.0.2 # needed in benchmarks
fastapi <0.87.0 # for `ServableModuleValidator`
uvicorn <0.22.1 # for `ServableModuleValidator`
2023-04-24 08:19:20 +00:00
tensorboard >=2.9.1, <2.12.0 # for `TensorBoardLogger`
protobuf <=3.20.1 # strict # an extra is updating protobuf, this pin prevents TensorBoard failure