gen

src/autotrain/datagen/clients.py

@@ -1,14 +1,14 @@
+import json
 import time
 from dataclasses import dataclass
 from typing import Optional
 
+import outlines
+import torch
+import transformers
 from huggingface_hub import InferenceClient
 
 from autotrain import logger
-import transformers
-import torch
-import outlines
-import json
 
 
 @dataclass

src/autotrain/datagen/text.py

@@ -1,4 +1,5 @@
 import hashlib
+import os
 import random
 import time
 
@@ -9,7 +10,6 @@
 from autotrain.datagen.clients import Client
 from autotrain.datagen.params import AutoTrainGenParams
 
-import os
 
 TEXT_CLASSIFICATION_SYSTEM_PROMPT = """
 You are an AI bot that generates data for text classification tasks.

src/autotrain/datagen/utils.py

@@ -3,7 +3,7 @@
 import subprocess
 from typing import Dict, List, Optional
 
-from datasets import Dataset, DatasetDict, ClassLabel
+from datasets import ClassLabel, Dataset, DatasetDict
 from huggingface_hub import HfApi, metadata_update
 
 from autotrain import logger

src/autotrain/trainers/text_classification/__main__.py

@@ -15,6 +15,8 @@
 from transformers.trainer_callback import PrinterCallback
 
 from autotrain import logger
+from autotrain.datagen.gen import AutoTrainGen
+from autotrain.datagen.params import AutoTrainGenParams
 from autotrain.trainers.common import (
     ALLOW_REMOTE_CODE,
     LossLoggingCallback,
@@ -28,8 +30,6 @@
 from autotrain.trainers.text_classification import utils
 from autotrain.trainers.text_classification.dataset import TextClassificationDataset
 from autotrain.trainers.text_classification.params import TextClassificationParams
-from autotrain.datagen.params import AutoTrainGenParams
-from autotrain.datagen.gen import AutoTrainGen
 
 
 def parse_args():