Class CallableLazyImport

dask/layers.py:29–42

Function Wrapper for Lazy Importing. This Class should only be used when materializing a graph on a distributed scheduler.

class CallableLazyImport:
    """Function Wrapper for Lazy Importing.

    This Class should only be used when materializing a graph
    on a distributed scheduler.
    """

    def __init__(self, function_path):
        self.function_path = function_path

    def __call__(self, *args, **kwargs):
        from distributed.utils import import_term

        return import_term(self.function_path)(*args, **kwargs)


#