mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
91fba2c2e9
* [Feat] support pass@ k * [Feat] support pass@k * [Feat] support pass@k * [Feat] support pass@k * [Feat] support pass@k * [Feat] support pass@k docs * update naming --------- Co-authored-by: Leymore <zfz-960727@163.com>
15 lines
361 B
Python
15 lines
361 B
Python
from itertools import islice
|
|
|
|
try:
|
|
# batched is in 3.12
|
|
from itertools import batched
|
|
except ImportError:
|
|
|
|
def batched(iterable, n):
|
|
# batched('ABCDEFG', 3) --> ABC DEF G
|
|
if n < 1:
|
|
raise ValueError('n must be at least one')
|
|
it = iter(iterable)
|
|
while batch := tuple(islice(it, n)):
|
|
yield batch
|