cannot import name 'Mapping' from 'collections'

I'm getting this error when calling:

from bertviz.transformers_neuron_view import BertModel, BertTokenizer

With Python 3.11. The Mapping location has been updated but this is not reflected here.