From b0cd40d7c7fd3828ae15bbbcf8b5f1f272ebf5c2 Mon Sep 17 00:00:00 2001 From: Dan Moldovan Date: Mon, 22 Jul 2019 12:24:09 -0700 Subject: [PATCH] Internal cleanup: avoid discarding the result of NodeTransformer visitor, for consistency. This is a no-op, because generic_visit doesn't ever replace the node, so the CL is purely for consistency. PiperOrigin-RevId: 259380818 --- tensorflow/python/autograph/converters/control_flow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/autograph/converters/control_flow.py b/tensorflow/python/autograph/converters/control_flow.py index 526c642c337..7f846bacf5f 100644 --- a/tensorflow/python/autograph/converters/control_flow.py +++ b/tensorflow/python/autograph/converters/control_flow.py @@ -347,7 +347,7 @@ class ControlFlowTransformer(converter.Base): return loop_vars, loop_vars_ast_tuple def visit_While(self, node): - self.generic_visit(node) + node = self.generic_visit(node) (basic_loop_vars, composite_loop_vars, reserved_symbols, possibly_undefs) = self._get_loop_vars( @@ -419,7 +419,7 @@ class ControlFlowTransformer(converter.Base): return undefined_assigns + node def visit_For(self, node): - self.generic_visit(node) + node = self.generic_visit(node) (basic_loop_vars, composite_loop_vars, reserved_symbols, possibly_undefs) = self._get_loop_vars(