resolve torchmetrics import issue (#1015)

unit8co · Jun 18, 2022 · c8169ec · c8169ec
1 parent 14d671a
commit c8169ec
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/darts/tests/models/forecasting/test_torch_forecasting_model.py b/darts/tests/models/forecasting/test_torch_forecasting_model.py
@@ -4,11 +4,6 @@
 from unittest.mock import patch
 
 import pandas as pd
-from torchmetrics import (
-    MeanAbsoluteError,
-    MeanAbsolutePercentageError,
-    MetricCollection,
-)
 
 from darts import TimeSeries
 from darts.logging import get_logger
@@ -19,6 +14,11 @@
 
 try:
     import torch
+    from torchmetrics import (
+        MeanAbsoluteError,
+        MeanAbsolutePercentageError,
+        MetricCollection,
+    )
 
     from darts.models.forecasting.rnn_model import RNNModel
 

diff --git a/requirements/torch.txt b/requirements/torch.txt
@@ -1,3 +1,2 @@
 pytorch-lightning>=1.5.0
 torch>=1.8.0
-torchmetrics>=0.9.1