Skip to content

Commit f3e4397

Browse files
committed
ENH: Make sure to return int for indices
1 parent 53a206b commit f3e4397

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pandas/util/testing.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -941,8 +941,8 @@ def _gen_unique_rand(rng, _extra_size):
941941
extra_size *= 1.05
942942
ind = _gen_unique_rand(random_state, extra_size)
943943

944-
j = np.floor(ind * 1. / nrows)
945-
i = (ind - j * nrows)
944+
j = np.floor(ind * 1. / nrows).astype(int)
945+
i = (ind - j * nrows).astype(int)
946946
return i.tolist(), j.tolist()
947947

948948

0 commit comments

Comments
 (0)