Skip to content

Commit 153cecc

Browse files
committed
fix future_stack warning
1 parent 0042079 commit 153cecc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bencheval/bencheval/tabarena.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1763,7 +1763,7 @@ def dataset_fold_similarity(
17631763
if fold_sim.shape[0] >= 2:
17641764
# consider only upper triangle (excluding diagonal)
17651765
sim_vals = fold_sim.where(np.triu(np.ones(fold_sim.shape), k=1).astype(bool))
1766-
stacked = sim_vals.stack(dropna=True) # MultiIndex (seed_i, seed_j) -> similarity
1766+
stacked = sim_vals.stack(future_stack=True) # MultiIndex (seed_i, seed_j) -> similarity
17671767
if len(stacked) > 0:
17681768
(i_max, j_max), v_max = stacked.idxmax(), float(stacked.max())
17691769
(i_min, j_min), v_min = stacked.idxmin(), float(stacked.min())

0 commit comments

Comments
 (0)