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
Original file line number	Diff line number	Diff line change
Expand Up		@@ -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

Expand Down