code: def test_model_with_param(jit_compile, num_chains):
      kernel = NUTS(
          model_with_param,
          jit_compile=jit_compile,
          ignore_jit_warnings=True,
      )
      mcmc = MCMC(kernel, 10, num_chains=num_chains, mp_context="spawn")
      mcmc.run()
  
file_path: pyro/tests/infer/mcmc/test_mcmc_util.py
imports: from functools import partial
  import pytest
  import torch
  import pyro
  import pyro.distributions
  from pyro.infer import Predictive
  from pyro.infer.autoguide import init_to_feasible, init_to_generated, init_to_mean, init_to_median, init_to_sample, init_to_uniform, init_to_value
  from pyro.infer.mcmc import NUTS
  from pyro.infer.mcmc.api import MCMC
  from pyro.infer.mcmc.util import initialize_model
  from pyro.util import optional
  from tests.common import assert_close, str_erase_pointers
label_desc: Use image node_id 4 for calling a global function with example usage: test_model_with_param(jit_compile, num_chains) without return types