Thanks for checking that! I use vscode so this is a no go...
For the time being, because Mito generates pandas code for every edit you make, you can always use Mito in Jupyter to generate code, and then copy it over to VSCode. Admittedly, its not as nice of a workflow, but it does work!
[1] https://docs.trymito.io/getting-started/installing-mito/inst...