diff --git a/requirements-test.txt b/requirements-test.txt index 493dc4a6..2763d5cd 100644 --- a/requirements-test.txt +++ b/requirements-test.txt @@ -5,7 +5,7 @@ flake8-quotes yapf==0.23.0 pytest pytorch-lightning==1.4.7 -lightning-bolts==0.3.3 +lightning-bolts==0.5.0 ray[tune] -torch==1.8.1 +torch==1.9.0 torchvision