MNT: move bz2 back to function level

closes matplotlib#10866
magnunor · Mar 23, 2018 · ffe9203 · ffe9203
1 parent 7a3248f
commit ffe9203
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/matplotlib/cbook/__init__.py b/lib/matplotlib/cbook/__init__.py
@@ -8,7 +8,6 @@
 
 import six
 from six.moves import xrange, zip
-import bz2
 import collections
 import contextlib
 import datetime
@@ -478,6 +477,9 @@ def to_filehandle(fname, flag='rU', return_opened=False, encoding=None):
             flag = flag.replace('U', '')
             fh = gzip.open(fname, flag)
         elif fname.endswith('.bz2'):
+            # python may not be complied with bz2 support,
+            # bury import until we need it
+            import bz2
             # get rid of 'U' in flag for bz2 files
             flag = flag.replace('U', '')
             fh = bz2.BZ2File(fname, flag)