Uncap the torchmetrics version, so we can get fixes in 1.0+. (#3741)

Co-authored-by: Trevor DiMartino <trevor.dimartino@.workday.com> Co-authored-by: Justin Zhao <justinxzhao@gmail.com>
ludwig-ai · Oct 20, 2023 · 77259c7 · 77259c7
1 parent e6c60ab
commit 77259c7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ludwig/modules/metric_modules.py b/ludwig/modules/metric_modules.py
@@ -136,7 +136,7 @@ class RMSEMetric(MeanSquaredError, LudwigMetric):
     """Root mean squared error metric."""
 
     def __init__(self, **kwargs):
-        super().__init__(squared=False, **kwargs)
+        super().__init__(squared=False)
 
 
 @register_metric(PRECISION, [BINARY], MAXIMIZE, PROBABILITIES)

diff --git a/requirements.txt b/requirements.txt
@@ -26,7 +26,7 @@ marshmallow-jsonschema
 marshmallow-dataclass==8.5.4
 tensorboard
 nltk    # Required for rouge scores.
-torchmetrics>=0.11.0,<=0.11.4
+torchmetrics>=0.11.0
 torchinfo
 filelock
 psutil==5.9.4