versae commited on
Commit
6e3e8e8
1 Parent(s): eae9a48

Update eval.py

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