Fastai v2 "TTA" is not working in google Colab

I am trying to use fastai v2 tta in google colab. I used the following command:

predictions, targets = learn.tta(dl=dl_test)

And I get the following error message:

AttributeError: ‘NBProgressBar’ object has no attribute ‘wait_for’

Any suggestions?
Thank you,

Hi hagarciapinto I hope you are having a wonderful day!

Maybe the posts in this thread can help you solve your error.

This one looks like it may do it.

Cheers mrfabuluous1 :smiley: :smile:

Thank you @mrfabulous1 , much appreciated.