commit 1528ec49557cb4b40a8a5d3782296724d7b46c45
author A. Unique TensorFlower <gardener@tensorflow.org> Mon Jan 10 15:26:54 2022 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Jan 10 15:31:54 2022 -0800
tree cf2304b65fd2e1d210f1034378ad7ed29da3da50
parent c40a23ddb6e8a4ae5b18f466adde3b367456a606

TF upgrade tool fix for new google/pasta versions.

PiperOrigin-RevId: 420871518
Change-Id: Iac67856b983427a89837a8eaf8a4a5268c182633

tensorflow/tools/compatibility/tf_upgrade_v2.py

diff --git a/tensorflow/tools/compatibility/tf_upgrade_v2.py b/tensorflow/tools/compatibility/tf_upgrade_v2.py
index 050fb3c..9ce6361 100644
--- a/tensorflow/tools/compatibility/tf_upgrade_v2.py
+++ b/tensorflow/tools/compatibility/tf_upgrade_v2.py
@@ -2074,8 +2074,8 @@
     if keyword_arg.arg == "step":
       return node
   default_value = "tf.compat.v1.train.get_or_create_global_step()"
-  # Parse with pasta instead of ast to avoid emitting a spurious trailing \n.
-  ast_value = pasta.parse(default_value)
+  ast_value = ast.parse(default_value).body[0].value
+  del ast_value.lineno  # hack to prevent spurious reordering of call args
   node.keywords.append(ast.keyword(arg="step", value=ast_value))
   logs.append((
       ast_edits.WARNING, node.lineno, node.col_offset,