fix flake

Lightning-AI · Jan 10, 2022 · e0b25c3 · e0b25c3
1 parent b88572b
commit e0b25c3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/tests/text/test_eed.py b/tests/text/test_eed.py
@@ -14,16 +14,11 @@
 
 from functools import partial
 
-import numpy as np
 import pytest
 from torch import Tensor, tensor
 
 from tests.text.helpers import TextTester
-from tests.text.inputs import (
-    _inputs_multiple_references,
-    _inputs_single_reference,
-    _inputs_single_sentence_multiple_references,
-)
+from tests.text.inputs import _inputs_single_reference, _inputs_single_sentence_multiple_references
 from torchmetrics.functional.text.eed import extended_edit_distance
 from torchmetrics.text.eed import ExtendedEditDistance
 

diff --git a/torchmetrics/image/__init__.py b/torchmetrics/image/__init__.py
@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from torchmetrics.image.psnr import PSNR  # noqa: F401
-from torchmetrics.image.ssim import SSIM  # noqa: F401
+from torchmetrics.image.ssim import SSIM, MultiScaleStructuralSimilarityIndexMeasure  # noqa: F401