code: def build_cnn_model() -> tf.keras.Model:
      model = tf.keras.Sequential(
          [
              tf.keras.Input(shape=(28, 28)),
              tf.keras.layers.Flatten(),
              tf.keras.layers.Dense(512, activation="relu"),
              tf.keras.layers.Dense(512, activation="relu"),
              tf.keras.layers.Dense(10),
          ]
      )
      return model
  
file_path: ray/release/air_tests/air_benchmarks/workloads/tensorflow_benchmark.py
imports: import json
  import os
  import time
  from pathlib import Path
  import click
  import numpy
  import tensorflow
  from typing import List, Tuple
label_desc: Use image node_id 2 for calling a global function with example usage: build_cnn_model() and returns: model