M1 Mac 上的 TensorFlow(任何版本 > 2.5.0):PyCharm 中没有代码完成

TensorFlow (any version > 2.5.0) on M1 Mac: No code completion in PyCharm

我正在使用 JetBrain 的 PyCharm IDE 在 M1 Mac 机器上使用 TensorFlow。 我已按给定顺序使用以下命令安装了 TensorFlow:

conda create -n data-science -y python=3.9
conda activate data-science
conda install -c apple tensorflow-deps
pip install tensorflow-macos
pip install tensorflow-metal

安装Conda,我是按照苹果官方的documentation

我可以毫无问题地导入 TensorFlow;它打印的版本号是 2.8.0。 但是,我没有收到任何代码完成建议。 输入 tensorflow.keras.l 时,应提示 layerslosses。 如所附屏幕截图所示,我没有得到任何可用的东西:

我已经重新启动 IDE,使所有缓存失效,重新索引项目,并多次从头开始重新创建环境,但都无济于事。其他建议按预期工作(例如屏幕截图中的 os.path.join 行)。令我困惑的是,在同一台机器 (M1 Mac) 上的另一个项目中,我正在使用 TensorFlow 2.5.0,它为我提供了所需的代码建议。

由于无法获得最新 TensorFlow 版本的代码建议,我无法更新到 2.8.0 等版本。您对解决此问题有何建议?

对于代码完成,您可以按照以下解决方法

from tensorflow import keras

if typing.TYPE_CHECKING:
    from keras.api._v2 import keras

详情请参考here