Skip to content

Commit 02ca580

Browse files
Fix "bad operand type for" (#955)
* Add parametrxp in vvsort Co-authored-by: Alexander-Makaryev <[email protected]>
1 parent c5c68ef commit 02ca580

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/test_linalg.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,11 @@
55
import numpy
66

77

8-
def vvsort(val, vec, size):
8+
def vvsort(val, vec, size, xp):
99
for i in range(size):
1010
imax = i
1111
for j in range(i + 1, size):
12-
if numpy.abs(val[imax]) < numpy.abs(val[j]):
12+
if xp.abs(val[imax]) < xp.abs(val[j]):
1313
imax = j
1414

1515
temp = val[i]
@@ -86,10 +86,10 @@ def test_eig_arange(type, size):
8686
np_val, np_vec = numpy.linalg.eig(symm)
8787

8888
# DPNP sort val/vec by abs value
89-
vvsort(dpnp_val, dpnp_vec, size)
89+
vvsort(dpnp_val, dpnp_vec, size, inp)
9090

9191
# NP sort val/vec by abs value
92-
vvsort(np_val, np_vec, size)
92+
vvsort(np_val, np_vec, size, numpy)
9393

9494
# NP change sign of vectors
9595
for i in range(np_vec.shape[1]):

0 commit comments

Comments
 (0)