As the title states, when I write a code in a Python script file (extension: .py), the code seems to be fully colored like this:

But when writing the same code in a Jupyter notebook (extension: .ipynb), the code seems to be partially colored like this:
Am I missing any configuration in the Python extension or is it a bug?
I am working on Windows and I have the extension pack from Microsoft for Pyhton and Jupyter:

The color theme is Dark+ (default dark).


