From 9dcd884eb62daad5424f0eb3f2c5e6361e3e695b Mon Sep 17 00:00:00 2001 From: Farrah Molina Date: Mon, 5 Aug 2024 11:30:30 -0500 Subject: [PATCH] Added brightness as a model prediction --- orbitize/system.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/orbitize/system.py b/orbitize/system.py index 14a92b66..c016b4ba 100644 --- 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)