Why is there a "Warning: Slow Fallback implementation" and a DimensionMismatch error?

Your train_X and test_X have element type Any. Choose something concrete.