Skip to content

Commit 658d5d6

Browse files
committed
Fix bitrot.
1 parent 9a8ba5b commit 658d5d6

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

bin.src/curate_templates.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,10 @@ def _make_parser():
4343
"repo",
4444
help="An existing data repository containing the input collections.",
4545
)
46+
parser.add_argument(
47+
"skymap",
48+
help="The skymap used for this batch of templates.",
49+
)
4650
parser.add_argument(
4751
"tag",
4852
help="A Jira ticket number for the new template collection name.",
@@ -60,7 +64,7 @@ def _make_parser():
6064
)
6165
parser.add_argument(
6266
"--where",
63-
default="",
67+
default="instrument='LSSTCam' AND skymap='lsst_cells_v2'",
6468
help="A string expression to select datasets in the input collections.",
6569
)
6670
parser.add_argument(
@@ -102,7 +106,7 @@ def select_ref(drefs, tract, patch, band, dtype="template_coadd"):
102106
return drefs[0]
103107

104108

105-
def make_threshold_cuts(butler, template_coadds, n_images, tracts, filter_by, cutoff):
109+
def make_threshold_cuts(butler, skymap, template_coadds, n_images, tracts, filter_by, cutoff):
106110
"""Select template_coadd and template_coadd_n_image datasets that pass a depth threshold.
107111
108112
Parameters
@@ -134,7 +138,7 @@ def make_threshold_cuts(butler, template_coadds, n_images, tracts, filter_by, cu
134138
rejected_drefs = []
135139

136140
for tract in tracts:
137-
coadd_depth_table = butler.get("template_coadd_depth_table", tract=tract)
141+
coadd_depth_table = butler.get("template_coadd_depth_table", skymap=skymap, tract=tract)
138142
mask = (coadd_depth_table[filter_by] > cutoff)
139143

140144
# --- Handle accepted patches/bands ---
@@ -345,9 +349,10 @@ def main():
345349

346350
# Filter out template_coads that don't meet the cutoff and save them to record.
347351
logging.info("Starting curation.")
348-
accepted_drefs, rejected_drefs, accepted_n_image_refs = make_threshold_cuts(butler, coadd_refs,
349-
n_image_refs, tracts,
350-
args.filter_by, args.cutoff
352+
accepted_drefs, rejected_drefs, accepted_n_image_refs = make_threshold_cuts(butler, args.skymap,
353+
coadd_refs, n_image_refs,
354+
tracts, args.filter_by,
355+
args.cutoff
351356
)
352357
logging.info(f"Curation complete. Accepted {len(accepted_drefs)} out of {len(coadd_refs)}"
353358
f" template_coadd datasets in {args.collections}.")

0 commit comments

Comments
 (0)