match data types for shuffle split