diff --git a/src/spikeinterface/widgets/sorting_summary.py b/src/spikeinterface/widgets/sorting_summary.py index 7cd5b47c46..4b722db233 100644 --- a/src/spikeinterface/widgets/sorting_summary.py +++ b/src/spikeinterface/widgets/sorting_summary.py @@ -202,9 +202,11 @@ def plot_figpack(self, data_plot, **backend_kwargs): if dp.curation: if use_sortingview: curation_class = vv_views.SortingCuration2 + curation_kwargs = {"label_choices": dp.label_choices} else: curation_class = vv_views.SortingCuration - v_curation = curation_class(label_choices=dp.label_choices) + curation_kwargs = {"default_label_options": dp.label_choices} + v_curation = curation_class(**curation_kwargs) v1 = vv_base.Splitter( direction="vertical", item1=vv_views.LayoutItem(v_units_table), item2=vv_views.LayoutItem(v_curation) ) diff --git a/src/spikeinterface/widgets/utils_figpack.py b/src/spikeinterface/widgets/utils_figpack.py index d43d3e1d21..0766a94390 100644 --- a/src/spikeinterface/widgets/utils_figpack.py +++ b/src/spikeinterface/widgets/utils_figpack.py @@ -101,9 +101,11 @@ def generate_unit_table_view( units_tables = make_units_table_from_sorting(sorting) # analyzer = None + unit_ids = make_serializable(sorting.unit_ids) + if unit_properties is None: ut_columns = [] - ut_rows = [vv_views.UnitsTableRow(unit_id=u, values={}) for u in sorting.unit_ids] + ut_rows = [vv_views.UnitsTableRow(unit_id=u, values={}) for u in unit_ids] else: # keep only selected columns unit_properties = np.array(unit_properties) @@ -125,7 +127,7 @@ def generate_unit_table_view( ut_columns.append(vv_views.UnitsTableColumn(key=col, label=col, dtype=txt_dtype)) ut_rows = [] - for unit_index, unit_id in enumerate(sorting.unit_ids): + for unit_index, unit_id in enumerate(unit_ids): row_values = {} for col in unit_properties: values = units_tables[col].to_numpy()