diff options
-rw-r--r-- | tensorflow/contrib/learn/__init__.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/learn/__init__.py b/tensorflow/contrib/learn/__init__.py index 05c4024d0b..aec4911e29 100644 --- a/tensorflow/contrib/learn/__init__.py +++ b/tensorflow/contrib/learn/__init__.py @@ -88,9 +88,11 @@ from __future__ import print_function from tensorflow.contrib.learn.python.learn import * # pylint: enable=wildcard-import +from tensorflow.contrib.learn.python.learn import learn_runner + from tensorflow.python.util.all_util import remove_undocumented -_allowed_symbols = ['datasets', 'head', 'io', 'models', +_allowed_symbols = ['datasets', 'head', 'io', 'learn_runner', 'models', 'monitors', 'NotFittedError', 'ops', 'preprocessing', 'utils', 'graph_actions'] |