jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
Dongsheng Zhu 2025-03-04 09:16:02 +00:00
parent 63c7970937
commit a7070ba2d2
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
opencompass/datasets/livecodebench/livecodebench.py
View File

@ -6,8 +6,8 @@ import json
import pickle
import zlib
from dataclasses import dataclass
from enum import Enum
from datetime import datetime
from enum import Enum
from datasets import DatasetDict, load_dataset, load_from_disk
@ -111,13 +111,14 @@ class LCBCodeGenerationDataset(BaseDataset):
dataset = dataset.map(transform)
if start_date is not None:
p_start_date = datetime.strptime(start_date, "%Y-%m-%d")
p_start_date = datetime.strptime(start_date, '%Y-%m-%d')
dataset = dataset.filter(
lambda e: p_start_date <= datetime.fromisoformat(e['contest_date'])) # noqa: E501
lambda e: p_start_date <= datetime.fromisoformat(e[
'contest_date'])) # noqa: E501
if end_date is not None:
p_end_date = datetime.strptime(end_date, "%Y-%m-%d")
dataset = dataset.filter(
lambda e: datetime.fromisoformat(e['contest_date']) <= p_end_date) # noqa: E501
p_end_date = datetime.strptime(end_date, '%Y-%m-%d')
dataset = dataset.filter(lambda e: datetime.fromisoformat(e[
'contest_date']) <= p_end_date) # noqa: E501
return DatasetDict({'test': dataset, 'train': dataset})