from hydra.core.config_store import ConfigStore

from trainer.tasks.step_flux_task import StepFluxTaskConfig

cs = ConfigStore.instance()
cs.store(group="task", name="step_flux", node=StepFluxTaskConfig)