Only add ModuleWrapper when lazy loading is requested or when using TF 1.x (to

print deprecation warnings).

PiperOrigin-RevId: 266944067

tensorflow/python/tools/api/generator/create_python_api.py

@@ -242,6 +242,9 @@ __all__ = [_s for _s in dir() if not _s.startswith('_')]
 __all__.extend([_s for _s in _names_with_underscore])
 ''' % underscore_names_str
 
+    # Add module wrapper if we need to print deprecation messages
+    # or if we use lazy loading.
+    if self._api_version == 1 or self._lazy_loading:
       for dest_module, _ in self._module_imports.items():
         deprecation = 'False'
         has_lite = 'False'