summaryrefslogtreecommitdiff
path: root/numpy/lib/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/lib/utils.py')
-rw-r--r--numpy/lib/utils.py7
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"