split_blocks: Shorten Python 2/3 compatible code.

Shorten the import to one line and the assignment to one line. This should keep things simple. Also should cutdown noise from linters and coverage tools.
jakirkham · Dec 1, 2016 · 10d9d08 · 10d9d08
1 parent a584343
commit 10d9d08
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/kenjutsu/kenjutsu.py b/kenjutsu/kenjutsu.py
@@ -315,11 +315,9 @@ def split_blocks(space_shape, block_shape, block_halo=None):
         xrange = range
 
     try:
-        from itertools import ifilter
-        from itertools import imap
+        from itertools import ifilter, imap
     except ImportError:
-        ifilter = filter
-        imap = map
+        ifilter, imap = filter, map
 
     if block_halo is not None:
         assert (len(space_shape) == len(block_shape) == len(block_halo)), \