tqdm in jupyter doesn't work correctly

edit

I've run the following to install tqdm:

sage --pip install --user tqdm
sage --pip install ipywidgets
sage --jupyter nbextension enable --py widgetsnbextension
sage --pip install --upgrade ipywidgets
sage --pip install --upgrade widgetsnbextension
sage --pip install --upgrade jupyterlab
sudo sage --jupyter nbextension enable --py widgetsnbextension --sys-prefix

and tested it with the following cell code:

from sage.all import Integer
from tqdm.notebook import tqdm

SAMPLE_COUNT = 100r  # replace with your actual sample count

with tqdm(total=int(SAMPLE_COUNT)) as progress:
    for i in range(SAMPLE_COUNT):
        progress.update(1r)
        sleep(0.1r)

it seems to load fine, but the progressbar never updates, and it also looks to be in text form, instead of the widget with pretty colours?