code: def build_model() -> tf.keras.Model:
      model = tf.keras.Sequential(
          [
              tf.keras.layers.InputLayer(input_shape=()),
              # Add feature dimension, expanding (batch_size,) to (batch_size, 1).
              tf.keras.layers.Flatten(),
              tf.keras.layers.Dense(10),
              tf.keras.layers.Dense(1),
          ]
      )
      return model
  
file_path: ray/doc/source/train/doc_code/tf_starter.py
imports: import ray
  import tensorflow
  from ray import train
  from ray.train import ScalingConfig
  from ray.train.tensorflow import TensorflowTrainer
  from ray.train.tensorflow.keras import ReportCheckpointCallback
label_desc: Use image node_id 1 for calling a global function with example usage: build_model() and returns: model