TypeError: Couldn't cast array of type int64 while mapping the dataset

This is not what I meant. What I meant is that the value of the labels key in the features dictionary should be Value("int64") (not the current Sequence(Value("int64")) ).