Include the operators module in the test framework as well.

PiperOrigin-RevId: 191756100

diff --git a/tensorflow/contrib/autograph/converters/BUILD b/tensorflow/contrib/autograph/converters/BUILD
index c5a0dc1..92cca30 100644 (file)
--- a/tensorflow/contrib/autograph/converters/BUILD
+++ b/tensorflow/contrib/autograph/converters/BUILD
@@ -49,6 +49,7 @@ py_library(
     visibility = ["//tensorflow:__subpackages__"],
     deps = [
         ":converters",
+        "//tensorflow/contrib/autograph/operators",
         "//tensorflow/contrib/autograph/pyct",
         "//tensorflow/contrib/autograph/pyct/static_analysis",
         "//tensorflow/contrib/autograph/utils",

diff --git a/tensorflow/contrib/autograph/converters/converter_test_base.py b/tensorflow/contrib/autograph/converters/converter_test_base.py
index 3ea2cfd..984e72c 100644 (file)
--- a/tensorflow/contrib/autograph/converters/converter_test_base.py
+++ b/tensorflow/contrib/autograph/converters/converter_test_base.py
@@ -21,6 +21,7 @@ from __future__ import print_function
 import contextlib
 import imp
 
+from tensorflow.contrib.autograph import operators
 from tensorflow.contrib.autograph import utils
 from tensorflow.contrib.autograph.pyct import compiler
 from tensorflow.contrib.autograph.pyct import context
@@ -77,6 +78,7 @@ class TestCase(test.TestCase):
       result.tf = self.make_fake_mod('fake_tf', *symbols)
       result.autograph_utils = utils
       result.autograph_api = self.make_fake_mod('fake_api', converted_call)
+      result.__ops = operators  # pylint:disable=protected-access
       yield result
     except Exception:  # pylint:disable=broad-except
       if source is None:

diff --git a/tensorflow/contrib/autograph/impl/api_test.py b/tensorflow/contrib/autograph/impl/api_test.py
index ee2d301..f156a87 100644 (file)
--- a/tensorflow/contrib/autograph/impl/api_test.py
+++ b/tensorflow/contrib/autograph/impl/api_test.py
@@ -37,8 +37,12 @@ class ApiTest(test.TestCase):
   def setUp(self):
     config.COMPILED_IMPORT_STATEMENTS = (
         'from __future__ import print_function',
-        'from tensorflow.contrib.autograph import utils as '
-        'autograph_utils', 'tf = autograph_utils.fake_tf()')
+        'from tensorflow.contrib.autograph import utils'
+        ' as autograph_utils',
+        'from tensorflow.contrib.autograph import operators'
+        ' as __ops',
+        'tf = autograph_utils.fake_tf()',
+    )
 
   def test_decorator_recurses(self):