Function get_or_none

axlearn/common/utils.py:1565–1566

(x: Optional[dict], key: Any)

                out_tree[k] = v
        in_tree = out_tree
    return in_tree


def non_empty_leaf_merge_fn(primary: Any, secondary: Any):
    """This function chooses the non-empty leaf. If both leaves are non-empty, an error