Pytorch + BERT+ batch_encode_plus() Code running fine in Colab but producing problems with Kaggle in mismatch input shapes

Question

I tried to use a Google Colab initialised Notebook for Kaggle and found a strange behaviour as it gave me something like:

 16   # text2tensor
---> 17   tra