def METHOD_NAME(caller_globals):
    """
    Restore function for safecopy_modules()
    Reassigns current dictionary of 'module.function': function
    saved by safecopy_modules to callers modules.
    Assumes caller_globals is a dict from module name str to caller module obj.
    Example:
        restore_saved_modules({'BertSelfAttention': BertSelfAttention})
    """
    original_functions = registry.get(ORIGINAL_PATCH_FUNCTIONS_KEY)
    for module_function_name, function in original_functions.items():
        module_name, function_name = module_function_name.split(".")
        if module_name in caller_globals:
            setattr(caller_globals[module_name], function_name, function)
    registry.register(ORIGINAL_PATCH_FUNCTIONS_KEY, {})