Error type: ArrowInvalid Details: Failed to parse string: '[254,254]' as a scalar of type int32

cc @abhishek as this seems to be related to AutoTrain.