Maps a function over a nested dictionary.
(fn: Callable, tree: dict)
| 36 | |
| 37 | |
| 38 | def tree_map(fn: Callable, tree: dict) -> dict: |
| 39 | """Maps a function over a nested dictionary.""" |
| 40 | return {k: tree_map(fn, v) if isinstance(v, dict) else fn(v) for k, v in tree.items()} |
| 41 | |
| 42 | |
| 43 | def tree_map_with_key(fn: Callable, tree: dict, keys: Sequence = ()) -> dict: |
no outgoing calls
no test coverage detected