code: def get_vars_maybe_avg(namespace, var_names, training, polyak_decay):
      """utility for retrieving polyak averaged params"""
      vars = []
      for vn in var_names:
          vars.append(
              get_var_maybe_avg(namespace, vn, training, polyak_decay)
          )
      return vars
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  from torch.nn import Parameter
label_desc: Use image node_id 2 for calling a global function with example usage: get_vars_maybe_avg(namespace, var_names, training, polyak_decay) and returns: vars