int32 to float

triplet_margin_loss.py

@@ -77,9 +77,9 @@ class TripletMarginLoss(evaluate.EvaluationModule):
             inputs_description=_KWARGS_DESCRIPTION,
             features=datasets.Features(
                 {
-                    "anchor": datasets.Sequence(datasets.Value("int32")),
-                    "positive": datasets.Sequence(datasets.Value("int32")),
-                    "negative": datasets.Sequence(datasets.Value("int32"))
+                    "anchor": datasets.Sequence(datasets.Value("float")),
+                    "positive": datasets.Sequence(datasets.Value("float")),
+                    "negative": datasets.Sequence(datasets.Value("float"))
                 }
             ),
             reference_urls=["https://proceedings.neurips.cc/paper/2003/hash/d3b1fb02964aa64e257f9f26a31f72cf-Abstract.html"],