diff --git a/tensorflow/python/autograph/pyct/static_analysis/type_inference.py b/tensorflow/python/autograph/pyct/static_analysis/type_inference.py index 3412a8a6aa0..4e8a9a90020 100644 --- a/tensorflow/python/autograph/pyct/static_analysis/type_inference.py +++ b/tensorflow/python/autograph/pyct/static_analysis/type_inference.py @@ -173,6 +173,7 @@ class StmtInferrer(gast.NodeVisitor): self.closure_types = closure_types self.types_in = types_in self.new_symbols = {} + self.rvalue = None def visit(self, node): types = super().visit(node)