Skip to content

GitLab

"profiler/vscode:/vscode.git/clone" did not exist on "74284dd29481283257b898026741e03d4cdd7ac8"
Commit 569f9e6e authored by Baber's avatar Baber
Browse files

nit

parent 6508efaa
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 4 deletions
lm_eval/evaluator_utils.py
View file @ 569f9e6e
...@@ -209,10 +209,12 @@ def get_sample_size( ...@@ -209,10 +209,12 @@ def get_sample_size(
"Limit is 1.0, adjusting the sample size to be a multiple of the batch size" "Limit is 1.0, adjusting the sample size to be a multiple of the batch size"
) )
return (len(task.eval_docs) // batch_size) * batch_size return (len(task.eval_docs) // batch_size) * batch_size
elif limit is not None: else:
limit = ( limit = (
int(math.ceil(len(task.eval_docs) * limit)) if limit < 1.0 else int(limit) int(math.ceil(len(task.eval_docs) * limit))
) if limit < 1.0
else int(limit)
)
return limit return limit
......
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