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

np.ndarray -> 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(np.ndarray(predictions), np.ndarray(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.array(predictions), np.array(references))
94
  psnrs = list(map(
95
  lambda args: peak_signal_noise_ratio(*args, data_range=data_range), samples
96
  ))