From b25a4738d60acbbf0be3819af1b368e44943a958 Mon Sep 17 00:00:00 2001 From: Niels Poulsen Date: Wed, 27 Mar 2024 17:58:56 +0100 Subject: [PATCH] bug fix: colormap --- src/napari_deeplabcut/_widgets.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/napari_deeplabcut/_widgets.py b/src/napari_deeplabcut/_widgets.py index 1eb1f60..5254d58 100644 --- 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