diff options
Diffstat (limited to 'numpy/linalg/linalg.py')
| -rw-r--r-- | numpy/linalg/linalg.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/numpy/linalg/linalg.py b/numpy/linalg/linalg.py index 1a0dd3c6e..142de2340 100644 --- a/numpy/linalg/linalg.py +++ b/numpy/linalg/linalg.py @@ -18,7 +18,8 @@ __all__ = ['solve', from numpy.core import array, asarray, zeros, empty, transpose, \ intc, single, double, csingle, cdouble, inexact, complexfloating, \ newaxis, ravel, all, Inf, dot, add, multiply, identity, sqrt, \ - maximum, nonzero, diagonal, arange, fastCopyAndTranspose, sum + maximum, nonzero, diagonal, arange, fastCopyAndTranspose, sum, \ + argsort from numpy.lib import triu from numpy.linalg import lapack_lite @@ -75,7 +76,7 @@ def _commonType(*arrays): result_type = _complex_types_map[result_type] else: t = double - return t, rt + return t, result_type def _castCopyAndTranspose(type, *arrays): if len(arrays) == 1: @@ -351,7 +352,7 @@ def eigh(a, UPLO='L'): if results['info'] > 0: raise LinAlgError, 'Eigenvalues did not converge' at = a.transpose().astype(result_t) - return w.astype(result_t), wrap(at) + return w.astype(_realType(result_t)), wrap(at) # Singular value decomposition |