Skip to content

Commit 5ee5315

Browse files
qwhelanMateusz Górski
authored and
Mateusz Górski
committed
BENCH: Improve perf of rolling.Apply.time_rolling (pandas-dev#29239)
1 parent 6bd5d0a commit 5ee5315

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

asv_bench/benchmarks/rolling.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -28,15 +28,15 @@ def peakmem_rolling(self, constructor, window, dtype, method):
2828
class Apply:
2929
params = (
3030
["DataFrame", "Series"],
31-
[10, 1000],
31+
[3, 300],
3232
["int", "float"],
3333
[sum, np.sum, lambda x: np.sum(x) + 5],
3434
[True, False],
3535
)
36-
param_names = ["contructor", "window", "dtype", "function", "raw"]
36+
param_names = ["constructor", "window", "dtype", "function", "raw"]
3737

3838
def setup(self, constructor, window, dtype, function, raw):
39-
N = 10 ** 5
39+
N = 10 ** 3
4040
arr = (100 * np.random.random(N)).astype(dtype)
4141
self.roll = getattr(pd, constructor)(arr).rolling(window)
4242

0 commit comments

Comments
 (0)