class_docstr: Model to test multiple dict input/output
class_docstr_tok: ["Model","to","test","multiple","dict","input\/output"]
code: def __init__(self, input_keys, output_keys):
      super().__init__()
      assert len(input_keys) == len(output_keys)
      inputs = {k: nn.Linear(1, 1) for k in input_keys}
      self.inputs = nn.ModuleDict(inputs)
  
      self.input_keys = input_keys
      self.output_keys = output_keys
  
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 1 to create a new DictIOModel object from inherited base classes: nn with example: obj = DictIOModel(input_keys, output_keys)