update dtype

cosine_similarity.py

@@ -71,8 +71,8 @@ class cosine_similarity(evaluate.Metric):
             inputs_description=_KWARGS_DESCRIPTION,
             # This defines the format of each prediction and reference
             features=datasets.Features({
-                'predictions': datasets.Value('float'),
-                'references': datasets.Value('float'),
+                'predictions': Array1D(shape=(1, ), dtype='float'),
+                'references': Array1D(shape=(1, ), dtype='float'),
             }),
             # Homepage of the module for documentation
             homepage="http://module.homepage",