Skip to content

Commit 2485e63

Browse files
committed
fix sub sampling of test data
1 parent 1ca2920 commit 2485e63

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -376,7 +376,7 @@ def prepare_data_index(self,
376376
if 0 < testing_max_sounds < len(self.data_index[set_index]):
377377
self.data_index[set_index] = self.np_rng.choice(self.data_index[set_index], \
378378
testing_max_sounds,
379-
replace=False)
379+
replace=False).tolist()
380380
if set_index == 'testing':
381381
self.catalog_overlaps(self.data_index['testing'])
382382
labels = [sound['label'] for sound in self.data_index['testing']]

0 commit comments

Comments
 (0)