a

my_metricv2.py

@@ -72,29 +72,12 @@ class MyMetricv2(evaluate.Metric):
             inputs_description=_KWARGS_DESCRIPTION,
             # This defines the format of each prediction and reference
             features=datasets.Features({
-                "predictions": [
-                    datasets.Features(
-                        {
-                            "scores": datasets.Sequence(datasets.Value("float")),
-                            "ids": datasets.Sequence(datasets.Value("int64")),
-                            "frames": datasets.Sequence(datasets.Value("int64")),
-                            "boxes": datasets.Sequence(
+                "predictions": datasets.Sequence(
                                 datasets.Sequence(datasets.Value("float"))
                             ),
-                        }
-                    )
-                ],
-                "references": [
-                    datasets.Features(
-                        {
-                            "ids": datasets.Sequence(datasets.Value("int64")),
-                            "frames": datasets.Sequence(datasets.Value("int64")),
-                            "boxes": datasets.Sequence(
+                "references": datasets.Sequence(
                                 datasets.Sequence(datasets.Value("float"))
-                            ),
-                        }
-                    )
-                ],
+                            )
             }),
             # Homepage of the module for documentation
             homepage="http://module.homepage",