add copyanno for safety

2020-03-23 22:58:01 +08:00 · 2020-03-23 22:58:01 +08:00 · 1a1b07304b
commit 1a1b07304b
parent e9e0d0f2bc
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/python/autograph/converters/break_statements.py
+++ b/tensorflow/python/autograph/converters/break_statements.py
@ -144,6 +144,7 @@ class BreakTransformer(converter.Base):
          orelse=node.orelse)

      new_for_node = node[0]
+      anno.copyanno(original_node, new_for_node, 'extra_test')
      anno.copyanno(original_node, new_for_node, anno.Basic.DIRECTIVES)

      return node