Skip to content

Commit da52e82

Browse files
committed
Addresses comments
1 parent 302a19b commit da52e82

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

benchmark_v2/optional/multidim_binop_benchmark.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import operator
33

44
import pytest
5-
5+
import functools
66
import arkouda as ak
77

88
from benchmark_v2.benchmark_utils import calc_num_bytes
@@ -13,6 +13,7 @@
1313
OPS = ("+", "-", "*", "/", "//", "&", "|", "^")
1414

1515

16+
@functools.cache
1617
def choose_shape(n: int, ndim: int) -> tuple[int, ...]:
1718
"""
1819
Choose an ``ndim``-dimensional shape whose element count is as close as possible

0 commit comments

Comments
 (0)