try to get iterator working in Python 3

jtauber · Mar 19, 2016 · 3b5c23c · 3b5c23c
1 parent 2f39598
commit 3b5c23c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sgf.py b/sgf.py
@@ -129,9 +129,13 @@ def output(self, f):
 
 
 class NodeIterator:
+
     def __init__(self, game_tree):
         self.node = game_tree.nodes[0]
 
+    def __iter__(self):
+        return self
+
     def next(self):
         if self.node:
             node = self.node