Class ExprScalar

dask/tests/test_composite_expr.py:73–104

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

    def __init__(self, expr):
        self._expr = expr

    @property
    def expr(self):
        return self._expr

    def __dask_graph__(self):
        return self.expr.__dask_graph__()

    def __dask_keys__(self):
        return self.expr.__dask_keys__()

    def __dask_layers__(self):
        return (self.expr._name,)

    def __dask_tokenize__(self):
        return self.expr

    def __dask_postcompute__(self):
        return _first, ()

    def __dask_postpersist__(self):
        return ExprScalar._rebuild, (self.expr,)

    @staticmethod
    def _rebuild(dsk, expr):
        return ExprScalar(LiteralExpr(expr.name, dsk[expr.__dask_keys__()[0]]))


@get_collection_type.register(CompositeScalarMeta)