Try dynamic value type

vendiscore.py

@@ -74,12 +74,11 @@ def get_dtype(config_name):
         return datasets.Features({"samples": datasets.Value("string")})
     if config_name == "image":
         return datasets.Features({"samples": datasets.Image})
-    elif config_name in ("X", "K"):
+    if config_name in ("X", "K"):
         return datasets.Array2D
-    elif config_name == "default":
+    if config_name == "default":
         return datasets.Features({"samples": datasets.Value("string")})
-    else:
-        return datasets.Features({"samples": datasets.Value(config_name)})
+    return datasets.Features({"samples": datasets.Value(config_name)})
 
 
 @evaluate.utils.file_utils.add_start_docstrings(
@@ -96,11 +95,7 @@ class VendiScore(evaluate.Metric):
             description=_DESCRIPTION,
             citation=_CITATION,
             inputs_description=_KWARGS_DESCRIPTION,
-            features=datasets.Features(
-                {
-                    "samples": get_dtype(self.config_name),
-                }
-            ),
+            features=get_dtype(self.config_name),
             homepage="http://github.com/Vertaix/Vendi-Score",
             codebase_urls=["http://github.com/Vertaix/Vendi-Score"],
             reference_urls=[],