diff options
Diffstat (limited to 'tensorflow/python/autograph/pyct/static_analysis/live_values.py')
| -rw-r--r-- | tensorflow/python/autograph/pyct/static_analysis/live_values.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/tensorflow/python/autograph/pyct/static_analysis/live_values.py b/tensorflow/python/autograph/pyct/static_analysis/live_values.py index 48b442f3bd..3963772dad 100644 --- a/tensorflow/python/autograph/pyct/static_analysis/live_values.py +++ b/tensorflow/python/autograph/pyct/static_analysis/live_values.py @@ -29,10 +29,11 @@ from tensorflow.python.autograph.pyct import anno from tensorflow.python.autograph.pyct import transformer from tensorflow.python.autograph.pyct.static_analysis.annos import NodeAnno + # TODO(aqj): Do we need this? Do other builtins fail in similar ways # See b/114389775 for a related bug in pyct # These symbols are legal in Python, but don't appear in the namespace. -_special_symbols = {'range': range} +_SPECIAL_SYMBOLS = {'range': range, 'print': print} class LiveValueResolver(transformer.Base): @@ -71,8 +72,10 @@ class LiveValueResolver(transformer.Base): # If the symbol value is for example a primitive, then it will not # have a name. pass - elif node.id in _special_symbols: - anno.setanno(node, 'live_val', _special_symbols[node.id]) + elif node.id in _SPECIAL_SYMBOLS: + # Note: if the user redefined any of these symbols, then they would + # be visible in the namespace and we would never reach this branch. + anno.setanno(node, 'live_val', _SPECIAL_SYMBOLS[node.id]) else: pass # TODO(mdan): Should we raise an error here? |