There is an additional step to have it properly working in jupyter lab: you must have the jupyter widgets extension properly installed, which I think is done with:
jupyter labextension install @jupyter-widgets/jupyterlab-manager
Let me know if it solves your issue.