Commit 569f9e6e authored by Baber's avatar Baber
Browse files

nit

parent 6508efaa
...@@ -209,9 +209,11 @@ def get_sample_size( ...@@ -209,9 +209,11 @@ 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