👽️ sparse: add axis param to _spbase.count_nonzero

.mypyignore-todo

@@ -1,7 +1,6 @@
 scipy\.sparse\.(_?construct\.)?rand(om(_array)?)?
 scipy\.sparse\._coo\._coo_base\.tensordot
 scipy\.sparse\._data\._minmax_mixin\.(arg|nan)?(max|min)
-scipy\.sparse\._(base\._sp|coo\._coo_|dia\._dia_|dok\._dok_|lil\._lil_)base\.count_nonzero
 
 scipy\.spatial\.distance\.directed_hausdorff
 scipy\.spatial\.transform\.(_?rotation\.)?Rotation\.random

scipy-stubs/sparse/_base.pyi

@@ -581,7 +581,7 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
 
     #
     def nonzero(self, /) -> tuple[Index1D, Index1D]: ...
-    def count_nonzero(self, /) -> int: ...
+    def count_nonzero(self, /, axis: int | None = None) -> int: ...
     def conjugate(self, /, copy: bool = True) -> Self: ...
     def conj(self, /, copy: bool = True) -> Self: ...
     def transpose(self, /, axes: None = None, copy: bool = False) -> Self: ...

scipy-stubs/sparse/_compressed.pyi

@@ -279,10 +279,6 @@ class _cs_matrix(
         maxprint: int | None = None,
     ) -> None: ...
 
-    #
-    @override
-    def count_nonzero(self, /, axis: None = None) -> int: ...
-
     #
     def sorted_indices(self, /) -> Self: ...
     def sort_indices(self, /) -> None: ...