lightning/requirements/pytorch/test.txt

20 lines
593 B
Plaintext

coverage ==7.3.1
pytest ==7.4.0
pytest-cov ==4.1.0
pytest-timeout ==2.1.0
pytest-rerunfailures ==12.0
pytest-random-order ==1.1.0
# needed in tests
cloudpickle >=1.3, <2.3.0
scikit-learn >0.22.1, <1.4.0
numpy >=1.17.2, <1.27.0
onnx >=1.12.0, <1.17.0
onnxruntime >=1.12.0, <1.19.0
psutil <5.9.6 # for `DeviceStatsMonitor`
pandas >1.0, <2.2.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.9.1, <2.15.0 # for `TensorBoardLogger`