danf0 commited on
Commit
3ae673b
1 Parent(s): d9ea262

Try float before int

Browse files
Files changed (1) hide show
  1. vendiscore.py +3 -3
vendiscore.py CHANGED
@@ -90,15 +90,15 @@ def get_features(config_name):
90
  if config_name in ("K", "X"):
91
  return [
92
  datasets.Features(
93
- {"samples": datasets.Sequence(datasets.Value("int32"))}
94
  ),
95
  datasets.Features(
96
- {"samples": datasets.Sequence(datasets.Value("float"))}
97
  ),
98
  ]
99
  return [
100
- datasets.Features({"samples": datasets.Value("int32")}),
101
  datasets.Features({"samples": datasets.Value("float")}),
 
102
  datasets.Features({"samples": datasets.Array2D}),
103
  ]
104
 
 
90
  if config_name in ("K", "X"):
91
  return [
92
  datasets.Features(
93
+ {"samples": datasets.Sequence(datasets.Value("float"))}
94
  ),
95
  datasets.Features(
96
+ {"samples": datasets.Sequence(datasets.Value("int32"))}
97
  ),
98
  ]
99
  return [
 
100
  datasets.Features({"samples": datasets.Value("float")}),
101
+ datasets.Features({"samples": datasets.Value("int32")}),
102
  datasets.Features({"samples": datasets.Array2D}),
103
  ]
104