code: def test_build_model(self):
      config = {
          "name": "mlp",
          "input_dim": 3,
          "output_dim": 1,
          "hidden_dims": [2],
      }
      model = build_model(config)
      self.assertTrue(isinstance(model, ClassyModel))
  
      tensor = torch.tensor([[1, 2, 3]], dtype=torch.float)
      output = model.forward(tensor)
      self.assertEqual(output.shape, torch.Size([1, 1]))
  
      tensor = torch.tensor([[1, 2, 3], [1, 2, 3]], dtype=torch.float)
      output = model.forward(tensor)
      self.assertEqual(output.shape, torch.Size([2, 1]))
  
file_path: ClassyVision/test/models_mlp_test.py
imports: import unittest
  import torch
  from classy_vision.generic.util import get_torch_version
  from classy_vision.models import build_model, ClassyModel
label_desc: Use image node_id 1 for calling the TestMLPModel obj's underlying member method code with example usage: obj.test_build_model() without return types