IteratorNode should not take kwargs. The iterator should be passed as…

… data
JohnVinyard · May 13, 2016 · eb8d54d · eb8d54d
1 parent bd4a9c5
commit eb8d54d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/featureflow/__init__.py b/featureflow/__init__.py
@@ -1,4 +1,4 @@
-__version__ = '0.6.9'
+__version__ = '0.7.9'
 
 from model import BaseModel
 

diff --git a/featureflow/iteratornode.py b/featureflow/iteratornode.py
@@ -2,10 +2,9 @@
 
 
 class IteratorNode(Node):
-    def __init__(self, iterable=None, needs=None):
-        self.iterable = iterable
+    def __init__(self, needs=None):
         super(IteratorNode, self).__init__(needs=needs)
 
     def _process(self, data):
-        for x in self.iterable:
+        for x in data:
             yield x
diff --git a/featureflow/test_integration.py b/featureflow/test_integration.py
@@ -310,7 +310,7 @@ def test_can_use_iterator_node(self):
         iterable = chunked(StringIO(data_source['mary']), chunksize=3)
 
         class D(BaseModel, self.Settings):
-            stream = Feature(IteratorNode, iterable=iterable, store=True)
+            stream = Feature(IteratorNode, store=True)
             words = Feature(Tokenizer, needs=stream, store=False)
             count = JSONFeature(WordCount, needs=words, store=True)