Update eval.py
Browse files
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
|