fix

Lightning-AI · Apr 20, 2021 · bb78a9d · bb78a9d
1 parent 9ba02e7
commit bb78a9d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/bases/test_metric.py b/tests/bases/test_metric.py
@@ -22,7 +22,7 @@
 
 from tests.helpers import seed_all
 from tests.helpers.testers import DummyListMetric, DummyMetric, DummyMetricSum
-from torchmetrics.utilities.imports import _LIGHTNING_GREATER_THAN_1_3, _TORCH_LOWER_1_6
+from torchmetrics.utilities.imports import _LIGHTNING_AVAILABLE, _LIGHTNING_GREATER_EQUAL_1_3, _TORCH_LOWER_1_6
 
 seed_all(42)
 
@@ -101,7 +101,7 @@ def test_reset_compute():
     a.update(tensor(5))
     assert a.compute() == 5
     a.reset()
-    if _LIGHTNING_GREATER_THAN_1_3:
+    if not _LIGHTNING_AVAILABLE or _LIGHTNING_GREATER_EQUAL_1_3:
         assert a.compute() == 0
     else:
         assert a.compute() == 5