danf0 commited on
Commit
d10ec79
·
1 Parent(s): 3258b73

Process images as numpy arrays

Browse files
Files changed (1) hide show
  1. vendiscore.py +4 -1
vendiscore.py CHANGED
@@ -75,7 +75,10 @@ def get_features(config_name):
75
  if config_name in ("text", "default"):
76
  return datasets.Features({"samples": datasets.Value("string")})
77
  if config_name == "image":
78
- return datasets.Features({"samples": datasets.Array3D})
 
 
 
79
  if config_name in ("K", "X"):
80
  return [
81
  datasets.Features(
 
75
  if config_name in ("text", "default"):
76
  return datasets.Features({"samples": datasets.Value("string")})
77
  if config_name == "image":
78
+ return [
79
+ datasets.Features({"samples": datasets.Array2D}),
80
+ datasets.Features({"samples": datasets.Array3D}),
81
+ ]
82
  if config_name in ("K", "X"):
83
  return [
84
  datasets.Features(