Ensure add_bright_limits_columns adds all columns

sdss · May 6, 2024 · 808fc34 · 808fc34
1 parent 9b54c2e
commit 808fc34
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/too/validate.py b/src/too/validate.py
@@ -807,6 +807,12 @@ def add_bright_limits_columns(
     # Get a list of design modes.
     design_modes: dict[str, DesignMode] = allDesignModes(database)
 
+    # Add all columns for all design modes and set to false for now.
+    targets = targets.with_columns(
+        **{f"bn_{dm}_valid": polars.lit(False) for dm in design_modes},
+        **{f"mag_lim_{dm}_valid": polars.lit(False) for dm in design_modes},
+    )
+
     processed: list[polars.DataFrame] = []
 
     # First we split the targets by observatory and sky_brightness_mode (bright or