minor fix to imports

Signed-off-by: Sukriti-Sharma4 <[email protected]>
foundation-model-stack · Jun 30, 2024 · 8fe0842 · 8fe0842
1 parent dbaa919
commit 8fe0842
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tuning/utils/preprocessing_utils.py b/tuning/utils/preprocessing_utils.py
@@ -18,10 +18,8 @@
 # Third Party
 from datasets import Dataset
 from transformers import AutoTokenizer, DataCollatorForSeq2Seq
-import datasets
-
-# First Party
 from trl import DataCollatorForCompletionOnlyLM
+import datasets
 
 # Local
 from tuning.config import configs