remove id

triplet_margin_loss.py

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