From a7070ba2d23ad560d66c896a06e805191f67a078 Mon Sep 17 00:00:00 2001 From: Dongsheng Zhu <2532956974@qq.com> Date: Tue, 4 Mar 2025 09:16:02 +0000 Subject: [PATCH] lint --- opencompass/datasets/livecodebench/livecodebench.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/opencompass/datasets/livecodebench/livecodebench.py b/opencompass/datasets/livecodebench/livecodebench.py index fd7eee8e..20bd2e0d 100644 --- a/opencompass/datasets/livecodebench/livecodebench.py +++ b/opencompass/datasets/livecodebench/livecodebench.py @@ -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})