code: def train_small_model():
      model_input = tf.keras.layers.Input((3,))
      model_output = tf.keras.layers.Dense(4)(model_input)
      model = tf.keras.Model(model_input, model_output)
      model.compile(loss="mse")
  
      x = np.random.uniform(size=(5, 3))
      y = np.random.uniform(size=(5, 4))
      model.fit(x, y, epochs=1)
  
file_path: addons/tensorflow_addons/utils/tests/test_utils_test.py
imports: import random
  from packaging.version import Version
  import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 3 for calling a global function with example usage: train_small_model() without return types