fix(python): Change recognition of numba ufunc (#15916)

pola-rs · Apr 28, 2024 · 49ef964 · 49ef964
1 parent 7ae1e58
commit 49ef964
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/py-polars/polars/expr/expr.py b/py-polars/polars/expr/expr.py
@@ -293,7 +293,8 @@ def __array_ufunc__(
         if method != "__call__":
             msg = f"Only call is implemented not {method}"
             raise NotImplementedError(msg)
-        is_custom_ufunc = ufunc.__class__ != np.ufunc
+        # Numpy/Scipy ufuncs have signature None but numba signatures always exists.
+        is_custom_ufunc = getattr(ufunc, "signature") is not None  # noqa: B009
         num_expr = sum(isinstance(inp, Expr) for inp in inputs)
         exprs = [
             (inp, Expr, i) if isinstance(inp, Expr) else (inp, None, i)