Skip to content

Commit

Permalink
Special-case component dataset types in execution reports.
Browse files Browse the repository at this point in the history
  • Loading branch information
TallJimbo committed Feb 20, 2024
1 parent a71a70a commit 269e678
Showing 1 changed file with 18 additions and 6 deletions.
24 changes: 18 additions & 6 deletions python/lsst/pipe/base/execution_reports.py
Original file line number Diff line number Diff line change
Expand Up @@ -359,12 +359,24 @@ def make_reports(
pipeline_dataset_types.intermediates,
pipeline_dataset_types.outputs,
):
refs[dataset_type.name] = {
ref.id: ref
for ref in butler.registry.queryDatasets(
dataset_type.name, collections=collection, findFirst=False
)
}
if dataset_type.component is not None:
# Work around the fact that component support has been phased
# out of daf_butler queries but not pipe_base's QGs. This
# should go away on DM-40441.
parent_dataset_type = dataset_type.makeCompositeDatasetType()
refs[dataset_type.name] = {
ref.id: ref.makeComponentRef(dataset_type.component)
for ref in butler.registry.queryDatasets(
parent_dataset_type.name, collections=collection, findFirst=False
)
}
else:
refs[dataset_type.name] = {
ref.id: ref
for ref in butler.registry.queryDatasets(
dataset_type.name, collections=collection, findFirst=False
)
}
for task_def in qg.iterTaskGraph():
for node in qg.getNodesForTask(task_def):
status_graph.add_node(node.nodeId)
Expand Down

0 comments on commit 269e678

Please sign in to comment.