Use a different way to import 'version' module, which is slightly different after namespace package.
Use a different way to import 'version' module, which is slightly different after namespace package.

PiperOrigin-RevId: 351889731

tensorflow/tfx,tensorflow/tfx