register template override for dataset_input.pt