code: def current_transform(
      self, stage: RunningStage, current_fn: str
  ) -> Callable:
      return self._transform[stage].transforms.get(
          current_fn, self._identity
      )
  
imports: import inspect
  from dataclasses import dataclass
  from functools import partial
  from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple
  from pytorch_lightning.utilities import rank_zero_warn
  from pytorch_lightning.utilities.enums import LightningEnum
  from flash.core.data.callback import ControlFlow
  from flash.core.data.utilities.collate import default_collate
  from flash.core.data.utils import _STAGES_PREFIX
  from flash.core.utilities.stages import RunningStage
  from flash.core.utilities.types import INPUT_TRANSFORM_TYPE
label_desc: Use image node_id 2 for calling the InputTransform obj's underlying member method code with example usage: obj.current_transform(stage, current_fn) and returns: self