code: def __init__(self):
      super().__init__()
      self.simple_model = tf.keras.Sequential(
          [tf.keras.layers.Dense(10), tf.keras.layers.Dense(1)]
      )
  
imports: import math
  import os
  import shutil
  import tempfile
  import pytest
  from unittest import TestCase
  import tensorflow
  import numpy
  from bigdl.orca.data.pandas import read_csv
  from bigdl.orca.data import SparkXShards, XShards
  from bigdl.orca.data.tf.data import Dataset
  from bigdl.orca.learn.tf2 import Estimator
  from bigdl.orca.ray import OrcaRayContext
  from bigdl.orca import OrcaContext
label_desc: Use image node_id 1 to create a new SimpleModel object from inherited base classes: tf.keras with example: obj = SimpleModel()