Try dynamic typing

vendiscore.py

@@ -86,16 +86,19 @@ def get_features(config_name):
     if config_name in ("text", "default"):
         return datasets.Features({"samples": datasets.Value("string")})
     if config_name == "image":
-        datasets.Features({"samples": datasets.Image})
+        return datasets.Features({"samples": datasets.Image})
+    if config_name in ("K", "X"):
+        return [
+            datasets.Features(
+                {"samples": datasets.Sequence(datasets.Value("int32"))}
+            ),
+            datasets.Features(
+                {"samples": datasets.Sequence(datasets.Value("float"))}
+            ),
+        ]
     return [
         datasets.Features({"samples": datasets.Value("int32")}),
-        datasets.Features(
-            {"samples": datasets.Sequence(datasets.Value("int32"))}
-        ),
         datasets.Features({"samples": datasets.Value("float")}),
-        datasets.Features(
-            {"samples": datasets.Sequence(datasets.Value("float"))}
-        ),
         datasets.Features({"samples": datasets.Array2D}),
     ]