Add additional quanta information for pipetask run

lsst · Oct 10, 2022 · 7b56b1b · 7b56b1b
1 parent dba8530
commit 7b56b1b
Showing 1 changed file with 1 addition and 1 deletion.
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`.