Skip to content

Commit 92fbb78

Browse files
committed
Update colormap based on config.yaml
1 parent 75036c5 commit 92fbb78

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

src/napari_deeplabcut/_widgets.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -453,17 +453,20 @@ def on_insert(self, event):
453453
self.viewer.status = f"New keypoint{'s' if len(diff) > 1 else ''} {', '.join(diff)} found."
454454
for _layer, store in self._stores.items():
455455
_layer.metadata["header"] = layer.metadata["header"]
456-
_layer.metadata["face_color_cycles"] = layer.metadata["face_color_cycles"]
457-
_layer.face_color = "label"
458-
_layer.face_color_cycle = layer.metadata["face_color_cycles"]["label"]
459-
_layer.events.face_color()
460456
store.layer = _layer
461457

462458
for menu in self._menus:
463459
menu._map_individuals_to_bodyparts()
464460
menu._update_items()
465461

466-
self._update_color_scheme()
462+
# Always update the colormap to reflect the one in the config.yaml file
463+
for _layer, store in self._stores.items():
464+
_layer.metadata["face_color_cycles"] = layer.metadata["face_color_cycles"]
465+
_layer.face_color = "label"
466+
_layer.face_color_cycle = layer.metadata["face_color_cycles"]["label"]
467+
_layer.events.face_color()
468+
store.layer = _layer
469+
self._update_color_scheme()
467470

468471
# Remove the unnecessary layer newly added
469472
QTimer.singleShot(10, self.viewer.layers.pop)

0 commit comments

Comments
 (0)