code: def simple_annotation():
      def simple_wrapper(old_func):
          @functools.wraps(old_func)
          def new_func(*args, **kwargs):
              output = old_func(*args, **kwargs)
              return output
  
          return new_func
  
      return simple_wrapper
  
imports: import unittest
  import functools
  import torch
  import torch.nn.functional
  from nni.common.concrete_trace_utils import concrete_trace
  from nni.compression.pruning import L1NormPruner
  from nni.compression.speedup import ModelSpeedup
  from nni.compression.utils.counter import count_flops_params
label_desc: Use image node_id 1 for calling a global function with example usage: simple_annotation() and returns: simple_wrapper, new_func, output