Making data set iterator compatible for Python 2 and 3.

multi_categorical_gans/datasets/dataset.py

@@ -1,5 +1,7 @@
 from __future__ import division
 
+from future.utils import implements_iterator
+
 import numpy as np
 
 
@@ -23,6 +25,7 @@ def batch_iterator(self, batch_size, shuffle=True):
         return DatasetIterator(self.features, indices, batch_size)
 
 
+@implements_iterator
 class DatasetIterator(object):
 
     def __init__(self, features, indices, batch_size):
@@ -36,7 +39,7 @@ def __init__(self, features, indices, batch_size):
     def __iter__(self):
         return self
 
-    def next(self):
+    def __next__(self):
         if self.batch_index >= self.num_batches:
             raise StopIteration
         else:

requirements.txt

@@ -1,3 +1,4 @@
+future==0.17.1
 numpy==1.16.0
 scikit-learn==0.20.2
 scipy==1.2.0