Added support for ML filling on dataset_collapse