revise unit test of ms-ssim

open-mmlab · Jan 18, 2023 · 6c17aa5 · 6c17aa5
1 parent 59f8433
commit 6c17aa5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_metrics/test_ms_ssim.py b/tests/test_metrics/test_ms_ssim.py
@@ -37,7 +37,7 @@ def test_ms_ssim(init_kwargs, inputs, results):
     ms_ssim = MS_SSIM(**init_kwargs)
     ms_ssim_results = ms_ssim(inputs)
     np.testing.assert_allclose(
-        ms_ssim_results['ms_ssim'], results)
+        ms_ssim_results['ms-ssim'], results)
 
 
 def test_raise_error():