update args

peak_signal_to_noise_ratio.py

@@ -91,8 +91,8 @@ class PeakSignaltoNoiseRatio(evaluate.Metric):
         sample_weight=None,
     ) -> Dict[str, float]:
         samples = zip(predictions, references)
-        psnrs = map(lambda args: peak_signal_noise_ratio(*args, data_range), samples)
-        return {
-            "Peak Signal-to-Noise Ratio": np.average(list(psnrs), weights=sample_weight)
-        }
+        psnrs = list(map(
+            lambda args: peak_signal_noise_ratio(*args, data_range=data_range), samples
+        ))
+        return {"Peak Signal-to-Noise Ratio": np.average(psnrs, weights=sample_weight)}