diff options
Diffstat (limited to 'taskflow/types/tree.py')
-rw-r--r-- | taskflow/types/tree.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/taskflow/types/tree.py b/taskflow/types/tree.py index 3681694..2aef2a1 100644 --- a/taskflow/types/tree.py +++ b/taskflow/types/tree.py @@ -15,11 +15,10 @@ # under the License. import collections +import io import itertools import os -import six - from taskflow.types import graph from taskflow.utils import iter_utils from taskflow.utils import misc @@ -279,9 +278,9 @@ class Node(object): """ if stringify_node is None: # Default to making a unicode string out of the nodes item... - stringify_node = lambda node: six.text_type(node.item) + stringify_node = lambda node: str(node.item) expected_lines = self.child_count(only_direct=False) + 1 - buff = six.StringIO() + buff = io.StringIO() conn = vertical_conn + horizontal_conn stop_at_parent = self for i, node in enumerate(self.dfs_iter(include_self=True), 1): |