Function _find_layer_containing_key

dask/highlevelgraph.py:37–41

(key)

    """Returns the dependencies between layers"""

    def _find_layer_containing_key(key):
        for k, v in layers.items():
            if key in v:
                return k
        raise RuntimeError(f"{key!r} not found")

    all_keys = {key for layer in layers.values() for key in layer}
    ret = {k: set() for k in layers}