[Fix] Unexpected string attribute length. (#136)

Author: Seventeen17

graphlearn/python/decoder.py

@@ -18,6 +18,7 @@
 from __future__ import division
 from __future__ import print_function
 
+import sys
 import numpy as np
 
 
@@ -123,5 +124,7 @@ def format_attrs(self, int_attrs, float_attrs, string_attrs):
 
     if string_attrs is not None:
       string_attrs = string_attrs.reshape(-1, self._string_attr_num)
-      string_attrs = string_attrs.astype('U13')
+      if (sys.version_info[0] == '3'):
+        # For Python 3.X, encode string attributes as Unicode.
+        string_attrs = string_attrs.astype('U')
     return int_attrs, float_attrs, string_attrs