code: def model_creator_pytorch(config):
      """Returns a torch.nn.Module object."""
      return nn.Linear(1, config.get("hidden_size", 1))
  
imports: from unittest import TestCase
  from zoo.orca.automl.model.base_pytorch_model import PytorchModelBuilder
  from torch.utils.data import Dataset, DataLoader
  import pytest
  import numpy
  import torch
  import torch.nn
label_desc: Use image node_id 5 for calling a global function with example usage: model_creator_pytorch(config) and returns: nn