code: def test_save_and_load_tf2_module(tmp_path, tf2_toy_model):
      model_path = os.path.join(tmp_path, "model")
      mlflow.tensorflow.save_model(tf2_toy_model.model, model_path)
  
      loaded_model = mlflow.tensorflow.load_model(model_path)
  
      predictions = loaded_model(tf2_toy_model.inference_data).numpy()
      np.testing.assert_allclose(
          predictions,
          tf2_toy_model.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 2 for calling a global function with example usage: test_save_and_load_tf2_module(tmp_path, tf2_toy_model) without return types