code: def test_save_with_model_config(model_path, model_config):
      model = InferenceContextModel()
      mlflow.pyfunc.save_model(
          model_path, python_model=model, model_config=model_config
      )
  
      loaded_model = mlflow.pyfunc.load_model(model_uri=model_path)
  
      assert loaded_model.model_config
      assert set(model_config.keys()) == set(loaded_model.model_config)
      assert all(
          loaded_model.model_config[k] == v
          for k, v in model_config.items()
      )
      assert all(
          loaded_model.model_config[k] == v
          for k, v in loaded_model.predict([[0]])
      )
  
file_path: mlflow/tests/pyfunc/test_pyfunc_model_config.py
imports: import os
  import pytest
  import mlflow
  from mlflow.models import Model
label_desc: Use image node_id 4 for calling a global function with example usage: test_save_with_model_config(model_path, model_config) without return types