How to import TensorFlow in VS Code