mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
34561ececb
* add InfiniteBench * add InfiniteBench --------- Co-authored-by: wangchonghua <wangchonghua@pjlab.org.cn>
20 lines
471 B
Python
20 lines
471 B
Python
import json
|
|
import re
|
|
|
|
from opencompass.registry import TEXT_POSTPROCESSORS
|
|
|
|
|
|
def iter_jsonl(path):
|
|
with open(path, 'r') as f:
|
|
for line in f:
|
|
yield json.loads(line)
|
|
|
|
|
|
@TEXT_POSTPROCESSORS.register_module()
|
|
def InfiniteBench_first_number_postprocess(text: str) -> str:
|
|
first_number = re.search(r'\d+\.\d+|\d+', text)
|
|
if first_number is None:
|
|
return None
|
|
first_number = first_number.group(0).strip()
|
|
return str(first_number)
|