[Python] Import "tensorflow.keras" could not be resolved after upgrading to TensorFlow 2.8.0

TensorFlow 2.8 was recently released and I installed it as soon as it was out. I really need it for support of higher NumPy versions and a few new features. However, after installing it in my conda environment with

python3 -m pip install --upgrade tensorflow

neither PyCharm nor VSCode can no longer resolve the import from tensorflow.keras import ....

The imports themselves seem to work at runtime, but because the import cannot be resolved I can't make use of code completion, visualizing signatures of functions and etc. Has anybody encountered a similar issue?

everything was working with TF 2.7 - the version I had before.

Note: I'm using Python 3.8

Vs Code

PyCharm

I tried to check the versions through the PyCharm interpreter tab and this is what I saw. For some reason PyCharm isn't aware that there are versions after 2.0 (I have the latest version of pip installed in that environment). I'm guessing this is related, but not sure what to do with that.



Continue reading...