dask__dask-10521

{
  "status": "ok",
  "stdout": "525:def get(\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "content": "def get(\n    key: str,\n    default: Any = no_default,\n    config: dict = config,\n    override_with: Any = no_default,\n) -> Any:\n    \"\"\"\n    Get elements from global config\n\n    If ``override_with`` is not None this value will be passed straight back.\n    Useful for getting kwarg defaults from Dask config.\n\n    Use '.' for nested access\n\n    Examples\n    --------\n    >>> from dask import config\n    >>> config.get('foo')  # doctest: +SKIP\n    {'x': 1, 'y': 2}\n\n    >>> config.get('foo.x')  # doctest: +SKIP\n    1\n\n    >>> config.get('foo.x.y', default=123)  # doctest: +SKIP\n    123\n\n    >>> config.get('foo.y', override_with=None)  # doctest: +SKIP\n    2\n\n    >>> config.get('foo.y', override_with=3)  # doctest: +SKIP\n    3\n\n    See Also\n    --------\n    dask.config.set\n    \"\"\"\n    if override_with is not no_default:\n        return override_with\n    keys = key.split(\".\")\n    result = config\n    for k in keys:\n        k = canonical_name(k, result)\n        try:\n            result = result[k]\n        except (TypeError, IndexError, KeyError):\n            if default is no_default:\n                raise\n            return default\n\n    return result\n\n\ndef pop(key: str, default: Any = no_default, config: dict = config) -> Any:\n    \"\"\"Like ``get``, but remove the element if found\n\n    See Also\n    --------\n    dask.config.get\n    dask.config.set\n    \"\"\"\n    keys = key.split(\".\")\n    result = config\n    for i, k in enumerate(keys):\n        k = canonical_name(k, result)\n        try:\n            if i == len(keys) - 1:\n                return result.pop(k)\n            else:\n                result = result[k]\n        except (TypeError, IndexError, KeyError):\n            if default is no_default:\n                raise\n            return default\n\n\ndef update_defaults(\n    new: Mapping, config: dict = config, defaults: list[Mapping] = defaults\n) -> None:\n    \"\"\"Add a new set of defaults to the configuration\n\n    It does two things:\n\n    1.  Add the defaults to a global collection to be used by refresh later\n    2.  Updates the global config with the new configuration.\n        Old values are prioritized over new ones, unless the current value\n        is the old default, in which case it's updated to the new default.\n    \"\"\"\n    current_defaults = merge(*defaults)\n    defaults.append(new)\n    update(config, new, priority=\"new-defaults\", defaults=current_defaults)\n\n\ndef expand_environment_variables(config: Any) -> Any:\n    \"\"\"Expand environment variables in a nested config dictionary\n\n    This function will recursively search through any nested dictionaries\n    and/or lists.\n\n    Parameters\n    ----------\n    config : dict, iterable, or str\n        Input object to search for environment variables\n\n    Returns\n    -------\n    config : same type as input\n\n    Examples\n    --------\n    >>> expand_environment_variables({'x': [1, 2, '$USER']})  # doctest: +SKIP\n    {'x': [1, 2, 'my-username']}\n    \"\"\"\n    if isinstance(config, Mapping):\n        return {k: expand_environment_variables(v) for k, v in config.items()}\n    elif isinstance(config, str):\n        return os.path.expandvars(config)\n    elif isinstance(config, (list, tuple, builtins.set)):\n        return type(config)(expand_environment_variables(v) for v in config)\n    else:\n        return config\n\n\n#: Mapping of {deprecated key: new key} for renamed keys, or {deprecated key: None} for\n#: removed keys. All deprecated keys must use '-' instead of '_'.\n#: This is used in three places:\n#: 1. In refresh(), which calls rename() to rename and warn upon loading\n#:    from ~/.config/dask.yaml, DASK_ env variables, etc.\n#: 2. in distributed/config.py and equivalent modules, where we perform additional\n#:    distributed-specific renames for the yaml/env config and enrich this dict\n#: 3. from individual calls to dask.config.set(), which internally invoke\n#     check_deprecations()\ndeprecations: dict[str, str | None] = {\n    \"fuse-ave-width\": \"optimization.fuse.ave-width\",\n    \"fuse-max-height\": \"optimization.fuse.max-height\",\n    \"fuse-max-width\": \"optimization.fuse.max-width\",\n    \"fuse-subgraphs\": \"optimization.fuse.subgraphs\",\n    \"fuse-rename-keys\": \"optimization.fuse.rename-keys\",\n    \"fuse-max-depth-new-edges\": \"optimization.fuse.max-depth-new-edges\",\n    # See https://github.com/dask/distributed/pull/4916\n    \"ucx.cuda-copy\": \"distributed.ucx.cuda_copy\",\n    \"ucx.tcp\": \"distributed.ucx.tcp\",\n    \"ucx.nvlink\": \"distributed.ucx.nvlink\",\n    \"ucx.infiniband\": \"distributed.ucx.infiniband\",\n    \"ucx.rdmacm\": \"distributed.ucx.rdmacm\",\n    \"ucx.net-devices\": \"distributed.ucx.net-devices\",\n    \"ucx.reuse-endpoints\": \"distributed.ucx.reuse-endpoints\",\n    \"rmm.pool-size\": \"distributed.rmm.pool-size\",\n    \"shuffle\": \"dataframe.shuffle.algorithm\",\n    \"array.rechunk-threshold\": \"array.rechunk.threshold\",\n    \"dataframe.shuffle.algorithm\": \"dataframe.shuffle.method\",",
  "start_line": 525,
  "end_line": 674,
  "total_lines": 802,
  "is_truncated": true
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "content": "from __future__ import annotations\n\nimport ast\nimport base64\nimport builtins  # Explicitly use builtins.set as 'set' will be shadowed by a function\nimport json\nimport os\nimport site\nimport sys\nimport threading\nimport warnings\nfrom collections.abc import Mapping, Sequence\nfrom typing import Any, Literal\n\nimport yaml\n\nfrom dask.typing import no_default\n\n\ndef _get_paths():\n    \"\"\"Get locations to search for YAML configuration files.\n\n    This logic exists as a separate function for testing purposes.\n    \"\"\"\n\n    paths = [\n        os.getenv(\"DASK_ROOT_CONFIG\", \"/etc/dask\"),\n        os.path.join(sys.prefix, \"etc\", \"dask\"),\n        *[os.path.join(prefix, \"etc\", \"dask\") for prefix in site.PREFIXES],\n        os.path.join(os.path.expanduser(\"~\"), \".config\", \"dask\"),\n    ]\n    if \"DASK_CONFIG\" in os.environ:\n        paths.append(os.environ[\"DASK_CONFIG\"])\n\n    # Remove duplicate paths while preserving ordering\n    paths = list(reversed(list(dict.fromkeys(reversed(paths)))))\n\n    return paths\n\n\npaths = _get_paths()\n\nif \"DASK_CONFIG\" in os.environ:\n    PATH = os.environ[\"DASK_CONFIG\"]\nelse:\n    PATH = os.path.join(os.path.expanduser(\"~\"), \".config\", \"dask\")\n\n\nconfig: dict = {}\nglobal_config = config  # alias\n\n\nconfig_lock = threading.Lock()\n\n\ndefaults: list[Mapping] = []\n\n\ndef canonical_name(k: str, config: dict) -> str:\n    \"\"\"Return the canonical name for a key.\n\n    Handles user choice of '-' or '_' conventions by standardizing on whichever\n    version was set first. If a key already exists in either hyphen or\n    underscore form, the existing version is the canonical name. If neither\n    version exists the original key is used as is.\n    \"\"\"\n    try:\n        if k in config:\n            return k\n    except TypeError:\n        # config is not a mapping, return the same name as provided\n        return k\n\n    altk = k.replace(\"_\", \"-\") if \"_\" in k else k.replace(\"-\", \"_\")\n\n    if altk in config:\n        return altk\n\n    return k\n\n\ndef update(\n    old: dict,\n    new: Mapping,\n    priority: Literal[\"old\", \"new\", \"new-defaults\"] = \"new\",\n    defaults: Mapping | None = None,\n) -> dict:\n    \"\"\"Update a nested dictionary with values from another\n\n    This is like dict.update except that it smoothly merges nested values\n\n    This operates in-place and modifies old\n\n    Parameters\n    ----------\n    priority: string {'old', 'new', 'new-defaults'}\n        If new (default) then the new dictionary has preference.\n        Otherwise the old dictionary does.\n        If 'new-defaults', a mapping should be given of the current defaults.\n        Only if a value in ``old`` matches the current default, it will be\n        updated with ``new``.\n\n    Examples\n    --------\n    >>> a = {'x': 1, 'y': {'a': 2}}\n    >>> b = {'x': 2, 'y': {'b': 3}}\n    >>> update(a, b)  # doctest: +SKIP\n    {'x': 2, 'y': {'a': 2, 'b': 3}}\n\n    >>> a = {'x': 1, 'y': {'a': 2}}\n    >>> b = {'x': 2, 'y': {'b': 3}}\n    >>> update(a, b, priority='old')  # doctest: +SKIP\n    {'x': 1, 'y': {'a': 2, 'b': 3}}\n\n    >>> d = {'x': 0, 'y': {'a': 2}}\n    >>> a = {'x': 1, 'y': {'a': 2}}\n    >>> b = {'x': 2, 'y': {'a': 3, 'b': 3}}\n    >>> update(a, b, priority='new-defaults', defaults=d)  # doctest: +SKIP\n    {'x': 1, 'y': {'a': 3, 'b': 3}}\n\n    See Also\n    --------\n    dask.config.merge\n    \"\"\"\n    for k, v in new.items():\n        k = canonical_name(k, old)\n\n        if isinstance(v, Mapping):\n            if k not in old or old[k] is None or not isinstance(old[k], dict):\n                old[k] = {}\n            update(\n                old[k],\n                v,\n                priority=priority,\n                defaults=defaults.get(k) if defaults else None,\n            )\n        else:\n            if (\n                priority == \"new\"\n                or k not in old\n                or (\n                    priority == \"new-defaults\"\n                    and defaults\n                    and k in defaults\n                    and defaults[k] == old[k]\n                )\n            ):\n                old[k] = v\n\n    return old",
  "start_line": 1,
  "end_line": 150,
  "total_lines": 802,
  "is_truncated": true
}

{
  "status": "ok",
  "filepath": "dask/typing.py",
  "content": "from __future__ import annotations\n\nimport abc\nfrom collections.abc import Callable, Hashable, Mapping, Sequence\nfrom enum import Enum\nfrom typing import (\n    TYPE_CHECKING,\n    Any,\n    Literal,\n    Protocol,\n    TypeVar,\n    Union,\n    runtime_checkable,\n)\n\nif TYPE_CHECKING:\n    # IPython import is relatively slow. Avoid if not necessary\n    from IPython.display import DisplayObject\n\n    # TODO import from typing (requires Python >=3.10)\n    from typing_extensions import TypeAlias\n\nCollType = TypeVar(\"CollType\", bound=\"DaskCollection\")\nCollType_co = TypeVar(\"CollType_co\", bound=\"DaskCollection\", covariant=True)\nPostComputeCallable = Callable\n\n\nKey: TypeAlias = Union[str, bytes, int, float, tuple[\"Key\", ...]]\nGraph: TypeAlias = Mapping[Key, Any]\n# Potentially nested list of Dask keys\nNestedKeys: TypeAlias = list[Union[Key, \"NestedKeys\"]]\n\n\nclass SchedulerGetCallable(Protocol):\n    \"\"\"Protocol defining the signature of a ``__dask_scheduler__`` callable.\"\"\"\n\n    def __call__(\n        self,\n        dsk: Graph,\n        keys: Sequence[Key] | Key,\n        **kwargs: Any,\n    ) -> Any:\n        \"\"\"Method called as the default scheduler for a collection.\n\n        Parameters\n        ----------\n        dsk :\n            The task graph.\n        keys :\n            Key(s) corresponding to the desired data.\n        **kwargs :\n            Additional arguments.\n\n        Returns\n        -------\n        Any\n            Result(s) associated with `keys`\n\n        \"\"\"\n        raise NotImplementedError(\"Inheriting class must implement this method.\")\n\n\nclass PostPersistCallable(Protocol[CollType_co]):\n    \"\"\"Protocol defining the signature of a ``__dask_postpersist__`` callable.\"\"\"\n\n    def __call__(\n        self,\n        dsk: Graph,\n        *args: Any,\n        rename: Mapping[str, str] | None = None,\n    ) -> CollType_co:\n        \"\"\"Method called to rebuild a persisted collection.\n\n        Parameters\n        ----------\n        dsk: Mapping\n            A mapping which contains at least the output keys returned\n            by __dask_keys__().\n        *args : Any\n            Additional optional arguments If no extra arguments are\n            necessary, it must be an empty tuple.\n        rename : Mapping[str, str], optional\n            If defined, it indicates that output keys may be changing\n            too; e.g. if the previous output of :meth:`__dask_keys__`\n            was ``[(\"a\", 0), (\"a\", 1)]``, after calling\n            ``rebuild(dsk, *extra_args, rename={\"a\": \"b\"})``\n            it must become ``[(\"b\", 0), (\"b\", 1)]``.\n            The ``rename`` mapping may not contain the collection\n            name(s); in such case the associated keys do not change.\n            It may contain replacements for unexpected names, which\n            must be ignored.\n\n        Returns\n        -------\n        Collection\n            An equivalent Dask collection with the same keys as\n            computed through a different graph.\n\n        \"\"\"\n        raise NotImplementedError(\"Inheriting class must implement this method.\")\n\n\n@runtime_checkable\nclass DaskCollection(Protocol):\n    \"\"\"Protocol defining the interface of a Dask collection.\"\"\"\n\n    @abc.abstractmethod\n    def __dask_graph__(self) -> Graph:\n        \"\"\"The Dask task graph.\n\n        The core Dask collections (Array, DataFrame, Bag, and Delayed)\n        use a :py:class:`~dask.highlevelgraph.HighLevelGraph` to\n        represent the collection task graph. It is also possible to\n        represent the task graph as a low level graph using a Python\n        dictionary.\n\n        Returns\n        -------\n        Mapping\n            The Dask task graph. If the instance returns a\n            :py:class:`dask.highlevelgraph.HighLevelGraph` then the\n            :py:func:`__dask_layers__` method must be implemented, as\n            defined by the :py:class:`~dask.typing.HLGDaskCollection`\n            protocol.\n\n        \"\"\"\n        raise NotImplementedError(\"Inheriting class must implement this method.\")\n\n    @abc.abstractmethod\n    def __dask_keys__(self) -> NestedKeys:\n        \"\"\"The output keys of the task graph.\n\n        Note that there are additional constraints on keys for a Dask\n        collection than those described in the :doc:`task graph\n        specification documentation <spec>`. These additional\n        constraints are described below.\n\n        All keys must either be non-empty strings or tuples where the first element is a\n        non-empty string, followed by zero or more arbitrary str, bytes, int, float, or\n        tuples thereof. The non-empty string is commonly known as the *collection name*.\n        All collections embedded in the dask package have exactly one name, but this is\n        not a requirement.\n\n        These are all valid outputs:\n\n        - ``[]``\n        - ``[\"x\", \"y\"]``\n        - ``[[(\"y\", \"a\", 0), (\"y\", \"a\", 1)], [(\"y\", \"b\", 0), (\"y\", \"b\", 1)]``\n\n        Returns",
  "start_line": 1,
  "end_line": 150,
  "total_lines": 439,
  "is_truncated": true
}

{
  "status": "ok",
  "stdout": "432:    no_default = \"NO_DEFAULT\"\n438:no_default = _NoDefault.no_default\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "filepath": "dask/typing.py",
  "content": "\n    Copied from pandas._libs.lib._NoDefault.\n\n    Usage\n    -----\n    from dask.typing import NoDefault, no_default\n\n    def f(x: int | None | NoDefault = no_default) -> int:\n        if x is no_default:\n            ...\n    \"\"\"\n\n    no_default = \"NO_DEFAULT\"\n\n    def __repr__(self) -> str:\n        return \"<no_default>\"\n\n\nno_default = _NoDefault.no_default\nNoDefault = Literal[_NoDefault.no_default]",
  "start_line": 420,
  "end_line": 439,
  "total_lines": 439,
  "is_truncated": false
}

{
  "status": "ok",
  "stdout": "Test 1: 128MiB (Expected: 128MiB)\nTest 2: None (Expected: 128MiB)\nTest 3: 3 (Expected: 3)\nTest 4: 128MiB (Expected: 128MiB)\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "occurrences": 1,
  "strategy": "exact",
  "diff": "--- a/dask/config.py\n+++ b/dask/config.py\n@@ -558,7 +558,7 @@\n     --------\n     dask.config.set\n     \"\"\"\n-    if override_with is not no_default:\n+    if override_with is not no_default and override_with is not None:\n         return override_with\n     keys = key.split(\".\")\n     result = config\n",
  "is_truncated": false
}

{
  "status": "ok",
  "stdout": "Test 1: 128MiB (Expected: 128MiB)\nTest 2: 128MiB (Expected: 128MiB)\nTest 3: 3 (Expected: 3)\nTest 4: 128MiB (Expected: 128MiB)\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "error",
  "error_type": "CommandError",
  "error_message": "\u001b[1m============================= test session starts ==============================\u001b[0m\ncollected 56 items\n\ndask/tests/test_config.py \u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31mF\u001b[0m\u001b[31mF\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31m [ 82%]\n\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31m                                                               [100%]\u001b[0m\n\n=================================== FAILURES ===================================\n\u001b[31m\u001b[1m________________ test_collect_yaml_permission_errors[directory] ________________\u001b[0m\n\u001b[1m\u001b[31m/testbed/dask/tests/test_config.py\u001b[0m:189: in test_collect_yaml_permission_errors\n    \u001b[0m\u001b[94massert\u001b[39;49;00m config == expected\u001b[90m\u001b[39;49;00m\n\u001b[1m\u001b[31mE   AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {}\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m\n\u001b[1m\u001b[31mE     Left contains 3 more items:\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m{\u001b[33m'\u001b[39;49;00m\u001b[33mx\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m1\u001b[39;49;00m, \u001b[33m'\u001b[39;49;00m\u001b[33my\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m3\u001b[39;49;00m, \u001b[33m'\u001b[39;49;00m\u001b[33mz\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m4\u001b[39;49;00m}\u001b[90m\u001b[39;49;00m\u001b[0m\n\u001b[1m\u001b[31mE     Use -v to get more diff\u001b[0m\n\u001b[31m\u001b[1m__________________ test_collect_yaml_permission_errors[file] ___________________\u001b[0m\n\u001b[1m\u001b[31m/testbed/dask/tests/test_config.py\u001b[0m:189: in test_collect_yaml_permission_errors\n    \u001b[0m\u001b[94massert\u001b[39;49;00m config == expected\u001b[90m\u001b[39;49;00m\n\u001b[1m\u001b[31mE   AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {'y': 3, 'z': 4}\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m\n\u001b[1m\u001b[31mE     Omitting 2 identical items, use -vv to show\u001b[0m\n\u001b[1m\u001b[31mE     Left contains 1 more item:\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m{\u001b[33m'\u001b[39;49;00m\u001b[33mx\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m1\u001b[39;49;00m}\u001b[90m\u001b[39;49;00m\u001b[0m\n\u001b[1m\u001b[31mE     Use -v to get more diff\u001b[0m\n\u001b[33m=============================== warnings summary ===============================\u001b[0m\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.bag.utils\n    pytest.register_assert_rewrite(\n\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.array.utils\n    pytest.register_assert_rewrite(\n\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.dataframe.utils\n    pytest.register_assert_rewrite(\n\n-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html\n============================= slowest 10 durations =============================\n0.07s call     dask/tests/test_config.py::test_schema\n0.04s call     dask/tests/test_config.py::test_collect_yaml_permission_errors[directory]\n0.01s call     dask/tests/test_config.py::test_schema_is_complete\n\n(7 durations < 0.005s hidden.  Use -vv to show these durations.)\n\u001b[36m\u001b[1m=========================== short test summary info ============================\u001b[0m\n\u001b[31mFAILED\u001b[0m dask/tests/test_config.py::\u001b[1mtest_collect_yaml_permission_errors[directory]\u001b[0m - AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {}\n\u001b[31mFAILED\u001b[0m dask/tests/test_config.py::\u001b[1mtest_collect_yaml_permission_errors[file]\u001b[0m - AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {'y': 3, 'z': 4}\n\u001b[31m=================== \u001b[31m\u001b[1m2 failed\u001b[0m, \u001b[32m54 passed\u001b[0m, \u001b[33m3 warnings\u001b[0m\u001b[31m in 0.31s\u001b[0m\u001b[31m ===================\u001b[0m\n",
  "details": {
    "stdout": "\u001b[1m============================= test session starts ==============================\u001b[0m\ncollected 56 items\n\ndask/tests/test_config.py \u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31mF\u001b[0m\u001b[31mF\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31m [ 82%]\n\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31m                                                               [100%]\u001b[0m\n\n=================================== FAILURES ===================================\n\u001b[31m\u001b[1m________________ test_collect_yaml_permission_errors[directory] ________________\u001b[0m\n\u001b[1m\u001b[31m/testbed/dask/tests/test_config.py\u001b[0m:189: in test_collect_yaml_permission_errors\n    \u001b[0m\u001b[94massert\u001b[39;49;00m config == expected\u001b[90m\u001b[39;49;00m\n\u001b[1m\u001b[31mE   AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {}\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m\n\u001b[1m\u001b[31mE     Left contains 3 more items:\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m{\u001b[33m'\u001b[39;49;00m\u001b[33mx\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m1\u001b[39;49;00m, \u001b[33m'\u001b[39;49;00m\u001b[33my\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m3\u001b[39;49;00m, \u001b[33m'\u001b[39;49;00m\u001b[33mz\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m4\u001b[39;49;00m}\u001b[90m\u001b[39;49;00m\u001b[0m\n\u001b[1m\u001b[31mE     Use -v to get more diff\u001b[0m\n\u001b[31m\u001b[1m__________________ test_collect_yaml_permission_errors[file] ___________________\u001b[0m\n\u001b[1m\u001b[31m/testbed/dask/tests/test_config.py\u001b[0m:189: in test_collect_yaml_permission_errors\n    \u001b[0m\u001b[94massert\u001b[39;49;00m config == expected\u001b[90m\u001b[39;49;00m\n\u001b[1m\u001b[31mE   AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {'y': 3, 'z': 4}\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m\n\u001b[1m\u001b[31mE     Omitting 2 identical items, use -vv to show\u001b[0m\n\u001b[1m\u001b[31mE     Left contains 1 more item:\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m{\u001b[33m'\u001b[39;49;00m\u001b[33mx\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m1\u001b[39;49;00m}\u001b[90m\u001b[39;49;00m\u001b[0m\n\u001b[1m\u001b[31mE     Use -v to get more diff\u001b[0m\n\u001b[33m=============================== warnings summary ===============================\u001b[0m\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.bag.utils\n    pytest.register_assert_rewrite(\n\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.array.utils\n    pytest.register_assert_rewrite(\n\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.dataframe.utils\n    pytest.register_assert_rewrite(\n\n-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html\n============================= slowest 10 durations =============================\n0.07s call     dask/tests/test_config.py::test_schema\n0.04s call     dask/tests/test_config.py::test_collect_yaml_permission_errors[directory]\n0.01s call     dask/tests/test_config.py::test_schema_is_complete\n\n(7 durations < 0.005s hidden.  Use -vv to show these durations.)\n\u001b[36m\u001b[1m=========================== short test summary info ============================\u001b[0m\n\u001b[31mFAILED\u001b[0m dask/tests/test_config.py::\u001b[1mtest_collect_yaml_permission_errors[directory]\u001b[0m - AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {}\n\u001b[31mFAILED\u001b[0m dask/tests/test_config.py::\u001b[1mtest_collect_yaml_permission_errors[file]\u001b[0m - AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {'y': 3, 'z': 4}\n\u001b[31m=================== \u001b[31m\u001b[1m2 failed\u001b[0m, \u001b[32m54 passed\u001b[0m, \u001b[33m3 warnings\u001b[0m\u001b[31m in 0.31s\u001b[0m\u001b[31m ===================\u001b[0m\n",
    "stderr": "",
    "exit_code": 1
  }
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "occurrences": 1,
  "strategy": "exact",
  "diff": "--- a/dask/config.py\n+++ b/dask/config.py\n@@ -558,7 +558,7 @@\n     --------\n     dask.config.set\n     \"\"\"\n-    if override_with is not no_default and override_with is not None:\n+    if override_with is not no_default:\n         return override_with\n     keys = key.split(\".\")\n     result = config\n",
  "is_truncated": false
}

{
  "status": "error",
  "error_type": "CommandError",
  "error_message": "\u001b[1m============================= test session starts ==============================\u001b[0m\ncollected 56 items\n\ndask/tests/test_config.py \u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31mF\u001b[0m\u001b[31mF\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31m [ 82%]\n\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31m                                                               [100%]\u001b[0m\n\n=================================== FAILURES ===================================\n\u001b[31m\u001b[1m________________ test_collect_yaml_permission_errors[directory] ________________\u001b[0m\n\u001b[1m\u001b[31m/testbed/dask/tests/test_config.py\u001b[0m:189: in test_collect_yaml_permission_errors\n    \u001b[0m\u001b[94massert\u001b[39;49;00m config == expected\u001b[90m\u001b[39;49;00m\n\u001b[1m\u001b[31mE   AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {}\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m\n\u001b[1m\u001b[31mE     Left contains 3 more items:\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m{\u001b[33m'\u001b[39;49;00m\u001b[33mx\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m1\u001b[39;49;00m, \u001b[33m'\u001b[39;49;00m\u001b[33my\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m3\u001b[39;49;00m, \u001b[33m'\u001b[39;49;00m\u001b[33mz\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m4\u001b[39;49;00m}\u001b[90m\u001b[39;49;00m\u001b[0m\n\u001b[1m\u001b[31mE     Use -v to get more diff\u001b[0m\n\u001b[31m\u001b[1m__________________ test_collect_yaml_permission_errors[file] ___________________\u001b[0m\n\u001b[1m\u001b[31m/testbed/dask/tests/test_config.py\u001b[0m:189: in test_collect_yaml_permission_errors\n    \u001b[0m\u001b[94massert\u001b[39;49;00m config == expected\u001b[90m\u001b[39;49;00m\n\u001b[1m\u001b[31mE   AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {'y': 3, 'z': 4}\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m\n\u001b[1m\u001b[31mE     Omitting 2 identical items, use -vv to show\u001b[0m\n\u001b[1m\u001b[31mE     Left contains 1 more item:\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m{\u001b[33m'\u001b[39;49;00m\u001b[33mx\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m1\u001b[39;49;00m}\u001b[90m\u001b[39;49;00m\u001b[0m\n\u001b[1m\u001b[31mE     Use -v to get more diff\u001b[0m\n\u001b[33m=============================== warnings summary ===============================\u001b[0m\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.bag.utils\n    pytest.register_assert_rewrite(\n\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.dataframe.utils\n    pytest.register_assert_rewrite(\n\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.array.utils\n    pytest.register_assert_rewrite(\n\n-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html\n============================= slowest 10 durations =============================\n0.04s call     dask/tests/test_config.py::test_schema\n0.02s call     dask/tests/test_config.py::test_collect_yaml_permission_errors[directory]\n0.01s call     dask/tests/test_config.py::test_schema_is_complete\n\n(7 durations < 0.005s hidden.  Use -vv to show these durations.)\n\u001b[36m\u001b[1m=========================== short test summary info ============================\u001b[0m\n\u001b[31mFAILED\u001b[0m dask/tests/test_config.py::\u001b[1mtest_collect_yaml_permission_errors[directory]\u001b[0m - AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {}\n\u001b[31mFAILED\u001b[0m dask/tests/test_config.py::\u001b[1mtest_collect_yaml_permission_errors[file]\u001b[0m - AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {'y': 3, 'z': 4}\n\u001b[31m=================== \u001b[31m\u001b[1m2 failed\u001b[0m, \u001b[32m54 passed\u001b[0m, \u001b[33m3 warnings\u001b[0m\u001b[31m in 0.18s\u001b[0m\u001b[31m ===================\u001b[0m\n",
  "details": {
    "stdout": "\u001b[1m============================= test session starts ==============================\u001b[0m\ncollected 56 items\n\ndask/tests/test_config.py \u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31mF\u001b[0m\u001b[31mF\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31m [ 82%]\n\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[32m.\u001b[0m\u001b[31m                                                               [100%]\u001b[0m\n\n=================================== FAILURES ===================================\n\u001b[31m\u001b[1m________________ test_collect_yaml_permission_errors[directory] ________________\u001b[0m\n\u001b[1m\u001b[31m/testbed/dask/tests/test_config.py\u001b[0m:189: in test_collect_yaml_permission_errors\n    \u001b[0m\u001b[94massert\u001b[39;49;00m config == expected\u001b[90m\u001b[39;49;00m\n\u001b[1m\u001b[31mE   AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {}\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m\n\u001b[1m\u001b[31mE     Left contains 3 more items:\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m{\u001b[33m'\u001b[39;49;00m\u001b[33mx\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m1\u001b[39;49;00m, \u001b[33m'\u001b[39;49;00m\u001b[33my\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m3\u001b[39;49;00m, \u001b[33m'\u001b[39;49;00m\u001b[33mz\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m4\u001b[39;49;00m}\u001b[90m\u001b[39;49;00m\u001b[0m\n\u001b[1m\u001b[31mE     Use -v to get more diff\u001b[0m\n\u001b[31m\u001b[1m__________________ test_collect_yaml_permission_errors[file] ___________________\u001b[0m\n\u001b[1m\u001b[31m/testbed/dask/tests/test_config.py\u001b[0m:189: in test_collect_yaml_permission_errors\n    \u001b[0m\u001b[94massert\u001b[39;49;00m config == expected\u001b[90m\u001b[39;49;00m\n\u001b[1m\u001b[31mE   AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {'y': 3, 'z': 4}\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m\n\u001b[1m\u001b[31mE     Omitting 2 identical items, use -vv to show\u001b[0m\n\u001b[1m\u001b[31mE     Left contains 1 more item:\u001b[0m\n\u001b[1m\u001b[31mE     \u001b[0m{\u001b[33m'\u001b[39;49;00m\u001b[33mx\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m: \u001b[94m1\u001b[39;49;00m}\u001b[90m\u001b[39;49;00m\u001b[0m\n\u001b[1m\u001b[31mE     Use -v to get more diff\u001b[0m\n\u001b[33m=============================== warnings summary ===============================\u001b[0m\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.bag.utils\n    pytest.register_assert_rewrite(\n\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.dataframe.utils\n    pytest.register_assert_rewrite(\n\nconftest.py:90\n  /workspace/conftest.py:90: PytestAssertRewriteWarning: Module already imported so cannot be rewritten: dask.array.utils\n    pytest.register_assert_rewrite(\n\n-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html\n============================= slowest 10 durations =============================\n0.04s call     dask/tests/test_config.py::test_schema\n0.02s call     dask/tests/test_config.py::test_collect_yaml_permission_errors[directory]\n0.01s call     dask/tests/test_config.py::test_schema_is_complete\n\n(7 durations < 0.005s hidden.  Use -vv to show these durations.)\n\u001b[36m\u001b[1m=========================== short test summary info ============================\u001b[0m\n\u001b[31mFAILED\u001b[0m dask/tests/test_config.py::\u001b[1mtest_collect_yaml_permission_errors[directory]\u001b[0m - AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {}\n\u001b[31mFAILED\u001b[0m dask/tests/test_config.py::\u001b[1mtest_collect_yaml_permission_errors[file]\u001b[0m - AssertionError: assert {'x': 1, 'y': 3, 'z': 4} == {'y': 3, 'z': 4}\n\u001b[31m=================== \u001b[31m\u001b[1m2 failed\u001b[0m, \u001b[32m54 passed\u001b[0m, \u001b[33m3 warnings\u001b[0m\u001b[31m in 0.18s\u001b[0m\u001b[31m ===================\u001b[0m\n",
    "stderr": "",
    "exit_code": 1
  }
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "occurrences": 1,
  "strategy": "exact",
  "diff": "--- a/dask/config.py\n+++ b/dask/config.py\n@@ -558,7 +558,7 @@\n     --------\n     dask.config.set\n     \"\"\"\n-    if override_with is not no_default:\n+    if override_with is not no_default and override_with is not None:\n         return override_with\n     keys = key.split(\".\")\n     result = config\n",
  "is_truncated": false
}

{
  "status": "ok",
  "stdout": "FAILED: 'str' object has no attribute 'items'\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "content": "\n    >>> config.get('foo.y', override_with=None)  # doctest: +SKIP\n    2\n\n    >>> config.get('foo.y', override_with=3)  # doctest: +SKIP\n    3\n\n    See Also\n    --------\n    dask.config.set\n    \"\"\"\n    if override_with is not no_default and override_with is not None:\n        return override_with\n    keys = key.split(\".\")\n    result = config\n    for k in keys:\n        k = canonical_name(k, result)\n        try:\n            result = result[k]\n        except (TypeError, IndexError, KeyError):\n            if default is no_default:\n                raise\n            return default\n\n    return result\n\n\ndef pop(key: str, default: Any = no_default, config: dict = config) -> Any:\n    \"\"\"Like ``get``, but remove the element if found\n\n    See Also\n    --------\n    dask.config.get\n    dask.config.set\n    \"\"\"\n    keys = key.split(\".\")\n    result = config\n    for i, k in enumerate(keys):\n        k = canonical_name(k, result)\n        try:\n            if i == len(keys) - 1:\n                return result.pop(k)\n            else:\n                result = result[k]\n        except (TypeError, IndexError, KeyError):\n            if default is no_default:\n                raise\n            return default\n\n\ndef update_defaults(",
  "start_line": 550,
  "end_line": 600,
  "total_lines": 802,
  "is_truncated": false
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "content": "    --------\n    >>> from dask import config\n    >>> config.get('foo')  # doctest: +SKIP\n    {'x': 1, 'y': 2}\n\n    >>> config.get('foo.x')  # doctest: +SKIP\n    1\n\n    >>> config.get('foo.x.y', default=123)  # doctest: +SKIP\n    123\n\n    >>> config.get('foo.y', override_with=None)  # doctest: +SKIP\n    2\n\n    >>> config.get('foo.y', override_with=3)  # doctest: +SKIP\n    3\n\n    See Also\n    --------\n    dask.config.set\n    \"\"\"",
  "start_line": 540,
  "end_line": 560,
  "total_lines": 802,
  "is_truncated": false
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "content": "\n    >>> config.get('foo.y', override_with=None)  # doctest: +SKIP\n    2\n\n    >>> config.get('foo.y', override_with=3)  # doctest: +SKIP\n    3",
  "start_line": 550,
  "end_line": 555,
  "total_lines": 802,
  "is_truncated": false
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "content": ") -> Any:\n    \"\"\"\n    Get elements from global config\n\n    If ``override_with`` is not None this value will be passed straight back.\n    Useful for getting kwarg defaults from Dask config.\n\n    Use '.' for nested access\n\n    Examples\n    --------\n    >>> from dask import config\n    >>> config.get('foo')  # doctest: +SKIP\n    {'x': 1, 'y': 2}\n\n    >>> config.get('foo.x')  # doctest: +SKIP\n    1\n\n    >>> config.get('foo.x.y', default=123)  # doctest: +SKIP\n    123\n",
  "start_line": 530,
  "end_line": 550,
  "total_lines": 802,
  "is_truncated": false
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "content": "\n    update(config, collect(**kwargs))\n    rename(deprecations, config)\n\n\ndef get(\n    key: str,\n    default: Any = no_default,\n    config: dict = config,\n    override_with: Any = no_default,\n) -> Any:",
  "start_line": 520,
  "end_line": 530,
  "total_lines": 802,
  "is_truncated": false
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "content": "        is the old default, in which case it's updated to the new default.\n    \"\"\"\n    current_defaults = merge(*defaults)\n    defaults.append(new)\n    update(config, new, priority=\"new-defaults\", defaults=current_defaults)\n\n\ndef expand_environment_variables(config: Any) -> Any:\n    \"\"\"Expand environment variables in a nested config dictionary\n\n    This function will recursively search through any nested dictionaries\n    and/or lists.\n\n    Parameters\n    ----------\n    config : dict, iterable, or str\n        Input object to search for environment variables\n\n    Returns\n    -------\n    config : same type as input",
  "start_line": 610,
  "end_line": 630,
  "total_lines": 802,
  "is_truncated": false,
  "budget_warning": "Only 10 tool call(s) remaining (20/30 used). Finalize your edits and call submit_patch soon."
}

{
  "status": "error",
  "error_type": "CommandError",
  "error_message": "",
  "details": {
    "stdout": "",
    "stderr": "",
    "exit_code": 1
  },
  "budget_warning": "Only 9 tool call(s) remaining (21/30 used). Finalize your edits and call submit_patch soon."
}

{
  "status": "ok",
  "stdout": "269:    set(d, config=result)\n355:    >>> with dask.config.set({'foo.bar': 123}):\n360:    >>> with dask.config.set(foo__bar=123):\n365:    >>> dask.config.set(foo__bar=123)  # doctest: +SKIP\n654:#: 3. from individual calls to dask.config.set(), which internally invoke\n699:            set({new: value}, config=config)\n",
  "stderr": "",
  "exit_code": 0,
  "budget_warning": "Only 8 tool call(s) remaining (22/30 used). Finalize your edits and call submit_patch soon."
}

{
  "status": "ok",
  "stdout": "def _get_paths():\ndef canonical_name(k: str, config: dict) -> str:\ndef update(\ndef merge(*dicts: Mapping) -> dict:\ndef _load_config_file(path: str) -> dict | None:\ndef collect_yaml(paths: Sequence[str] = paths) -> list[dict]:\ndef collect_env(env: Mapping[str, str] | None = None) -> dict:\ndef ensure_file(\n    def __init__(\n    def __enter__(self):\n    def __exit__(self, type, value, traceback):\n    def _assign(\ndef collect(paths: list[str] = paths, env: Mapping[str, str] | None = None) -> dict:\ndef refresh(\ndef get(\ndef pop(key: str, default: Any = no_default, config: dict = config) -> Any:\ndef update_defaults(\ndef expand_environment_variables(config: Any) -> Any:\ndef rename(\ndef check_deprecations(\ndef serialize(data: Any) -> str:\ndef deserialize(data: str) -> Any:\ndef _initialize() -> None:\n",
  "stderr": "",
  "exit_code": 0,
  "budget_warning": "Only 7 tool call(s) remaining (23/30 used). Finalize your edits and call submit_patch soon."
}

{
  "status": "ok",
  "filepath": "dask/config.py",
  "content": "from __future__ import annotations\n\nimport ast\nimport base64\nimport builtins  # Explicitly use builtins.set as 'set' will be shadowed by a function\nimport json\nimport os\nimport site\nimport sys\nimport threading\nimport warnings\nfrom collections.abc import Mapping, Sequence\nfrom typing import Any, Literal\n\nimport yaml\n\nfrom dask.typing import no_default\n\n\ndef _get_paths():\n    \"\"\"Get locations to search for YAML configuration files.\n\n    This logic exists as a separate function for testing purposes.\n    \"\"\"\n\n    paths = [\n        os.getenv(\"DASK_ROOT_CONFIG\", \"/etc/dask\"),\n        os.path.join(sys.prefix, \"etc\", \"dask\"),\n        *[os.path.join(prefix, \"etc\", \"dask\") for prefix in site.PREFIXES],\n        os.path.join(os.path.expanduser(\"~\"), \".config\", \"dask\"),\n    ]\n    if \"DASK_CONFIG\" in os.environ:\n        paths.append(os.environ[\"DASK_CONFIG\"])\n\n    # Remove duplicate paths while preserving ordering\n    paths = list(reversed(list(dict.fromkeys(reversed(paths)))))\n\n    return paths\n\n\npaths = _get_paths()\n\nif \"DASK_CONFIG\" in os.environ:\n    PATH = os.environ[\"DASK_CONFIG\"]\nelse:\n    PATH = os.path.join(os.path.expanduser(\"~\"), \".config\", \"dask\")\n\n\nconfig: dict = {}\nglobal_config = config  # alias\n\n\nconfig_lock = threading.Lock()\n\n\ndefaults: list[Mapping] = []\n\n\ndef canonical_name(k: str, config: dict) -> str:\n    \"\"\"Return the canonical name for a key.\n\n    Handles user choice of '-' or '_' conventions by standardizing on whichever\n    version was set first. If a key already exists in either hyphen or\n    underscore form, the existing version is the canonical name. If neither\n    version exists the original key is used as is.\n    \"\"\"\n    try:\n        if k in config:\n            return k\n    except TypeError:\n        # config is not a mapping, return the same name as provided\n        return k\n\n    altk = k.replace(\"_\", \"-\") if \"_\" in k else k.replace(\"-\", \"_\")\n\n    if altk in config:\n        return altk\n\n    return k\n\n\ndef update(\n    old: dict,\n    new: Mapping,\n    priority: Literal[\"old\", \"new\", \"new-defaults\"] = \"new\",\n    defaults: Mapping | None = None,\n) -> dict:\n    \"\"\"Update a nested dictionary with values from another\n\n    This is like dict.update except that it smoothly merges nested values\n\n    This operates in-place and modifies old\n\n    Parameters\n    ----------\n    priority: string {'old', 'new', 'new-defaults'}\n        If new (default) then the new dictionary has preference.\n        Otherwise the old dictionary does.\n        If 'new-defaults', a mapping should be given of the current defaults.\n        Only if a value in ``old`` matches the current default, it will be",
  "start_line": 1,
  "end_line": 100,
  "total_lines": 802,
  "is_truncated": false,
  "budget_warning": "Only 6 tool call(s) remaining (24/30 used). Finalize your edits and call submit_patch soon."
}

{
  "status": "error",
  "error_type": "CommandError",
  "error_message": "",
  "details": {
    "stdout": "",
    "stderr": "",
    "exit_code": 1
  },
  "budget_warning": "Only 5 tool call(s) remaining (25/30 used). Finalize your edits and call submit_patch soon."
}