Extend GzipFile by iterator to support pandas 0.2

cangermueller · May 22, 2017 · bc0ff3a · bc0ff3a
1 parent 8be6c39
commit bc0ff3a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/deepcpg/data/utils.py b/deepcpg/data/utils.py
@@ -299,6 +299,12 @@ def __init__(self, filename, mode='r', *args, **kwargs):
         else:
             self.fh = open(filename, mode, *args, **kwargs)
 
+    def __iter__(self):
+        return self.fh.__iter__()
+
+    def __next__(self):
+        return self.fh.__next__()
+
     def read(self, *args, **kwargs):
         return self.fh.read(*args, **kwargs)