Unverified Commit 37fab200 authored by Stella Biderman's avatar Stella Biderman Committed by GitHub
Browse files

Changed arg name to match master

parent 760788d5
......@@ -16,7 +16,7 @@ def parse_args():
parser.add_argument('--num_fewshot', type=int, default=1)
parser.add_argument('--seed', type=int, default=1234)
parser.add_argument('--output_path', default=None)
parser.add_argument('--truncate', default=None)
parser.add_argument('--limit', default=None)
return parser.parse_args()
......@@ -36,7 +36,7 @@ def main():
if not task.has_validation_docs():
continue
result = task.evaluate(
docs=itertools.isslice(task.validation_docs(), stop=args.truncate),
docs=itertools.isslice(task.validation_docs(), stop=args.limit),
lm=lm,
provide_description=args.provide_description,
num_fewshot=args.num_fewshot,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment