code: def restore_saved_modules(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, {})
  
imports: import importlib
  import logging
  import sys
  from mmf.common.registry import registry
  from packaging import version
label_desc: Use image node_id 3 for calling a global function with example usage: restore_saved_modules(caller_globals) without return types