code: def model_creator(config):
      """Returns a torch.nn.Module object."""
      return nn.Linear(1, config.get("hidden_size", 1))
  
imports: from bigdl.orca.learn.pytorch import Estimator
  from unittest import TestCase
  import pytest
  import torch
  import torch.nn
  import os
  import numpy
  from tempfile import TemporaryDirectory
label_desc: Use image node_id 1 for calling a global function with example usage: model_creator(config) and returns: nn