Skip to content

Commit aaddb60

Browse files
Merge remote-tracking branch 'origin/albert/new-dataset-support' into albert/new-dataset-support
2 parents 069474f + 6fb07d8 commit aaddb60

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lib/src/otx/data/samplers/class_incremental_sampler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def __init__(
6464
super().__init__(dataset)
6565

6666
# Need to split new classes dataset indices & old classses dataset indices
67-
ann_stats = dataset.get_idx_list_per_classes(True)
67+
ann_stats = dataset.get_idx_list_per_classes(use_string_label=True)
6868
new_indices, old_indices = [], []
6969
for cls in new_classes:
7070
new_indices.extend(ann_stats[cls])

0 commit comments

Comments
 (0)