specify arg in datasets.Array2D

peak_signal_to_noise_ratio.py

@@ -58,8 +58,8 @@ class StructuralSimilarityIndexMeasure(evaluate.Metric):
             citation=_CITATION,
             inputs_description=_KWARGS_DESCRIPTION,
             features=datasets.Features({
-                "predictions": datasets.Sequence(datasets.Array2D("float32")),
-                "references": datasets.Sequence(datasets.Array2D("float32")),
+                "predictions": datasets.Sequence(datasets.Array2D(dtype="float32")),
+                "references": datasets.Sequence(datasets.Array2D(dtype="float32")),
             }),
             reference_urls=["https://en.wikipedia.org/wiki/Peak_signal-to-noise_ratio"],
         )