Merge pull request #19 from lsst/tickets/OPSIM-1105

Tickets/opsim 1105
lsst · Jan 4, 2024 · 4d0b64e · 4d0b64e
2 parents 371ce16 + dc7fd36
commit 4d0b64e
Show file tree

Hide file tree

Showing 3 changed files with 201 additions and 165 deletions.
diff --git a/rubin_scheduler/scheduler/basis_functions/basis_functions.py b/rubin_scheduler/scheduler/basis_functions/basis_functions.py
@@ -874,13 +874,13 @@ def __init__(self, nside=None, max_airmass=2.5, penalty=np.nan):
 
     def _calc_value(self, conditions, indx=None):
         result = self.result.copy()
-        valid_airmass = np.isfinite(conditions.airmass)
+        valid_airmass = np.where(np.isfinite(conditions.airmass) == True)[0]
         good_pix = np.where(
             (conditions.airmass[valid_airmass] >= 1.0)
             & (IntRounded(conditions.airmass[valid_airmass]) < self.max_airmass)
             & (IntRounded(np.abs(conditions.az_to_sun[valid_airmass])) < IntRounded(np.pi / 2.0))
         )
-        result[valid_airmass][good_pix] = (
+        result[valid_airmass[good_pix]] = (
             conditions.airmass[valid_airmass][good_pix] / self.max_airmass.initial
         )
         return result