change float to int32

triplet_margin_loss.py

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