code: def train_batch(
      self, batch: TorchData, epoch_idx: int, batch_idx: int
  ) -> Any:
      raise NotImplementedError
  
file_path: determined/e2e_tests/tests/fixtures/trial_error/model_def.py
imports: from typing import Any
  from determined.pytorch import DataLoader, PyTorchTrial, PyTorchTrialContext, TorchData
label_desc: Use image node_id 4 for calling the ErrorTrial obj's underlying member method code with example usage: obj.train_batch(batch, epoch_idx, batch_idx) without return types