From 6f112cde9d0f23ddd68e1830919635e1be598725 Mon Sep 17 00:00:00 2001 From: helloyongyang Date: Thu, 14 May 2026 08:51:54 +0000 Subject: [PATCH] fix code format --- llmc/data/dataset/base_dataset.py | 3 +-- llmc/eval/eval_base.py | 3 +-- llmc/eval/eval_ppl.py | 3 +-- tools/download_calib_dataset.py | 3 +-- tools/download_eval_dataset.py | 3 +-- 5 files changed, 5 insertions(+), 10 deletions(-) diff --git a/llmc/data/dataset/base_dataset.py b/llmc/data/dataset/base_dataset.py index 85260e83d..206c1cf1d 100755 --- a/llmc/data/dataset/base_dataset.py +++ b/llmc/data/dataset/base_dataset.py @@ -3,12 +3,11 @@ from abc import ABCMeta import torch +from datasets import load_dataset, load_from_disk from loguru import logger from PIL import Image from torch.nn import functional as F -from datasets import load_dataset, load_from_disk - from .specified_preproc import PREPROC_REGISTRY diff --git a/llmc/eval/eval_base.py b/llmc/eval/eval_base.py index 098c9bb8f..60a60589e 100755 --- a/llmc/eval/eval_base.py +++ b/llmc/eval/eval_base.py @@ -5,11 +5,10 @@ import torch import torch.nn as nn +from datasets import load_dataset, load_from_disk from human_eval.data import read_problems from loguru import logger -from datasets import load_dataset, load_from_disk - class BaseEval: def __init__(self, model, config): diff --git a/llmc/eval/eval_ppl.py b/llmc/eval/eval_ppl.py index bb41329f1..d598218c5 100644 --- a/llmc/eval/eval_ppl.py +++ b/llmc/eval/eval_ppl.py @@ -3,11 +3,10 @@ import torch import torch.nn as nn +from datasets import load_dataset, load_from_disk from loguru import logger from tqdm import tqdm -from datasets import load_dataset, load_from_disk - from .eval_base import BaseEval diff --git a/tools/download_calib_dataset.py b/tools/download_calib_dataset.py index 31fe14772..37ce76bab 100644 --- a/tools/download_calib_dataset.py +++ b/tools/download_calib_dataset.py @@ -6,9 +6,8 @@ import argparse import os -from loguru import logger - from datasets import load_dataset +from loguru import logger def download(calib_dataset_name, path): diff --git a/tools/download_eval_dataset.py b/tools/download_eval_dataset.py index 12f1f2a6a..7eddd8bd0 100644 --- a/tools/download_eval_dataset.py +++ b/tools/download_eval_dataset.py @@ -6,9 +6,8 @@ import argparse import os -from loguru import logger - from datasets import load_dataset +from loguru import logger def download(calib_dataset_name, path):