code: def test_train_on_batch_with_pred_multi_input(self):
      x1 = torch.rand(ModelMultiInputTest.batch_size, 1)
      x2 = torch.rand(ModelMultiInputTest.batch_size, 1)
      y = torch.rand(ModelMultiInputTest.batch_size, 1)
      loss, pred_y = self.model.train_on_batch(
          (x1, x2), y, return_pred=True
      )
      self.assertEqual(type(loss), float)
      self.assertEqual(
          pred_y.shape, (ModelMultiInputTest.batch_size, 1)
      )
  
imports: import numpy
  import torch
  import torch.nn
  from torch.utils.data import DataLoader
  from poutyne import Model, TensorDataset
  from tests.framework.model.base import ModelFittingTestCase, MultiIOModel
label_desc: Use image node_id 5 for calling the ModelMultiInputTest obj's underlying member method code with example usage: obj.test_train_on_batch_with_pred_multi_input() without return types