If you're running this in a Jupyter Notebook, then you need to install the extensions.
See the following from the repo's readme:
Jupyter Notebook Support
For use in the Jupyter Notebook, install the notebook and ipywidgets
packages using pip...
pip install "notebook>=5.3" "ipywidgets==7.5"
or conda.
conda install "notebook>=5.3" "ipywidgets=7.5"
JupyterLab Support (Python 3.5+)
For use in JupyterLab, install the jupyterlab and ipywidgets
packages using pip...
pip install jupyterlab "ipywidgets==7.5"
or conda.
conda install jupyterlab "ipywidgets=7.5"
Then run the following commands to install the required JupyterLab extensions (note that this will require node to be installed):
# Basic JupyterLab renderer support
jupyter labextension install [email protected]
# OPTIONAL: Jupyter widgets extension for FigureWidget support
jupyter labextension install @jupyter-widgets/jupyterlab-manager [email protected]
Please check out our Troubleshooting guide if you run into any problems with JupyterLab.