experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use importlib instead of __import__ since that;s the recommended way.

Browse Source 
PiperOrigin-RevId: 267462334
This commit is contained in:
Yash Katariya 2019-09-05 15:01:12 -07:00 committed by TensorFlower Gardener
parent a8045ee92e
commit d6eab240b9
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/tools/docs/tf_doctest.py
View File

@ -18,6 +18,7 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import importlib
import os
import pkgutil
import re
@ -48,7 +49,7 @@ def find_modules():
for _, name, _ in pkgutil.walk_packages(
core.__path__, prefix=core.__name__ + '.'):
try:
tf_modules.append(__import__(name, fromlist=['']))
tf_modules.append(importlib.import_module(name))
except (ImportError, AttributeError):
pass
@ -105,8 +106,10 @@ def load_tests(unused_loader, tests, unused_ignore):
tf_modules = filter_on_submodules(tf_modules, FLAGS.module)
if FLAGS.list:
print('**************************************************')
for mod in tf_modules:
print(mod.__name__)
print('**************************************************')
return tests
for module in tf_modules: