Get the first layer from a HighLevelGraph whose name starts with a prefix
(hlg: HighLevelGraph, prefix: str)
| 139 | |
| 140 | |
| 141 | def hlg_layer(hlg: HighLevelGraph, prefix: str) -> Layer: |
| 142 | "Get the first layer from a HighLevelGraph whose name starts with a prefix" |
| 143 | for key, lyr in hlg.layers.items(): |
| 144 | if key.startswith(prefix): |
| 145 | return lyr |
| 146 | raise KeyError(f"No layer starts with {prefix!r}: {list(hlg.layers)}") |
| 147 | |
| 148 | |
| 149 | def hlg_layer_topological(hlg: HighLevelGraph, i: int) -> Layer: |
no test coverage detected