fix code format

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
 
 

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):

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
 
 

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):

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):