code: def assert_ok(model, guide, elbo, steps=2, *args, **kwargs):
      """
      Assert that inference works without warnings or errors.
      """
      inference = build_svi(model, guide, elbo)
      for i in range(steps):
          inference.step(*args, **kwargs)
  
imports: import warnings
  import pytest
  import torch
  from pytest import approx
  from torch.distributions import constraints
  from pyro.generic import distributions
  from pyro.generic import infer, ops, optim, pyro, pyro_backend
  from tests.common import assert_close, xfail_param
label_desc: Use image node_id 2 for calling a global function with example usage: assert_ok(model, guide, elbo, steps) without return types