code: def get_batch(batch_size):
      x1 = torch.rand(batch_size, 1)
      x2 = torch.rand(batch_size, 1)
      y1 = torch.rand(batch_size, 1)
      y2 = torch.rand(batch_size, 1)
      return dict(x1=x1, x2=x2), dict(y1=y1, y2=y2)
  
imports: import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  from poutyne import Model
  from tests.framework.model.base import ModelFittingTestCase
label_desc: Use image node_id 2 for calling a global function with example usage: get_batch(batch_size) and returns: dict,  dict