Visual Studio Code: highlighted imports in python

Question

I use Visual Studio Code both on my work Windows laptop, and on my Ubuntu PC at home.

There\'s a functionality that I can see from my Windows vscode that I\'d like to hav