code: def forward(self, x: Any) -> Any:
      if isinstance(x, dict):
          x_ = {
              self._mapping.get(key, key): np.array(value)
              for key, value in x.items()
              if key in self._mapping
          }
      else:
          x_ = {"image": x}
      x_ = self.transform(**x_)
      if isinstance(x, dict):
          x.update(
              {
                  self._mapping_rev.get(k, k): x_[k]
                  for k in self._mapping_rev
                  if k in x_
              }
          )
          return x
  
      return x_["image"]
  
file_path: lightning-flash/src/flash/core/data/transforms.py
imports: from typing import Any, Mapping, Sequence, Union
  import numpy
  from torch import nn
  from flash.core.data.io.input import DataKeys
  from flash.core.data.utils import convert_to_modules
  from flash.core.utilities.imports import _ALBUMENTATIONS_AVAILABLE, requires
label_desc: Use image node_id 2 for calling the AlbumentationsAdapter obj's underlying member method code with example usage: obj.forward(x) and returns: x_, x