Class ExprTuple

dask/tests/test_composite_expr.py:120–153

class ExprTuple(DaskMethodsMixin):
    __dask_scheduler__ = staticmethod(dask.threaded.get)
    __dask_optimize__ = None

    def __init__(self, *children):
        self.children = tuple(children)

    def __dask_exprs__(self):
        return tuple(child.expr for child in self.children)

    def __dask_rebuild_from_exprs__(self, exprs):
        return ExprTuple(*(ExprScalar(expr) for expr in exprs))

    def __dask_graph__(self):
        return merge(*(child.__dask_graph__() for child in self.children))

    def __dask_keys__(self):
        return [child.__dask_keys__() for child in self.children]

    def __dask_layers__(self):
        return tuple(child.expr._name for child in self.children)

    def __dask_tokenize__(self):
        return self.children

    def __dask_postcompute__(self):
        return _finalize_expr_tuple, ()

    def __dask_postpersist__(self):
        return ExprTuple._rebuild, (self.children,)

    @staticmethod
    def _rebuild(dsk, children):
        return ExprTuple(*(ExprScalar._rebuild(dsk, child.expr) for child in children))


def test_collections_to_expr_uses_composite_protocol():