just in case you too were facing this issue on an HPC/ssh system, you need to separately install the Python extension for the ssh on VS code :)