Replace removed np.int alias with int in visualize

Signed-off-by: Sai Asish Y <say.apm35@gmail.com>
This commit is contained in:
Sai Asish Y 2026-06-18 12:40:36 -07:00
parent 67aee0b79c
commit 7446cc09b2
2 changed files with 24 additions and 2 deletions

View File

@ -52,9 +52,9 @@ class InteractivePlotter:
self.spec_slices = spec_slices
self.call_info = call_info
# _, self.labels = np.unique([cc['class'] for cc in call_info], return_inverse=True)
self.labels = np.zeros(len(call_info), dtype=np.int)
self.labels = np.zeros(len(call_info), dtype=int)
self.annotated = np.zeros(
self.labels.shape[0], dtype=np.int
self.labels.shape[0], dtype=int
) # can populate this with 1's where we have labels
self.labels_cols = [
colors[self.labels[ii]] for ii in range(len(self.labels))

View File

@ -0,0 +1,22 @@
import numpy as np
from batdetect2.utils.visualize import InteractivePlotter
def test_interactive_plotter_init_builds_integer_label_arrays():
feats_ds = np.zeros((2, 2))
spec_slices = [np.zeros((4, 6)), np.zeros((4, 8))]
call_info = [{"class": "a"}, {"class": "b"}]
plotter = InteractivePlotter(
feats_ds=feats_ds,
feats=feats_ds,
spec_slices=spec_slices,
call_info=call_info,
freq_lims=[0, 1],
allow_training=False,
)
assert plotter.labels.shape == (2,)
assert np.issubdtype(plotter.labels.dtype, np.integer)
assert np.issubdtype(plotter.annotated.dtype, np.integer)