diff options
Diffstat (limited to 'numpy/lib/utils.py')
-rw-r--r-- | numpy/lib/utils.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/numpy/lib/utils.py b/numpy/lib/utils.py index ee49b41d6..c4b692a69 100644 --- a/numpy/lib/utils.py +++ b/numpy/lib/utils.py @@ -848,7 +848,12 @@ def _lookfor_generate_cache(module, import_modules, regenerate): global _lookfor_caches # Local import to speed up numpy's import time. import inspect - from cStringIO import StringIO + + if sys.version_info[0] >= 3: + # In Python3 stderr, stdout are text files. + from io import StringIO + else: + from StringIO import StringIO if module is None: module = "numpy" |