Как сделать так, чтобы IDEA / PyCharm правильно обрабатывал @tf_export?

Question

Как заставить IDEA / PyCharm правильно обрабатывать аннотацию @tf_export?

Я перешел на TF 1.12 и для следующего кода

import tensorflow as tf
from tensorflow.keras import layers

model = tf.keras.Sequential()
model.add(layers.Dense(64, activation='relu'))

Я вижу, что IDEA выделяет импорт layers.

Я не могу перейти к объявлению Dense, щелкнув по нему правой кнопкой мыши.

Код выполняется правильно.

Есть ли способ это исправить?

Answer 1

Вы можете сделать это:

import tensorflow.python.keras
from keras import Sequential, layers