code: def tf2_toy_model():
      tf.random.set_seed(1337)
      rand_w = tf.random.uniform(shape=[3, 1], dtype=tf.float32)
      rand_b = tf.random.uniform(shape=[], dtype=tf.float32)
  
      inference_data = np.array(
          [[2, 3, 4], [5, 6, 7]], dtype=np.float32
      )
      model = ToyModel(rand_w, rand_b)
      expected_results = model(inference_data)
  
      return TF2ModelInfo(
          model=model,
          inference_data=inference_data,
          expected_results=expected_results,
      )
  
file_path: mlflow/tests/tensorflow/test_tensorflow2_core_model_export.py
imports: import collections
  import os
  from unittest import mock
  import numpy
  import pytest
  import tensorflow
  import mlflow.tensorflow
  from mlflow.models import Model, infer_signature
label_desc: Use image node_id 1 for calling a global function with example usage: tf2_toy_model() and returns: TF2ModelInfo