BUG: fix sinc for torch (#56)

data-apis · Dec 9, 2024 · ee12cd1 · ee12cd1
1 parent a89ffdd
commit ee12cd1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/array_api_extra/_funcs.py b/src/array_api_extra/_funcs.py
@@ -543,6 +543,8 @@ def sinc(x: Array, /, *, xp: ModuleType | None = None) -> Array:
         raise ValueError(err_msg)
     # no scalars in `where` - array-api#807
     y = xp.pi * xp.where(
-        x, x, xp.asarray(xp.finfo(x.dtype).eps, dtype=x.dtype, device=x.device)
+        xp.astype(x, xp.bool),
+        x,
+        xp.asarray(xp.finfo(x.dtype).eps, dtype=x.dtype, device=x.device),
     )
     return xp.sin(y) / y