Fix --filter
Browse files
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
|