code: def test_save_with_options(tmp_path, tf2_toy_model):
      model_path = os.path.join(tmp_path, "model")
  
      saved_model_kwargs = {
          "signatures": [tf.TensorSpec(shape=None, dtype=tf.float32)],
          "options": tf.saved_model.SaveOptions(save_debug_info=True),
      }
  
      with mock.patch("tensorflow.saved_model.save") as mock_save:
          mlflow.tensorflow.save_model(
              tf2_toy_model.model,
              model_path,
              saved_model_kwargs=saved_model_kwargs,
          )
          mock_save.assert_called_once_with(
              mock.ANY, mock.ANY, **saved_model_kwargs
          )
  
          mock_save.reset_mock()
  
          with mlflow.start_run():
              mlflow.tensorflow.log_model(
                  tf2_toy_model.model,
                  "model",
                  saved_model_kwargs=saved_model_kwargs,
              )
  
          mock_save.assert_called_once_with(
              mock.ANY, mock.ANY, **saved_model_kwargs
          )
  
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 5 for calling a global function with example usage: test_save_with_options(tmp_path, tf2_toy_model) without return types