Can't use learn.get_preds() OR learn.TTA() with contrastive loss

Using the following pipeline, https://colab.research.google.com/drive/1G-PXfzBa2UPNymt_Hi5QQBc8l4TXBkvX, I am unable to run learn.get_preds() or learn.TTA(). I get the following error:

TypeError: expected Tensor as element 0 in argument 0, but got list