make izip python 2 & 3 compatible

CGATOxford · May 23, 2016 · feda398 · feda398
1 parent c2b4556
commit feda398
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/umi_tools/extract.py b/umi_tools/extract.py
@@ -83,7 +83,15 @@
 
 '''
 import sys
-from itertools import izip
+
+# python 3 doesn't require izip
+try:
+    # Python 2
+    from itertools import izip
+except ImportError:
+    # Python 3
+    izip = zip
+
 import umi_tools.Utilities as U