danf0 commited on
Commit
607b435
1 Parent(s): 41e5c28

Try dynamic value type again

Browse files
Files changed (1) hide show
  1. vendiscore.py +2 -1
vendiscore.py CHANGED
@@ -85,6 +85,8 @@ def get_dtype(config_name):
85
  def get_features(config_name):
86
  if config_name in ("text", "default"):
87
  return datasets.Features({"samples": datasets.Value("string")})
 
 
88
  return [
89
  datasets.Features({"samples": datasets.Value("int32")}),
90
  datasets.Features(
@@ -94,7 +96,6 @@ def get_features(config_name):
94
  datasets.Features(
95
  {"samples": datasets.Sequence(datasets.Value("float"))}
96
  ),
97
- datasets.Features({"samples": datasets.Image}),
98
  datasets.Features({"samples": datasets.Array2D}),
99
  ]
100
 
 
85
  def get_features(config_name):
86
  if config_name in ("text", "default"):
87
  return datasets.Features({"samples": datasets.Value("string")})
88
+ if config_name == "image":
89
+ datasets.Features({"samples": datasets.Image})
90
  return [
91
  datasets.Features({"samples": datasets.Value("int32")}),
92
  datasets.Features(
 
96
  datasets.Features(
97
  {"samples": datasets.Sequence(datasets.Value("float"))}
98
  ),
 
99
  datasets.Features({"samples": datasets.Array2D}),
100
  ]
101