def update_max_depth(depth_node_dict):
    """
    Maximum tree depth update.

    Input:  - depth_node_dict: A map from node depth to node ids as a python dictionary.

    Output: - max_depth: The maximum depth of the tree.
    """
    max_depth = max(depth_node_dict.keys())
    return max_depth