From a27eac54b099d2cc024cbaf36900ec97dbc7256a Mon Sep 17 00:00:00 2001 From: "Philipp A." Date: Tue, 18 Feb 2025 13:43:01 +0100 Subject: [PATCH] Use our sum --- tests/test_stats.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/tests/test_stats.py b/tests/test_stats.py index 818c8cf..91eb374 100644 --- a/tests/test_stats.py +++ b/tests/test_stats.py @@ -96,10 +96,5 @@ def test_sum_benchmark( except NotImplementedError: pytest.skip("random_array not implemented for dtype") - def sum(arr: Array[Any], axis: int | None) -> Array[Any]: # noqa: A001 - if hasattr(arr, "sum"): - return arr.sum(axis=axis) - return np.sum(arr, axis=axis) # type: ignore[arg-type] - - sum(arr, axis=axis) # warmup: numba compile - benchmark(sum, arr, axis=axis) + stats.sum(arr, axis=axis) # type: ignore[arg-type] # warmup: numba compile + benchmark(stats.sum, arr, axis=axis)