It seems like VS Community with python tools has a better syntax highlighting than VS Code with \"Python\" by Don Jayamanne. As you can see in the picture attached, VS Community
I was facing the same issue. I used the Chromodynamics theme which is compatible with MagicPython. You can find it here or install directly in the VS-CODE