Have scale_data return generators instead of lists