【Tensorflow】VSCodeのPylanceコード補完がTensorflowのKerasに効かない【VSCode】

2022/3/12追記

下記WorkAroundがうまく行かない場合があります。
僕の環境でもなんか気づいたらうまくいかなくなってました。2.7とかそのへんからかもしれません。
その場合は

from keras.api._v2 import keras

とするとうまく回避できます。
見た目が気持ち悪いのでうまくpylanceに適合できるようにTensorflow側にPRが立っています。
進行中の議論の内容はTensorflowのIssueでご確認ください。

KerasがTensorflowから独立したらしい

Keras been split into a separate PIP package (keras), and its code has been moved to the GitHub repositorykeras-team/keras. The API endpoints for tf.keras stay unchanged, but are now backed by the keras PIP package. The existing code in tensorflow/python/keras is a staled copy and will be removed in future release (2.7). Please remove any imports to tensorflow.python.keras and replace them with public tf.keras API instead.

Tensorflow Release Note Version 2.6.0

こうしたいろいろの関係があって内部処理ではTensorflowはKerasをLazyにLoadすることにしました。
これはPylanceなどによる静的型チェックの対象外です。

その結果VSCodeにPython/Pylance拡張を入れてもコード補完などのサポートが受けられなくなってしまうのでした。

Solution: tf.kerasにimport文をかける

# pylance works!
import tensorflow.keras as keras
layer = keras.layers.Dense(128)

# pylance does not work!
import tensorflow as tf
layer = tf.keras.layers.Dense(128)

LazyLoaderによってロードされたkerasモジュールをPylanceに解析させるにはimport文をtensorflow.kerasに対して実行する必要があります。

そうするとLazyLoaderが処理されてPylanceは今まで通りのkerasモジュールと解釈してくれるようになるのでした。

最後に

最近PythonでもTypeScriptでもTypeHintingアツい！PylanceによるCode補完を使うとVSCodeの画面がカラフルになって楽しい！
LazyLoaderをやっつけてよいTypeHintingライフを〜

それでは最後にLGTMとフォローも忘れずにお願いします！