Fix py2.* issue.

c0fec0de · Mar 13, 2017 · f8d109d · f8d109d
1 parent f6f0aed
commit f8d109d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/anytree/iterators.py b/anytree/iterators.py
@@ -1,4 +1,8 @@
 
+import six
+
+
+@six.python_2_unicode_compatible
 class PreOrderIter(object):
 
     def __init__(self, node):
@@ -43,6 +47,7 @@ def __iter__(self):
             stack = node.children + stack[1:]
 
 
+@six.python_2_unicode_compatible
 class PostOrderIter(object):
 
     def __init__(self, node):

diff --git a/anytree/walker.py b/anytree/walker.py
@@ -1,6 +1,9 @@
 # -*- coding: utf-8 -*-
 
+import six
 
+
+@six.python_2_unicode_compatible
 class Walker(object):
 
     def __init__(self):