From 8944a3eeb18c2374f02759324cbeded018c5868b Mon Sep 17 00:00:00 2001 From: Dan Moldovan Date: Thu, 18 Jun 2020 13:23:05 -0700 Subject: [PATCH] Enable type annotations for python/autograph. PiperOrigin-RevId: 317168016 Change-Id: I57341cc0347ab8eb008949e191da1415069203b7 --- tensorflow/python/autograph/pyct/parser.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/tensorflow/python/autograph/pyct/parser.py b/tensorflow/python/autograph/pyct/parser.py index 747d56e401d..9ac7c2ef2a6 100644 --- a/tensorflow/python/autograph/pyct/parser.py +++ b/tensorflow/python/autograph/pyct/parser.py @@ -22,6 +22,7 @@ from __future__ import division from __future__ import print_function import re +import sys import textwrap import tokenize @@ -33,11 +34,18 @@ from tensorflow.python.autograph.pyct import errors from tensorflow.python.autograph.pyct import inspect_utils -STANDARD_PREAMBLE = textwrap.dedent(""" - from __future__ import division - from __future__ import print_function +PY2_PREAMBLE = textwrap.dedent(""" +from __future__ import division +from __future__ import print_function """) -STANDARD_PREAMBLE_LEN = 2 +PY3_PREAMBLE = '' + +if sys.version_info >= (3,): + STANDARD_PREAMBLE = PY3_PREAMBLE +else: + STANDARD_PREAMBLE = PY2_PREAMBLE + +STANDARD_PREAMBLE_LEN = STANDARD_PREAMBLE.count('__future__') _LEADING_WHITESPACE = re.compile(r'\s*')