jpxkqx commited on
Commit
df0ed01
·
1 Parent(s): 5efd316

convert list of list to np.array

Browse files
Files changed (1) hide show
  1. peak_signal_to_noise_ratio.py +1 -1
peak_signal_to_noise_ratio.py CHANGED
@@ -90,7 +90,7 @@ class PeakSignaltoNoiseRatio(evaluate.Metric):
90
  data_range: Optional[float] = None,
91
  sample_weight=None,
92
  ) -> Dict[str, float]:
93
- samples = zip(predictions, references)
94
  psnrs = list(map(
95
  lambda args: peak_signal_noise_ratio(*args, data_range=data_range), samples
96
  ))
 
90
  data_range: Optional[float] = None,
91
  sample_weight=None,
92
  ) -> Dict[str, float]:
93
+ samples = zip(np.ndarray(predictions), np.ndarray(references))
94
  psnrs = list(map(
95
  lambda args: peak_signal_noise_ratio(*args, data_range=data_range), samples
96
  ))