Fix

google · Jul 8, 2020 · 7d7b28f · 7d7b28f
1 parent 280a362
commit 7d7b28f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/apitools/gen/gen_client_test.py b/apitools/gen/gen_client_test.py
@@ -73,6 +73,8 @@ def testGenClient_SimpleDocEmptyInit(self):
             init_file = _GetContent(os.path.join(tmp_dir_path, '__init__.py'))
             self.assertEqual("""\"""Package marker file.\"""
 
+from __future__ import absolute_import
+
 import pkgutil
 
 __path__ = pkgutil.extend_path(__path__, __name__)

diff --git a/apitools/gen/service_registry.py b/apitools/gen/service_registry.py
@@ -207,7 +207,7 @@ def WriteFile(self, printer):
         if self.__root_package:
             import_prefix = 'from {0} '.format(self.__root_package)
         else:
-            import_prefix = 'from . '
+            import_prefix = ''
         printer('%simport %s as messages', import_prefix,
                 client_info.messages_rule_name)
         printer()