Add unicode string support to pyct pretty printer.

PiperOrigin-RevId: 237161597

tensorflow/python/autograph/pyct/pretty_printer.py

@@ -18,7 +18,9 @@
 from __future__ import division
 from __future__ import print_function
 
+
 import gast
+import six
 import termcolor
 
 
@@ -106,9 +108,12 @@ def generic_visit(self, node, name=None):
           self._print('%s%s=()' % (self._indent(), self._field(f)))
       elif isinstance(v, gast.AST):
         self.generic_visit(v, f)
-      elif isinstance(v, str):
+      elif isinstance(v, six.binary_type):
+        self._print('%s%s=%s' % (self._indent(), self._field(f),
+                                 self._value('b"%s"' % v)))
+      elif isinstance(v, six.text_type):
         self._print('%s%s=%s' % (self._indent(), self._field(f),
-                                 self._value('"%s"' % v)))
+                                 self._value('u"%s"' % v)))
       else:
         self._print('%s%s=%s' % (self._indent(), self._field(f),
                                  self._value(v)))

tensorflow/python/autograph/pyct/pretty_printer_test.py

@@ -19,13 +19,22 @@
 from __future__ import print_function
 
 import ast
+import textwrap
 
 from tensorflow.python.autograph.pyct import pretty_printer
 from tensorflow.python.platform import test
 
 
 class PrettyPrinterTest(test.TestCase):
 
+  def test_unicode_bytes(self):
+    source = textwrap.dedent('''
+    def f():
+      return b'b', u'u', 'depends_py2_py3'
+    ''')
+    node = ast.parse(source)
+    self.assertIsNotNone(pretty_printer.fmt(node))
+
   def test_format(self):
     node = ast.FunctionDef(
         name='f',