🐛 special.logsumexp: add missing axis parameter defaults

Author: jorenham

scipy-stubs/special/_logsumexp.pyi

@@ -22,15 +22,15 @@ def logsumexp(
 @overload
 def logsumexp(
     a: onp.ToFloatND,
-    axis: AnyShape,
+    axis: AnyShape | None = None,
     b: onp.ToFloat | onp.ToFloatND | None = None,
     keepdims: bool = False,
     return_sign: Falsy = False,
 ) -> np.float64 | onp.ArrayND[np.float64]: ...
 @overload
 def logsumexp(
     a: onp.ToComplexND,
-    axis: AnyShape,
+    axis: AnyShape | None = None,
     b: onp.ToFloat | onp.ToFloatND | None = None,
     keepdims: bool = False,
     return_sign: Falsy = False,
@@ -45,12 +45,17 @@ def logsumexp(
 ) -> tuple[np.float64 | np.complex128, bool | np.bool_]: ...
 @overload
 def logsumexp(
-    a: onp.ToFloatND, axis: AnyShape, b: onp.ToFloat | onp.ToFloatND | None = None, keepdims: bool = False, *, return_sign: Truthy
+    a: onp.ToFloatND,
+    axis: AnyShape | None = None,
+    b: onp.ToFloat | onp.ToFloatND | None = None,
+    keepdims: bool = False,
+    *,
+    return_sign: Truthy,
 ) -> tuple[np.float64, bool | np.bool_] | tuple[onp.ArrayND[np.float64], onp.ArrayND[np.bool_]]: ...
 @overload
 def logsumexp(
     a: onp.ToComplexND,
-    axis: AnyShape,
+    axis: AnyShape | None = None,
     b: onp.ToFloat | onp.ToFloatND | None = None,
     keepdims: bool = False,
     *,

tests/special/teest_logsumexp.pyi

@@ -0,0 +1,7 @@
+import numpy as np
+
+from scipy.special import logsumexp
+
+# https://github.com/scipy/scipy-stubs/issues/697
+x = np.asarray([1, 2, 3], dtype=np.float64)
+logsumexp(x)