further ComponentID related fix

spacetelescope · Aug 16, 2023 · d728308 · d728308
1 parent 417fea9
commit d728308
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lcviz/plugins/ephemeris/ephemeris.py b/lcviz/plugins/ephemeris/ephemeris.py
@@ -189,6 +189,12 @@ def _update_all_phase_arrays(self, *args, component=None):
             phases = _times_to_phases(times)
             if component not in self.phase_cids:
                 self.phase_cids[component] = ComponentID(phase_comp_lbl)
+
+            if self.phase_cids[component] in data.components:
+                data.update_components({self.phase_cids[component]: phases})
+            else:
+                data.add_component(phases, self.phase_cids[component])
+
             # this loop catches phase components generated automatically by
             # when add_results is triggered in other plugins:
             for comp in data.components: