code: def simple_model(config):
      import tensorflow as tf
  
      model = tf.keras.models.Sequential(
          [
              tf.keras.layers.Dense(10, input_shape=(1,)),
              tf.keras.layers.Dense(1),
          ]
      )
      return model
  
imports: import shutil
  from unittest import TestCase
  import numpy
  import pytest
  import tensorflow
  from zoo import init_nncontext
  from zoo.orca.data import XShards
  import zoo.orca.data.pandas
  from zoo.orca.learn.tf2 import Estimator
  from zoo.ray import RayContext
  import ray
  import os
label_desc: Use image node_id 4 for calling a global function with example usage: simple_model(config) and returns: model