Thanks @Adhe11. Unfortunately this seems to be an issue with VS Code. I am also attempting to use Ipywidget’s @interact, used in Lesson 3, and facing a similar issue. VS Code has a known issue with that library Github issue link. So potentially this is similar.
In case anyone else comes across this thread, my workaround has been to launch JupyterLab from the Paperspace notebook.