diff options
| -rw-r--r-- | numpy/core/src/multiarraymodule.c | 4 | ||||
| -rw-r--r-- | numpy/linalg/linalg.py | 4 |
2 files changed, 3 insertions, 5 deletions
diff --git a/numpy/core/src/multiarraymodule.c b/numpy/core/src/multiarraymodule.c index 4e323e818..303506eca 100644 --- a/numpy/core/src/multiarraymodule.c +++ b/numpy/core/src/multiarraymodule.c @@ -3054,9 +3054,7 @@ PyArray_ArgMax(PyArrayObject *op, int axis) } Py_DECREF(ap); - SWAPBACK(op, rp); /* op now contains the return */ - - return (PyObject *)op; + return (PyObject *)rp; fail: Py_DECREF(ap); diff --git a/numpy/linalg/linalg.py b/numpy/linalg/linalg.py index bd75f9396..35f8d469c 100644 --- a/numpy/linalg/linalg.py +++ b/numpy/linalg/linalg.py @@ -24,8 +24,8 @@ class LinAlgError(Exception): pass # Helper routines -_array_kind = {'i':0, 'l': 0, 'f': 0, 'd': 0, 'F': 1, 'D': 1} -_array_precision = {'i': 1, 'l': 1, 'f': 0, 'd': 1, 'F': 0, 'D': 1} +_array_kind = {'i':0, 'l': 0, 'q': 0, 'f': 0, 'd': 0, 'F': 1, 'D': 1} +_array_precision = {'i': 1, 'l': 1, 'q': 1, 'f': 0, 'd': 1, 'F': 0, 'D': 1} _array_type = [['f', 'd'], ['F', 'D']] def _makearray(a): |