code: def test_model_log_with_signature_inference(tf2_toy_model):
      artifact_path = "model"
      example = tf2_toy_model.inference_data
  
      with mlflow.start_run():
          mlflow.tensorflow.log_model(
              tf2_toy_model.model,
              artifact_path=artifact_path,
              input_example=example,
          )
          model_uri = mlflow.get_artifact_uri(artifact_path)
  
      mlflow_model = Model.load(model_uri)
      assert mlflow_model.signature == infer_signature(
          tf2_toy_model.inference_data,
          tf2_toy_model.expected_results.numpy(),
      )
  
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 4 for calling a global function with example usage: test_model_log_with_signature_inference(tf2_toy_model) without return types