code: def model_creator(config):
      model = tf.keras.models.Sequential(
          [
              tf.keras.layers.Dense(
                  config["hidden_size"], input_shape=(1,)
              ),
              tf.keras.layers.Dense(1),
          ]
      )
      model.compile(
          loss="mse",
          optimizer=tf.keras.optimizers.SGD(config["lr"]),
          metrics=["mse"],
      )
      return model
  
imports: import tensorflow
  import numpy
  from unittest import TestCase
  from bigdl.orca.automl.auto_estimator import AutoEstimator
  import pytest
label_desc: Use image node_id 1 for calling a global function with example usage: model_creator(config) and returns: model