Kernel hangs after itkwidgets visualization (only on VSCode)

[mgenet]

I have some notebooks that use itkwidgets (0.32, as 1.0 does not support unstructured grids yet) for visualization of vtk objects. They run fine in the jupyter notebook interface, but the kernel seems to hang after a few seconds when the notebook runs within VSCode. I made a MWE, it is attached. (Also attaching the conda environment file I use to run the notebooks.) If you run all, it works fine, but after a few second you can see that you cannot run a cell, the notebook hangs. Any help would be greatly appreciated, especially as I using these notebooks in a class and some students are affected. Many thanks in advance!

PS: Somehow I cannot attach my notebook. Here it is: https://www.dropbox.com/scl/fi/89oiq0zho876evlekpkfy/test_itkwidgets_in_vscode.ipynb?rlkey=0b0964h6t137gu9uoq0gl2ybq&dl=0

And the conda env file: https://www.dropbox.com/scl/fi/d44g6o08har98qyzmc4ro/environment.yml?rlkey=0s6g74cxq05mnhocg3h0kevvz&dl=0