code: def _setup_input_type(self):
      self._training_batch_type = None
      sig = inspect.signature(self.train_step_gen)
      assert "training_batch" in sig.parameters
      param = sig.parameters["training_batch"]
      annotation = param.annotation
      if annotation == inspect.Parameter.empty:
          return
      if hasattr(annotation, "from_dict"):
          self._training_batch_type = annotation
  
imports: import inspect
  import logging
  import pytorch_lightning
  import torch
  from reagent.core.tensorboardX import SummaryWriterContext
  from reagent.core.utils import lazy_property
  from typing_extensions import final
label_desc: Use image node_id 2 for calling the ReAgentLightningModule obj's underlying member method code with example usage: obj._setup_input_type() without return types