ModuleNotFoundError: No module named 'pymavlink'

I am trying to run a basic script by following this tutorial: https://www.youtube.com/watch?v=kecnaxlUiTY
I got to the part where he runs the code with a print statement saying that hearbeat has been found.
I copied the same code on my vscode in windows with sitl setup through mission planner and got the error in the title even though I had done pip install pymavlink.
Here is an image: