src/tuning/__init__.py

"""Tuning subpackage for Précis."""

from src.tuning.lora import get_lora_config, apply_lora, merge_and_save
from src.tuning.data import SummarizationDataset, prepare_dataset, create_dummy_data
from src.tuning.trainer import PrecisTrainer

__all__ = [
    "get_lora_config",
    "apply_lora",
    "merge_and_save",
    "SummarizationDataset",
    "prepare_dataset",
    "create_dummy_data",
    "PrecisTrainer",
]