code: def set_relaxed_distributions(relaxed=True):
      global _RELAX
      old = _RELAX
      try:
          _RELAX = relaxed
          yield
      finally:
          _RELAX = old
  
file_path: pyro/pyro/contrib/epidemiology/distributions.py
imports: import math
  from contextlib import contextmanager
  import torch
  from torch.distributions.utils import broadcast_all
  import pyro.distributions
  from pyro.distributions.util import is_validation_enabled
label_desc: Use image node_id 5 for calling a global function with example usage: set_relaxed_distributions(relaxed) without return types