diff options
| author | Pauli Virtanen <pav@iki.fi> | 2009-12-06 12:07:49 +0000 |
|---|---|---|
| committer | Pauli Virtanen <pav@iki.fi> | 2009-12-06 12:07:49 +0000 |
| commit | 9854a4f6d52c1fca0b5cb0e3fcafae5894ea5302 (patch) | |
| tree | f55043e74c681d73963c5a88af30788d7d775fd8 /numpy/polynomial/polytemplate.py | |
| parent | 3031a0045ce97cc6e0a4d906bcc75713672992aa (diff) | |
| download | numpy-9854a4f6d52c1fca0b5cb0e3fcafae5894ea5302.tar.gz | |
3K: polynomial: Python3 friendlier template
Diffstat (limited to 'numpy/polynomial/polytemplate.py')
| -rw-r--r-- | numpy/polynomial/polytemplate.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/numpy/polynomial/polytemplate.py b/numpy/polynomial/polytemplate.py index ca6c71a46..512fd9201 100644 --- a/numpy/polynomial/polytemplate.py +++ b/numpy/polynomial/polytemplate.py @@ -2,10 +2,16 @@ """ import string +import sys + +if sys.version_info[0] >= 3: + rel_import = "from . import" +else: + rel_import = "import" polytemplate = string.Template(''' from __future__ import division -import polyutils as pu +REL_IMPORT polyutils as pu import numpy as np class $name(pu.PolyBase) : @@ -594,4 +600,4 @@ class $name(pu.PolyBase) : off, scl = pu.mapparms($domain, domain) coef = ${nick}line(off, scl) return $name(coef, domain) -''') +'''.replace('REL_IMPORT', rel_import)) |