[vscode] tensorflow.keras 자동완성 안됨

 

안된다.

pycharm에선 되는데 vscode에서는 tensorflow.keras내의 메소드들이 자동완성이 안된다. 자동완성이 안 되는 환경에서 개발은 정말 열받는다. 정말 너무 짜증 나서 이것저것 깔아보고 지워보고 찾아보고 하면서 겨우 방법을 알아내서 정리한다.

 

https://stackoverflow.com/questions/68860879/vscode-keras-intellisensesuggestion-not-working-properly

Vscode keras intellisense(suggestion) not working properly

위의 stackoverflow 글을 읽어보면 keras는 lazy loading이 되는 특별한 클래스라서 그런다고 한다. 그래서 vscode에서 읽어오지 않는 것 같다. 그러니 init을 하는 부분에서 import를 당겨주면 잘 된다고 한다.

일단 vscode에 python과 pylance extension이 깔려있어야 한다.

아래 코드를 tensorflow/__init__.py맨 밑에 추가하면 자동완성이 된다.

__init__.py는 vscode에서 import tensorflow를 입력하고 tensorflow를 cmd+클릭하면 바로 찾을 수 있다.

if _typing.TYPE_CHECKING:
  from tensorflow_estimator.python.estimator.api._v2 import estimator as estimator
  from keras.api._v2 import keras
  from keras.api._v2.keras import losses
  from keras.api._v2.keras import metrics
  from keras.api._v2.keras import optimizers
  from keras.api._v2.keras import initializers
# pylint: enable=g-import-not-at-top

이제 잘 된다.