Function _can_fuse_annotations

dask/blockwise.py:1211–1225

Treat the special annotation keys, as fusable since we can apply simple rules to capture their intent in a fused layer.

(a: dict | None, b: dict | None)

def _can_fuse_annotations(a: dict | None, b: dict | None) -> bool:
    """
    Treat the special annotation keys, as fusable since we can apply simple
    rules to capture their intent in a fused layer.
    """
    if a == b:
        return True

    if dask.config.get("optimization.annotations.fuse") is False:
        return False

    fusable = {"retries", "priority", "resources", "workers", "allow_other_workers"}
    return (not a or all(k in fusable for k in a)) and (
        not b or all(k in fusable for k in b)
    )


def _fuse_annotations(*args: dict) -> dict: