code: def register_augmentation_config(
      name: str, features: Union[str, List[str]]
  ):
      if isinstance(features, str):
          features = [features]
  
      def wrap(cls):
          for feature in features:
              augmentation_registry = (
                  get_augmentation_config_registry().get(feature, {})
              )
              augmentation_registry[name] = cls
              get_augmentation_config_registry()[
                  feature
              ] = augmentation_registry
          return cls
  
      return wrap
  
file_path: ludwig/ludwig/schema/features/augmentation/utils.py
imports: import copy
  from dataclasses import field
  from typing import Any, Dict, List, Optional, Union
  from marshmallow import fields, ValidationError
  from ludwig.api_annotations import DeveloperAPI
  from ludwig.constants import TYPE
  from ludwig.schema import utils
  from ludwig.schema.features.augmentation.base import BaseAugmentationConfig
  from ludwig.utils.registry import Registry
label_desc: Use image node_id 2 for calling a global function with example usage: register_augmentation_config(name, features) and returns: wrap, cls