code: def build_model():
      import tensorflow as tf
  
      model = tf.keras.Sequential(
          [
              tf.keras.layers.InputLayer(input_shape=()),
              tf.keras.layers.Flatten(),
              tf.keras.layers.Dense(1),
          ]
      )
      model.compile(optimizer="sgd", loss="mse")
      return model
  
file_path: ray/python/ray/data/tests/test_iterator.py
imports: from typing import Dict
  from unittest.mock import MagicMock, patch
  import numpy
  import pytest
  import tensorflow
  import torch
  import ray
label_desc: Use image node_id 1 for calling a global function with example usage: build_model() and returns: model