docstr: Implement training step as generator here
docstr_tok: ["Implement","training","step","as","generator","here"]
code: def train_step_gen(self, training_batch, batch_idx: int):
      """
      Implement training step as generator here
      """
      raise NotImplementedError
  
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 7 for calling the ReAgentLightningModule obj's underlying member method code with example usage: obj.train_step_gen(training_batch, batch_idx) without return types