Draft: Fixing tensorflow import errors
Summary
This MR introduces the following changes
- Fixing the import errors from
tensorflow
. The lazy import tensorflow uses is not accepted by the linters/interpreters correctly, so you get errors although this works fine. Adding the.python
fixes the issue for the linters/interpreters.