versae commited on
Commit
b5a132d
·
1 Parent(s): d3d8955

Fix --filter

Browse files
Files changed (1) hide show
  1. eval.py +1 -1
eval.py CHANGED
@@ -130,7 +130,7 @@ def main(args):
130
  if args.filter:
131
  attribute, value = list(map(str.strip, args.filter.split(":")))
132
  dataset = dataset.filter(
133
- lambda x: x[attribute == value],
134
  desc=f"Filtering on {args.filter}",
135
  )
136
  # for testing: only process the first two examples as a test
 
130
  if args.filter:
131
  attribute, value = list(map(str.strip, args.filter.split(":")))
132
  dataset = dataset.filter(
133
+ lambda x: x[attribute] == value,
134
  desc=f"Filtering on {args.filter}",
135
  )
136
  # for testing: only process the first two examples as a test