fixed find_empty_masks

nanograv · Dec 10, 2024 · a8a811d · a8a811d
1 parent ee1852d
commit a8a811d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/CHANGELOG-unreleased.md b/CHANGELOG-unreleased.md
@@ -26,6 +26,7 @@ the released changes.
 - When EQUAD is created from TNEQ, has proper TCB->TDB conversion info
 - TOA selection masks will work when only TOA is the first one
 - Condense code in Glitch model and add test coverage.
+- `find_empty_masks` will now search through `CMX` parameters
 ### Removed
 - macOS 12 CI 
 
diff --git a/src/pint/models/timing_model.py b/src/pint/models/timing_model.py
@@ -109,6 +109,9 @@
 
 ignore_prefix = {"DMXF1_", "DMXF2_", "DMXEP_"}
 
+# prefixes of parameters that may need to be checked for empty ranges
+prefixes = ["DM", "SW", "CM"]
+
 DEFAULT_ORDER = [
     "astrometry",
     "jump_delay",
@@ -2901,7 +2904,7 @@ def find_empty_masks(self, toas, freeze=False):
                 if freeze:
                     log.info(f"'{maskpar}' has no TOAs so freezing")
                     getattr(self, maskpar).frozen = True
-        for prefix in ["DM", "SW"]:
+        for prefix in prefixes:
             mapping = pint.utils.xxxselections(self, toas, prefix=prefix)
             for k in mapping:
                 if len(mapping[k]) == 0: