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: