diff --git a/sleap/gui/app.py b/sleap/gui/app.py index 0c95dff04..184a13b96 100644 --- a/sleap/gui/app.py +++ b/sleap/gui/app.py @@ -300,7 +300,7 @@ def _initialize_gui(self): """Creates menus, dock windows, starts timers to update gui state.""" self.state["distinctly_color"] = prefs["distinctly_color"] - + self._create_color_manager() self._create_video_player() self.statusBar() @@ -658,7 +658,7 @@ def prev_vid(): key="palette", ) - distinctly_color_options = ("instance_groups", "instances", "nodes", "edges") + distinctly_color_options = ("instance groups", "instances", "nodes", "edges") add_submenu_choices( menu=viewMenu, diff --git a/sleap/gui/color.py b/sleap/gui/color.py index 219406974..889381ca6 100644 --- a/sleap/gui/color.py +++ b/sleap/gui/color.py @@ -283,7 +283,7 @@ def get_item_color( return (128, 128, 128) - if self.distinctly_color == "instance_groups" and parent_instance: + if self.distinctly_color == "instance groups" and parent_instance: if parent_frame is None and parent_instance: parent_frame = parent_instance.frame diff --git a/sleap/gui/dataviews.py b/sleap/gui/dataviews.py index 59477a57b..d3651bc96 100644 --- a/sleap/gui/dataviews.py +++ b/sleap/gui/dataviews.py @@ -706,7 +706,7 @@ def item_to_data(self, obj, item: InstanceGroup): def get_item_color(self, instance_group: InstanceGroup, key: str): color_manager = self.context.app.color_manager - if color_manager.distinctly_color == "instance_groups" and key == "name": + if color_manager.distinctly_color == "instance groups" and key == "name": # Get the RecordingSession state = self.context.state diff --git a/sleap/io/visuals.py b/sleap/io/visuals.py index f2dde0be3..adb51f602 100644 --- a/sleap/io/visuals.py +++ b/sleap/io/visuals.py @@ -456,7 +456,7 @@ def _plot_instance_cv( ) if self.show_edges: - for (src, dst) in instance.skeleton.edge_inds: + for src, dst in instance.skeleton.edge_inds: # Get points for the nodes connected by this edge src_x, src_y = points_array[src] dst_x, dst_y = points_array[dst] @@ -542,8 +542,9 @@ def save_labeled_video( for what instance/node/edge palette: SLEAP color palette to use. Options include: "alphabet", "five+", "solarized", or "standard". Only used if `color_manager` is None. - distinctly_color: Specify how to color instances. Options include: "instances", - "edges", and "nodes". Only used if `color_manager` is None. + distinctly_color: Specify how to color instances. Options include: + "instance groups", "instances", "edges", and "nodes". Only used if + `color_manager` is None. gui_progress: Whether to show Qt GUI progress dialog. Returns: