named_schedulers, get_scheduler, collections_to_dsk, get_collection_names, get_name_from_key, replace_name_in_key, clone_key, ) from dask.core import literal from dask.delayed import Delayed from dask.utils import tmpdir, tmpfile from dask.utils_test import dec, inc, import_or_none from dask.diagnostics import Profiler tz = pytest.importorskip("tlz") da = import_or_none("dask.array") db = import_or_none("dask.bag") dd = import_or_none("dask.dataframe") np = import_or_none("numpy") sp = import_or_none("scipy.sparse") pd = import_or_none("pandas") def f1(a, b, c=1): pass def f2(a, b=1, c=2): pass
import time from operator import add import pytest import dask import dask.bag as db from dask import delayed from dask.base import clone_key from dask.blockwise import Blockwise from dask.graph_manipulation import bind, checkpoint, chunks, clone, wait_on from dask.highlevelgraph import HighLevelGraph from dask.tests.test_base import Tuple from dask.utils_test import import_or_none da = import_or_none("dask.array") dd = import_or_none("dask.dataframe") pd = import_or_none("pandas") class NodeCounter: def __init__(self): self.n = 0 def f(self, x): time.sleep(random.random() / 100) self.n += 1 return x def assert_no_common_keys(a, b, omit=None, *, layers: bool) -> None: