code: def print_debug_info(model_trace):
      model_trace.compute_log_prob()
      for name, site in model_trace.nodes.items():
          if site["type"] == "sample" and not isinstance(
              site["fn"], _Subsample
          ):
              logger.debug(
                  "prob( {} ):\n {}".format(
                      name, site["log_prob"].exp()
                  )
              )
  
imports: import io
  import logging
  import pytest
  import torch
  import pyro
  import pyro.distributions
  import pyro.poutine
  from pyro.infer import config_enumerate
  from pyro.infer.mcmc import HMC, NUTS
  from pyro.infer.mcmc.api import MCMC
  from pyro.infer.mcmc.util import TraceEinsumEvaluator, TraceTreeEvaluator, initialize_model
  from pyro.infer.reparam import LatentStableReparam
  from pyro.poutine.subsample_messenger import _Subsample
  from tests.common import assert_close, assert_equal, xfail_param
label_desc: Use image node_id 3 for calling a global function with example usage: print_debug_info(model_trace) without return types