Added brightness as a model prediction

sblunt · Aug 5, 2024 · 9dcd884 · 9dcd884
1 parent 37bbb19
commit 9dcd884
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/orbitize/system.py b/orbitize/system.py
@@ -667,6 +667,10 @@ def compute_model(self, params_arr, use_rebound=False):
                 model[self.rv[body_num], 1] = np.nan
 
             # TODO (farrah): add brightness to model here (use RV block above as template)
+            # Brightness
+            if len(self.brightness[body_num]) > 0:
+                model[self.brightness[body_num], 0] = brightness[self.brightness[body_num], body_num, :]
+                model[self.brightness[body_num], 1] = np.nan
             # (assume self.brightness is array of indices of epochs with brightness measurements)