Change torchmetrics version (#902)

Lightning-Universe · Oct 29, 2021 · 6c2e8ac · 6c2e8ac
1 parent 9927853
commit 6c2e8ac
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -18,7 +18,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 
 - Fixed a bug where validation metrics could be aggregated together with test metrics in some cases ([#900](https://github.com/PyTorchLightning/lightning-flash/pull/900))
-
+- Fixed a bug where the latest versions of torchmetrics and Lightning Flash could not be installed together ([#902](https://github.com/PyTorchLightning/lightning-flash/pull/902))
 
 ## [0.5.1] - 2021-10-26
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 packaging
 numpy
 torch>=1.7.1
-torchmetrics<=0.5.0
+torchmetrics>=0.4.0,!=0.5.1
 pytorch-lightning==1.4.9
 pyDeprecate
 pandas<1.3.0