Skip to content

Commit

Permalink
FIX bool type for cupy
Browse files Browse the repository at this point in the history
  • Loading branch information
mvdoc committed Mar 31, 2024
1 parent b0eaeb5 commit bc0e1d5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion himalaya/backend/cupy.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ def to_numpy(array):
def isin(x, y):
import numpy as np # XXX
np_result = np.isin(to_numpy(x), to_numpy(y))
return asarray(np_result, dtype=cupy.bool)
return asarray(np_result, dtype=bool)


def searchsorted(x, y):
Expand Down

0 comments on commit bc0e1d5

Please sign in to comment.