Issue with TTA in fastai v2

I’m getting this error when I try learn.tta():
‘NBProgressBar’ object has no attribute ‘wait_for’

It seems like a bug, because there is nothing special about my learner and I’m just doing what the book suggested.
Any idea?