Can we edit .py files in VSCode and use Google Colab extension to run them on Google servers like Jupyter Notebooks?

I have just get my hands dirty with VSCode's extension Google Colab (by Google).

While I can connect to google servers for running Jupyter Notebooks (.ipynb files), Is there a way to use VSCode directly on .py files of my project and execute them on Google servers.

Note that I dont want to use the command !python <filename.py> from my notebooks to run the .py files of my project, I want to run them directly with VSCode just like I am able to with Jupyter Notebook.