vscode-jupyter: Azure ML Widgets not working when connecting to Azure ML Notebooks from VS Code

Environment data

• VS Code version: 1.55.2
• Jupyter Extension version (available under the Extensions sidebar): v2021.5.745244803
• Python Extension version (available under the Extensions sidebar): v2021.4.765268190
• OS (Windows | Mac | Linux distro) and version: Windows 20H2
• Python and/or Anaconda version: 3.6
• Type of virtual environment used (N/A | venv | virtualenv | conda | …): XXX
• Jupyter server running: Local | Remote | N/A: remote

Expected behaviour

Widget should be displayed.

Actual behaviour

Output: No renderer could be found for MIME type: application/aml.mini.widget.v1.

Steps to reproduce:

Run the following, replacing source directory and compute target as necessary:

from azureml.widgets import RunDetails

ws = Workspace.from_config()
experiment = Experiment(workspace=ws, name='day1-experiment-hello')

config = ScriptRunConfig(source_directory='./code/hello', script='hello.py', compute_target='TestComputeForDon')

run = experiment.submit(config)
RunDetails(run).show()