diff options
Diffstat (limited to 'numpy/lib/_datasource.py')
| -rw-r--r-- | numpy/lib/_datasource.py | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/numpy/lib/_datasource.py b/numpy/lib/_datasource.py index 2d35065b0..617acdac1 100644 --- a/numpy/lib/_datasource.py +++ b/numpy/lib/_datasource.py @@ -275,8 +275,12 @@ class DataSource (object): """ # We import these here because importing urllib2 is slow and # a significant fraction of numpy's total import time. - from urllib2 import urlopen - from urllib2 import URLError + if sys.version_info[0] >= 3: + from urllib.request import urlopen + from urllib.error import URLError + else: + from urllib2 import urlopen + from urllib2 import URLError upath = self.abspath(path) @@ -421,8 +425,12 @@ class DataSource (object): """ # We import this here because importing urllib2 is slow and # a significant fraction of numpy's total import time. - from urllib2 import urlopen - from urllib2 import URLError + if sys.version_info[0] >= 3: + from urllib.request import urlopen + from urllib.error import URLError + else: + from urllib2 import urlopen + from urllib2 import URLError # Test local path if os.path.exists(path): |