Function optimizer_with_annotations

dask/tests/test_hlgexpr.py:179–189

(dsk, keys)

def optimizer_with_annotations(dsk, keys):
    annots = {}
    for layer in dsk.layers.values():
        if layer.annotations:
            annots.update(layer.annotations)
    out = optimizer(dsk, keys)
    out = {f"{key}-optimized": val for key, val in out.items()}
    return HighLevelGraph(
        layers={"fused": MaterializedLayer(out, annotations=annots)},
        dependencies={"fused": set()},
    )


def test_hlg_sequence_uses_annotations_of_optimized_dsk():