Bugfix

paulsengroup · May 7, 2024 · e2298c4 · e2298c4
1 parent 9c62edb
commit e2298c4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/benchmarks/bin_table_merge_bin_ids.py b/benchmarks/bin_table_merge_bin_ids.py
@@ -10,7 +10,6 @@
 
 import numpy as np
 import numpy.typing as npt
-import bioframe as bf
 
 import hictkpy
 import pandas as pd
@@ -80,8 +79,9 @@ def benchmark_pandas_vec(bins: pd.DataFrame, chroms: npt.NDArray, starts: npt.ND
                 (bins["start"] <= row["start"]) &
                 (bins["end"] > row["start"])
             ].index[0],
-            number=iters,
-        )
+            axis="columns"
+        ),
+        number=iters
     )
     # fmt: on