diff --git a/nwbwidgets/ophys.py b/nwbwidgets/ophys.py index 8ca02a54..3cf14bb9 100644 --- a/nwbwidgets/ophys.py +++ b/nwbwidgets/ophys.py @@ -279,7 +279,7 @@ def show_plane_segmentation_2d( plane_seg_hover_dict = { key: self.plane_seg[key].data for key in self.plane_seg.colnames - if key not in ["pixel_mask", "image_mask"] + if len(self.plane_seg[key].data.shape) == 1 } plane_seg_hover_dict.update(id=self.plane_seg.id.data) plane_seg_hover_df = pd.DataFrame(plane_seg_hover_dict)