Skip to content

Commit 797fade

Browse files
authored
Fix mint.nonzero interface call (#2001)
1 parent a660aec commit 797fade

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mindnlp/core/ops/array.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def narrow(input, dim, start, length):
130130
has_nonzero = hasattr(mindspore.mint, 'nonzero')
131131
def nonzero(input, *, as_tuple=False):
132132
if use_pyboost() and has_nonzero:
133-
return mindspore.mint.nonzero(input, as_tuple)
133+
return mindspore.mint.nonzero(input, as_tuple=as_tuple)
134134
_nonzero = _get_cache_prim(ops.NonZero)()
135135
out = _nonzero(input)
136136
if as_tuple:

0 commit comments

Comments
 (0)