Fix import formatting issues

Author: codewithdark-git

.github/workflows/docs.yml

@@ -1,23 +1,12 @@
-from .data import LoadDataset, DatasetPreprocessor, DatasetSplitter, DataLoader
-from .trainer import FineTuningTrainer, ModelEvaluator
-from .hub import HubManager, CheckpointManager
-from .utils import (
-    get_optimal_training_settings,
-    configure_logging,
-    enable_logging,
-    QuantizationBenchmark,
-)
 from .api import QuantLLM
-
-from .quant import (
-    QuantizationConfig,
-    QuantizationEngine,
-    QuantizedLinear,
-    GGUFQuantizer,
-)
-
-
-from .config import ModelConfig, DatasetConfig, TrainingConfig
+from .config import DatasetConfig, ModelConfig, TrainingConfig
+from .data import DataLoader, DatasetPreprocessor, DatasetSplitter, LoadDataset
+from .hub import CheckpointManager, HubManager
+from .quant import (GGUFQuantizer, QuantizationConfig, QuantizationEngine,
+                    QuantizedLinear)
+from .trainer import FineTuningTrainer, ModelEvaluator
+from .utils import (QuantizationBenchmark, configure_logging, enable_logging,
+                    get_optimal_training_settings)
 
 # Configure package-wide logging
 configure_logging()

quantllm/__init__.py

@@ -1,9 +1,12 @@
-from typing import Optional, Dict, Any, Union, Tuple
+from typing import Any, Dict, Optional, Tuple, Union
+
 import torch
-from transformers import PreTrainedModel, AutoTokenizer
-from ..quant.gguf import GGUFQuantizer, SUPPORTED_GGUF_BITS, SUPPORTED_GGUF_TYPES
-from ..utils.logger import logger
+from transformers import AutoTokenizer, PreTrainedModel
+
+from ..quant.gguf import (SUPPORTED_GGUF_BITS, SUPPORTED_GGUF_TYPES,
+                          GGUFQuantizer)
 from ..utils.benchmark import QuantizationBenchmark
+from ..utils.logger import logger
 
 
 class QuantLLM:

quantllm/api/high_level.py

@@ -1,6 +1,6 @@
 """Command-line interface for QuantLLM package."""
 
-from .commands import train, evaluate, quantize, serve
+from .commands import evaluate, quantize, serve, train
 from .parser import create_parser
 
 __all__ = ["main", "train", "evaluate", "quantize", "serve"]

quantllm/cli/__init__.py

@@ -1,9 +1,10 @@
 from argparse import Namespace
-from ..models import ModelLoader, QuantConfig
+
+from ..config import DatasetConfig, ModelConfig, TrainingConfig
 from ..data import DatasetLoader
-from ..training import FineTuningTrainer, ModelEvaluator
-from ..config import ModelConfig, TrainingConfig, DatasetConfig
+from ..models import ModelLoader, QuantConfig
 from ..runtime import DeviceManager
+from ..training import FineTuningTrainer, ModelEvaluator
 from ..utils.logger import logger
 
 

quantllm/cli/commands.py

@@ -1,6 +1,7 @@
 import argparse
 from typing import Any
-from .commands import train, evaluate, quantize, serve
+
+from .commands import evaluate, quantize, serve, train
 
 
 def create_parser() -> argparse.ArgumentParser:

quantllm/cli/parser.py

@@ -1,6 +1,6 @@
 from .config_manager import ConfigManager
-from .training_config import TrainingConfig
-from .model_config import ModelConfig
 from .dataset_config import DatasetConfig
+from .model_config import ModelConfig
+from .training_config import TrainingConfig
 
 __all__ = ["ConfigManager", "TrainingConfig", "ModelConfig", "DatasetConfig"]

quantllm/config/__init__.py

@@ -1,7 +1,9 @@
-import yaml
 import json
-from typing import Dict, Any, Optional
 from pathlib import Path
+from typing import Any, Dict, Optional
+
+import yaml
+
 from ..utils.logger import logger
 
 

quantllm/config/config_manager.py

@@ -1,6 +1,6 @@
 from dataclasses import dataclass
-from typing import Optional, Dict, Any, List, Union
 from pathlib import Path
+from typing import Any, Dict, List, Optional, Union
 
 
 @dataclass

quantllm/config/dataset_config.py

@@ -1,6 +1,6 @@
 from dataclasses import dataclass
-from typing import Optional, Dict, Any, List
 from pathlib import Path
+from typing import Any, Dict, List, Optional
 
 
 @dataclass