Skip to content

Commit ec4dc36

Browse files
committed
correct index
1 parent 31e3813 commit ec4dc36

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

benchmarks/test_benchmark_coo.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,9 +115,8 @@ def bench():
115115
def test_index_fancy(benchmark, indexing_args, seed):
116116
x = indexing_args
117117
side = x.shape[0]
118-
rank = x.ndim
119118
rng = np.random.default_rng(seed=seed)
120-
index = rng.integers(0, side, size=(side // 2,) * rank)
119+
index = rng.integers(0, side, size=(side // 2,))
121120

122121
x[index] # Numba compilation
123122

0 commit comments

Comments
 (0)