code: def test_augmentation_setup_update():
      def dummy_mag_to_param(magnitude):
          return magnitude + 1
  
      initial = {
          "mag_range": (0, 10),
          "randomly_negate": True,
          "mag_to_param": dummy_mag_to_param,
          "param_device": "gpu",
          "name": "some_other_dummy_name",
      }
  
      @augmentation
      def default_aug(data, _):
          return data
  
      defaults = {attr: getattr(default_aug, attr) for attr in initial}
      defaults["name"] = "dummy"
  
      @augmentation(**initial)
      def dummy(data, _):
          return data
  
      for reset_attr in initial:
          reset_attr_aug = dummy.augmentation(**{reset_attr: None})
          for attr in initial:
              reset = getattr(reset_attr_aug, attr)
              ref = (defaults if attr == reset_attr else initial)[attr]
              assert (
                  reset == ref
              ), f"{attr}: {reset}, {ref} ({reset_attr})"
  
imports: import numpy
  import nvidia.dali.tensors
  from nvidia.dali import pipeline_def, fn, types
  from nvidia.dali.auto_aug.core import augmentation, signed_bin
  from nvidia.dali.auto_aug.core._args import forbid_unused_kwargs
  from nose2.tools import params
  from test_utils import check_batch
  from nose_utils import assert_warns, assert_raises
label_desc: Use image node_id 9 for calling a global function with example usage: test_augmentation_setup_update() and returns: unknown, data, data