from hydra.core.config_store import ConfigStore

from trainer.datasets.step_sana_hf_dataset import StepSanaHFDatasetConfig

cs = ConfigStore.instance()
cs.store(group="dataset", name="step_sana", node=StepSanaHFDatasetConfig)