update marks when selected dataset is changed

(including hiding when no dataset is selected)
spacetelescope · Jan 24, 2024 · e4eff61 · e4eff61
1 parent 8348b5b
commit e4eff61
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/jdaviz/core/template_mixin.py b/jdaviz/core/template_mixin.py
@@ -2038,6 +2038,8 @@ def __init__(self, plugin, items, selected, scale_factor, multiselect=None,
 
         self.add_observe('is_active', self._plugin_active_changed)
         self.add_observe(selected, self._update_mark_coords)
+        if dataset is not None:
+            self.add_observe(self.dataset._plugin_traitlets['selected'], self._update_mark_coords)
         self.add_observe(scale_factor, self._update_mark_coords)
         # add marks to any new viewers
         self.hub.subscribe(self, ViewerAddedMessage, handler=self._update_mark_coords)