Modify getQuantaForTask to support extra log info

lsst · Oct 10, 2022 · 3e6f774 · 3e6f774
1 parent 33d0155
commit 3e6f774
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/doc/changes/DM-36145.feature.md b/doc/changes/DM-36145.feature.md
@@ -0,0 +1 @@
+* modifications to getQuantaForTask to support showing added additional quanta information in the logger
diff --git a/python/lsst/pipe/base/graph/graph.py b/python/lsst/pipe/base/graph/graph.py
@@ -440,7 +440,7 @@ def getQuantaForTask(self, taskDef: TaskDef) -> FrozenSet[Quantum]:
             The `set` of `Quantum` that is associated with the specified
             `TaskDef`.
         """
-        return frozenset(node.quantum for node in self._taskToQuantumNode[taskDef])
+        return frozenset(node.quantum for node in (self._taskToQuantumNode.get(taskDef) or tuple()))
 
     def getNodesForTask(self, taskDef: TaskDef) -> FrozenSet[QuantumNode]:
         """Return all the `QuantumNodes` associated with a `TaskDef`.