diff --git a/task.py b/task.py
index d062253eeee5e38db3388d5879da7064d3afee4b..abb26c3fb7ca74950dc4e670f920c29bbcd6d8e6 100644
--- a/task.py
+++ b/task.py
@@ -5,15 +5,13 @@ from functools import lru_cache
 from statistics import mean
 from typing import Union
 
+from cli import argument_parser
 from datasets import Dataset, load_dataset
 from evaluate import load as load_metric
 from llama_cpp import LlamaGrammar, deque
-from torch.utils import data
-from tqdm import tqdm
-
-from cli import argument_parser
 from models import Llama2, LLMModel, OpenAI
 from opt_types import ModelUsage
+from tqdm import tqdm
 from utils import log_calls, logger
 
 SYSTEM_MESSAGE = """