code: def get_adversarial_fn(adversarial_fn_name):
      if adversarial_fn_name == "advprop_pyramid":
          return get_adv_pyramid
      else:
          raise NotImplementedError(
              "No implementation for %s" % adversarial_fn_name
          )
  
file_path: scenic/scenic/projects/adversarialtraining/attacks/attack_compute.py
imports: import functools
  import jax
  import jax.numpy
  import ml_collections
  from scenic.projects.adversarialtraining.attacks import attack_losses
  from scenic.projects.adversarialtraining.attacks import attack_methods
  from scenic.projects.adversarialtraining.attacks import attack_transforms
  from scenic.train_lib import lr_schedules
  from scenic.train_lib import optimizers
label_desc: Use image node_id 3 for calling a global function with example usage: get_adversarial_fn(adversarial_fn_name) and returns: get_adv_pyramid