bug fix: colormap

DeepLabCut · Mar 27, 2024 · b25a473 · b25a473
1 parent ff8ae9a
commit b25a473
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/napari_deeplabcut/_widgets.py b/src/napari_deeplabcut/_widgets.py
@@ -1088,9 +1088,8 @@ def _update_colormap(self, colormap_name):
                     layer.metadata["header"], colormap_name,
                 )
                 layer.metadata["face_color_cycles"] = face_color_cycle_maps
-                if self.color_mode == keypoints.ColorMode.BODYPART:
-                    face_color_prop = "label"
-                else:
+                face_color_prop = "label"
+                if self.color_mode == str(keypoints.ColorMode.INDIVIDUAL):
                     face_color_prop = "id"
 
                 layer.face_color = face_color_prop