Merge pull request #103 from eevee/py3

Python 3 compatibility, sort of, maybe

1 files changed, 6 insertions, 2 deletions

diff --git a/pokemontools/interval_map.py b/pokemontools/interval_map.py
index daf22cd..fc53e69 100644
--- a/pokemontools/interval_map.py
+++ b/pokemontools/interval_map.py
@@ -1,7 +1,11 @@
 # -*- coding: utf-8 -*-
 
 from bisect import bisect_left, bisect_right
-from itertools import izip
+try:
+    from future_builtins import zip
+except ImportError:
+    pass
+
 
 class IntervalMap(object):
     """
@@ -75,7 +79,7 @@ class IntervalMap(object):
         ((low_bound, high_bound), value)
         these items are returned in order"""
         previous_bound = None
-        for (b, v) in izip(self._bounds, self._items):
+        for (b, v) in zip(self._bounds, self._items):
             if v is not None:
                 yield (previous_bound, b), v
             previous_bound = b