Function recursive_dict_update

cdslib/core/utils/argparse_utils.py:195–242

Recursively update the content in `src_dict` into `tgt_dict`. For example, tgt_dict = { 'a': 1, 'b': {'c': 2}, 'd': 3, } src_dict = { 'a': 4, 'b': {'e': 5}, 'f': 6, } After the function tgt_dict will contain: tgt_dict

(
    tgt_dict: T.MutableMapping[T.Any, T.Any],
    src_dict: T.MutableMapping[T.Any, T.Any],
)

def recursive_dict_update(
    tgt_dict: T.MutableMapping[T.Any, T.Any],
    src_dict: T.MutableMapping[T.Any, T.Any],
):
    """
    Recursively update the content in `src_dict` into `tgt_dict`.

    For example,
    tgt_dict = {
        'a': 1,
        'b': {'c': 2},
        'd': 3,
    }
    src_dict = {
        'a': 4,
        'b': {'e': 5},
        'f': 6,
    }

    After the function tgt_dict will contain:
    tgt_dict = {
        'a': 4,
        'b': {'c': 2, 'e': 5},
        'd': 3,
        'f': 6,
    }
    Notice that the function does not replace tgt_dict['b'] with the new
    dictionary, but update it.

    Args:
        tgt_dict:
            dictionary to be updated
        src_dict:
            dictionary containing the source content
    """

    if src_dict is None:
        return

    for key, val in src_dict.items():
        if key not in tgt_dict or tgt_dict[key] is None or isinstance(tgt_dict[key], HyperParams.TBD):
            tgt_dict[key] = val
        else:
            if isinstance(val, (dict, HyperParams)):
                assert isinstance(tgt_dict[key], (dict, HyperParams)), f"{key}, {type(tgt_dict[key])}"
                recursive_dict_update(tgt_dict[key], src_dict[key])
            else:
                tgt_dict[key] = val