docstr: Activate the training loop for the pl.LightningModule.
  
  Override this method for implementing a new algorithm.
docstr_tok: ["Activate","the","training","loop","for","the","pl.LightningModule",".","Override","this","method","for","implementing","a","new","algorithm","."]
code: def training_step(self, train_batch, batch_idx):
      """
      Activate the training loop for the pl.LightningModule.
  
      Override this method for implementing a new algorithm.
      """
      raise NotImplementedError
  
file_path: dowhy/dowhy/causal_prediction/algorithms/base_algorithm.py
imports: import pytorch_lightning
  import torch
  from torch.nn import functional
label_desc: Use image node_id 2 for calling the PredictionAlgorithm obj's underlying member method code with example usage: obj.training_step(train_batch, batch_idx) without return types