diff options
Diffstat (limited to 'tensorflow/contrib/autograph/converters/asserts_test.py')
| -rw-r--r-- | tensorflow/contrib/autograph/converters/asserts_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/autograph/converters/asserts_test.py b/tensorflow/contrib/autograph/converters/asserts_test.py index 2cd0e626bc..9c58ae3acc 100644 --- a/tensorflow/contrib/autograph/converters/asserts_test.py +++ b/tensorflow/contrib/autograph/converters/asserts_test.py @@ -32,8 +32,8 @@ class AssertsTest(converter_testing.TestCase): def test_fn(a): assert a > 0 - node = self.parse_and_analyze(test_fn, {}) - node = asserts.transform(node, self.ctx) + node, ctx = self.prepare(test_fn, {}) + node = asserts.transform(node, ctx) self.assertTrue(isinstance(node.body[0].body[0].value, gast.Call)) |