from src.settings.cherry_pick import ChatCherryPickSettings
from src.settings.datasets.chat import ChatMultiDatasetSettings
from src.settings.pipelines.train.base import BaseTrainExperimentSettings


class SftTrainExperimentSettings(BaseTrainExperimentSettings):
    train_dataset_settings: ChatMultiDatasetSettings
    val_dataset_settings: ChatMultiDatasetSettings

    cherry_pick_settings: ChatCherryPickSettings | None = None
