bug fix with debug prints: change color map

DeepLabCut · Mar 27, 2024 · a36fdfe · a36fdfe
1 parent f2d859b
commit a36fdfe
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/napari_deeplabcut/_widgets.py b/src/napari_deeplabcut/_widgets.py
@@ -865,7 +865,18 @@ def to_hex(nparray):
             return res
 
         self._display.reset()
-        mode = "label" if self.color_mode == str(keypoints.ColorMode.BODYPART) else "id"
+        logging.warning(self.color_mode)
+        logging.warning(str(keypoints.ColorMode.BODYPART))
+        logging.warning(str(keypoints.ColorMode.INDIVIDUAL))
+        logging.warning(str(self.color_mode == str(keypoints.ColorMode.BODYPART)))
+        logging.warning(str(self.color_mode == str(keypoints.ColorMode.INDIVIDUAL)))
+        logging.warning("---")
+        mode = "label"
+        if self.color_mode == str(keypoints.ColorMode.BODYPART):
+            mode = "id"
+
+        logging.warning(f"SELECTED MODE: {mode}")
+        logging.warning("---")
         for layer in self.viewer.layers:
             if isinstance(layer, Points) and layer.metadata:
                 self._display.update_color_scheme(
@@ -874,7 +885,6 @@ def to_hex(nparray):
                         for name, color in layer.metadata["face_color_cycles"][mode].items()
                     }
                 )
-                break
 
     def _remap_frame_indices(self, layer):
         if not self._images_meta.get("paths"):