Skip to content

Commit 3ff8522

Browse files
authored
add caching to _mean_var (#74)
1 parent 42137b1 commit 3ff8522

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/fast_array_utils/stats/_mean_var.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ def _sparse_mean_var(
8383
)
8484

8585

86-
@numba.njit
86+
@numba.njit(cache=True)
8787
def sparse_mean_var_minor_axis(
8888
data: NDArray[np.number[Any]],
8989
indptr: NDArray[np.integer[Any]],
@@ -115,7 +115,7 @@ def sparse_mean_var_minor_axis(
115115
return means, variances
116116

117117

118-
@numba.njit
118+
@numba.njit(cache=True)
119119
def sparse_mean_var_major_axis(
120120
data: NDArray[np.number[Any]],
121121
indptr: NDArray[np.integer[Any]],

0 commit comments

Comments
 (0)