Handle iterables in __prepare_labeled_tweets

JulianKniephoff · Jan 2, 2017 · 739fea5 · 739fea5
1 parent 6f44f72
commit 739fea5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cnn.py b/cnn.py
@@ -68,6 +68,9 @@ def output_for_class(class_number):
             output[class_number] = 1
             return output
 
+        # In case we are getting an iterator, we collect it here
+        # since we iterate over it twice
+        tweets = list(tweets)
         return {
             'input': self.__tweets_to_indices(
                 labeled_tweet.tweet for labeled_tweet in tweets