From 41cfa5da49577fb04908997c70946a4881c85430 Mon Sep 17 00:00:00 2001 From: Dan Moldovan Date: Mon, 10 Dec 2018 09:30:43 -0800 Subject: [PATCH] Actually restrict parse_expression to expression nodes. PiperOrigin-RevId: 224826621 --- tensorflow/python/autograph/pyct/parser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/autograph/pyct/parser.py b/tensorflow/python/autograph/pyct/parser.py index 39fc1a7ed05..d04a40157e7 100644 --- a/tensorflow/python/autograph/pyct/parser.py +++ b/tensorflow/python/autograph/pyct/parser.py @@ -117,7 +117,7 @@ def parse_expression(src): """ node = parse_str(src) assert isinstance(node, gast.Module) - if len(node.body) != 1 and not isinstance(node.body[0], gast.Expr): + if len(node.body) != 1 or not isinstance(node.body[0], gast.Expr): raise ValueError( 'Expected a single expression, found instead %s' % node.body) return node.body[0].value -- GitLab