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.

Conformity

• Changelog entry
• Documentation
• Development guidelines
• Style guides