fix logic when selection mask is array([0])

nanograv · Nov 15, 2024 · 7c2bf80 · 7c2bf80
1 parent ab4a22e
commit 7c2bf80
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/pint/models/noise_model.py b/src/pint/models/noise_model.py
@@ -164,14 +164,14 @@ def scale_toa_sigma(self, toas, warn=True):
             if equad.quantity is None:
                 continue
             mask = equad.select_toa_mask(toas)
-            if np.any(mask):
+            if len(mask) > 0:
                 sigma_scaled[mask] = np.hypot(sigma_scaled[mask], equad.quantity)
             elif warn:
                 warnings.warn(f"EQUAD {equad} has no TOAs")
         for efac_name in self.EFACs:
             efac = getattr(self, efac_name)
             mask = efac.select_toa_mask(toas)
-            if np.any(mask):
+            if len(mask) > 0:
                 sigma_scaled[mask] *= efac.quantity
             elif warn:
                 warnings.warn(f"EFAC {efac} has no TOAs")