Copy module list before iterating over it. Also, import python module…

… for clarity
STCYC · May 2, 2018 · fb82066 · fb82066
1 parent bc86da0
commit fb82066
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tensorflow/tools/api/generator/create_python_api.py b/tensorflow/tools/api/generator/create_python_api.py
@@ -23,6 +23,7 @@
 import os
 import sys
 
+from tensorflow import python  # pylint: disable=unused-import
 from tensorflow.python.util import tf_decorator
 
 
@@ -158,7 +159,8 @@ def get_api_init_text():
 
   # Traverse over everything imported above. Specifically,
   # we want to traverse over TensorFlow Python modules.
-  for module in sys.modules.values():
+  module_list = list(sys.modules.values())
+  for module in module_list:
     # Only look at tensorflow modules.
     if (not module or not hasattr(module, '__name__') or
         'tensorflow.' not in module.__name__):