prepare the dataset for cli worker