============================= test session starts ==============================
platform linux -- Python 3.5.6, pytest-3.8.1, py-1.9.0, pluggy-0.11.0
rootdir: /home/aryaman4/pyro, inifile: setup.cfg
collected 20697 items

tests/test_examples.py .FFF...............................FFF...FFFFF... [  0%]
FFFFFFFFFFFFFFFF....FF.........FFFFXFFFFFFssssssssssssssssssssssssssssss [  0%]
sssssssssFF......xxxx..FFFFFFxxxx......xFFFFF                            [  0%]
tests/test_generic.py ........xx.....sxx.sss.sxx.sss..xx....             [  0%]
tests/test_primitives.py ....                                            [  1%]
tests/test_util.py ..                                                    [  1%]
tests/contrib/test_hessian.py ..                                         [  1%]
tests/contrib/test_minipyro.py ..................xx....                  [  1%]
tests/contrib/test_util.py ........                                      [  1%]
tests/contrib/autoguide/test_inference.py .........................      [  1%]
tests/contrib/autoguide/test_mean_field_entropy.py ..                    [  1%]
tests/contrib/autoname/test_named.py .....                               [  1%]
tests/contrib/autoname/test_scoping.py ........                          [  1%]
tests/contrib/bnn/test_hidden_layer.py ....                              [  1%]
tests/contrib/cevae/test_cevae.py ...................................... [  1%]
............                                                             [  1%]
tests/contrib/easyguide/test_easyguide.py ............                   [  1%]
tests/contrib/epidemiology/test_distributions.py ....................... [  1%]
........................................................................ [  2%]
........................................................................ [  2%]
........................................................................ [  2%]
.....................................................................    [  3%]
tests/contrib/epidemiology/test_models.py .............................. [  3%]
........................................................................ [  3%]
........................................................................ [  4%]
........xx.............................................................. [  4%]
..............                                                           [  4%]
tests/contrib/epidemiology/test_quant.py .....                           [  4%]
tests/contrib/epidemiology/test_util.py ................................ [  4%]
....                                                                     [  4%]
tests/contrib/forecast/test_evaluate.py .........................        [  4%]
tests/contrib/forecast/test_forecaster.py .............................. [  4%]
........................................................................ [  5%]
........................................................................ [  5%]
........................................................................ [  5%]
........................................................................ [  6%]
........................................................................ [  6%]
........................................................................ [  6%]
........................................................................ [  7%]
........................................................................ [  7%]
........................................................................ [  8%]
........................................................................ [  8%]
..........................................ssssssssssssssssssssssssssssss [  8%]
ssssssssssssssssssssssssssssssssssssssssss.............................. [  9%]
..........................................ssssssssssssssssssssssssssssss [  9%]
ssssssssssssssssssssssssssssssssssssssssss.............................. [  9%]
..........................................ssssssssssssssssssssssssssssss [ 10%]
ssssssssssssssssssssssssssssssssssssssssss.............................. [ 10%]
..........................................ssssssssssssssssssssssssssssss [ 10%]
ssssssssssssssssssssssssssssssssssssssssss.............................. [ 11%]
..........................................ssssssssssssssssssssssssssssss [ 11%]
ssssssssssssssssssssssssssssssssssssssssss.............................. [ 11%]
....                                                                     [ 11%]
tests/contrib/forecast/test_util.py .................................... [ 12%]
........................................................................ [ 12%]
........................................................................ [ 12%]
........................................................................ [ 13%]
..................xxxxxxxxxxxxxxxxxx.................................... [ 13%]
........................................................................ [ 13%]
........................................................................ [ 14%]
........................................................................ [ 14%]
........................................................................ [ 14%]
........................................................................ [ 15%]
........................................................................ [ 15%]
........................................................................ [ 15%]
........................................................................ [ 16%]
..................                                                       [ 16%]
tests/contrib/gp/test_conditional.py ..............                      [ 16%]
tests/contrib/gp/test_kernels.py .....................                   [ 16%]
tests/contrib/gp/test_likelihoods.py ................................... [ 16%]
.............                                                            [ 16%]
tests/contrib/gp/test_models.py ........................................ [ 16%]
....................................                                     [ 17%]
tests/contrib/gp/test_parameterized.py ...                               [ 17%]
tests/contrib/oed/test_ewma.py .....                                     [ 17%]
tests/contrib/oed/test_finite_spaces_eig.py ...x...                      [ 17%]
tests/contrib/oed/test_glmm.py .......                                   [ 17%]
tests/contrib/oed/test_linear_models_eig.py ........                     [ 17%]
tests/contrib/oed/test_xexpx.py ...                                      [ 17%]
tests/contrib/randomvariable/test_random_variable.py ........            [ 17%]
tests/contrib/timeseries/test_gp.py ................................     [ 17%]
tests/contrib/timeseries/test_lgssm.py ................                  [ 17%]
tests/contrib/tracking/test_assignment.py .............................. [ 17%]
........................................................................ [ 18%]
........................................................................ [ 18%]
........................................................................ [ 18%]
........................................................................ [ 19%]
                                                                         [ 19%]
tests/contrib/tracking/test_distributions.py ........                    [ 19%]
tests/contrib/tracking/test_dynamic_models.py ....                       [ 19%]
tests/contrib/tracking/test_ekf.py ..                                    [ 19%]
tests/contrib/tracking/test_em.py .....                                  [ 19%]
tests/contrib/tracking/test_hashing.py ................................. [ 19%]
.......                                                                  [ 19%]
tests/contrib/tracking/test_measurements.py .                            [ 19%]
tests/distributions/test_binomial.py ................................... [ 19%]
...............................                                          [ 19%]
tests/distributions/test_categorical.py ..............                   [ 19%]
tests/distributions/test_coalescent.py ................................. [ 19%]
........................................................................ [ 20%]
........................................................................ [ 20%]
........................................................................ [ 20%]
........................................................................ [ 21%]
........................................................................ [ 21%]
...........................................                              [ 21%]
tests/distributions/test_conjugate.py .................................. [ 21%]
.....................                                                    [ 22%]
tests/distributions/test_conjugate_update.py ........................... [ 22%]
                                                                         [ 22%]
tests/distributions/test_cuda.py sssssssssssssssssssssssssssssssssssssss [ 22%]
sssssssssssssssssssssssssssssssssssssssssssssssssssssssssssssss          [ 22%]
tests/distributions/test_delta.py ..............................         [ 22%]
tests/distributions/test_distributions.py ............s......sss..s.s.ss [ 23%]
s.......................x...............x.........x.....xxxxx.xxxx.xx.x. [ 23%]
..........ss........................................................s.ss [ 23%]
sssssssssssssssssssssssssssssss.xxxx.ss.ss.............................. [ 24%]
........................................................................ [ 24%]
........................................................................ [ 24%]
............sssssssss................................................... [ 25%]
........................................................................ [ 25%]
........................................................................ [ 25%]
..............................xxxxxxxxx................................. [ 26%]
........................................................................ [ 26%]
..............................xxxxxxxxx................................. [ 26%]
..................................................................ssssss [ 27%]
ssssssssssss............................................................ [ 27%]
........................................................................ [ 27%]
........................................................................ [ 28%]
..................................................................xxx... [ 28%]
........................................................................ [ 28%]
......ssssss............................................................ [ 29%]
..........................................xx............................ [ 29%]
..xx......................xxxx.......................................... [ 29%]
..............................................xxx....................... [ 30%]
......................xxx...........................ssssss.............. [ 30%]
........................................................................ [ 31%]
.......                                                                  [ 31%]
tests/distributions/test_empirical.py .................................. [ 31%]
........................................................................ [ 31%]
........                                                                 [ 31%]
tests/distributions/test_extended.py ........                            [ 31%]
tests/distributions/test_gaussian_mixtures.py .......................... [ 31%]
...                                                                      [ 31%]
tests/distributions/test_haar.py ......                                  [ 31%]
tests/distributions/test_hmm.py ........................................ [ 31%]
........................................................................ [ 32%]
........................................................................ [ 32%]
........................................................................ [ 33%]
........................................................................ [ 33%]
........................................................................ [ 33%]
........................................................................ [ 34%]
........................................................................ [ 34%]
........................................................................ [ 34%]
........................................................................ [ 35%]
........................................................................ [ 35%]
........................................................................ [ 35%]
........................................................................ [ 36%]
........................................................................ [ 36%]
........................................................................ [ 36%]
........................................................................ [ 37%]
........................................................................ [ 37%]
........................................................................ [ 37%]
........................................................................ [ 38%]
........................................................................ [ 38%]
........................................................................ [ 38%]
........................................................................ [ 39%]
........................................................................ [ 39%]
........................................................................ [ 39%]
........................................................................ [ 40%]
........................................................................ [ 40%]
........................................................................ [ 41%]
........................................................................ [ 41%]
........................................................................ [ 41%]
........................................................................ [ 42%]
........................................................................ [ 42%]
..................................................                       [ 42%]
tests/distributions/test_ig.py ............                              [ 42%]
tests/distributions/test_improper_uniform.py ........................... [ 42%]
                                                                         [ 42%]
tests/distributions/test_independent.py ................................ [ 43%]
........................................................................ [ 43%]
......................................................................   [ 43%]
tests/distributions/test_kl.py ......................................... [ 43%]
..........                                                               [ 43%]
tests/distributions/test_lkj.py ...........................              [ 44%]
tests/distributions/test_lowrank_mvn.py ...                              [ 44%]
tests/distributions/test_mask.py ....................................... [ 44%]
....................................                                     [ 44%]
tests/distributions/test_mixture.py .................................... [ 44%]
........................................................................ [ 44%]
.............................................                            [ 45%]
tests/distributions/test_mvn.py ...........................              [ 45%]
tests/distributions/test_mvt.py ........................................ [ 45%]
........................................................                 [ 45%]
tests/distributions/test_omt_mvn.py ........ssssssssssssssss..           [ 45%]
tests/distributions/test_one_hot_categorical.py ...........              [ 45%]
tests/distributions/test_pickle.py ........sss..sss..s.s.s.ss..X..s.s.s. [ 46%]
s...s...s..ss....s....s.........sss                                      [ 46%]
tests/distributions/test_polya_gamma.py ...                              [ 46%]
tests/distributions/test_rejector.py ................................... [ 46%]
........................................................................ [ 46%]
...............................................                          [ 47%]
tests/distributions/test_relaxed_straight_through.py .............       [ 47%]
tests/distributions/test_reshape.py .................................... [ 47%]
....................................                                     [ 47%]
tests/distributions/test_shapes.py ..........                            [ 47%]
tests/distributions/test_spanning_tree.py ...FFF.....FFFFF.....FFFFF.... [ 47%]
.x.....x.....xssssssssssssssssssssssssssssssssssssssssssssssssssssssssss [ 48%]
ss                                                                       [ 48%]
tests/distributions/test_stable.py ..................................... [ 48%]
........................................................................ [ 48%]
........................................................................ [ 48%]
........................................................................ [ 49%]
........................................................................ [ 49%]
........................................................................ [ 49%]
........................................................................ [ 50%]
........................................................................ [ 50%]
........................................................................ [ 50%]
........................................................................ [ 51%]
..                                                                       [ 51%]
tests/distributions/test_tensor_type.py ...                              [ 51%]
tests/distributions/test_torch_patch.py ss............                   [ 51%]
tests/distributions/test_transforms.py ............................      [ 51%]
tests/distributions/test_unit.py ............                            [ 51%]
tests/distributions/test_util.py ....................................... [ 51%]
......                                                                   [ 51%]
tests/distributions/test_von_mises.py .................................. [ 52%]
.........                                                                [ 52%]
tests/distributions/test_zero_inflated.py .............................. [ 52%]
........................................................................ [ 52%]
........................................................................ [ 52%]
........................................................................ [ 53%]
.                                                                        [ 53%]
tests/infer/test_abstract_infer.py ..                                    [ 53%]
tests/infer/test_autoguide.py .......................................... [ 53%]
....................................................................xxxx [ 53%]
xxxxxxxxxx.............................................................. [ 54%]
........................................................................ [ 54%]
........................................................................ [ 54%]
.........                                                                [ 54%]
tests/infer/test_compute_downstream_costs.py ........................... [ 55%]
....................                                                     [ 55%]
tests/infer/test_conjugate_gradients.py .                                [ 55%]
tests/infer/test_csis.py ...                                             [ 55%]
tests/infer/test_discrete.py ...........................xx               [ 55%]
tests/infer/test_elbo_mapdata.py .........                               [ 55%]
tests/infer/test_enum.py ............................................... [ 55%]
........................................................................ [ 55%]
........................................................................ [ 56%]
........................................................................ [ 56%]
........................................................................ [ 56%]
........................................................................ [ 57%]
........................................................................ [ 57%]
........................................................................ [ 57%]
........................................................................ [ 58%]
..............................................                           [ 58%]
tests/infer/test_gradient.py ........................................... [ 58%]
.......xx..xx..xx..xx................................................... [ 59%]
..........x.xxxxxxxxxxxxxxxxx                                            [ 59%]
tests/infer/test_inference.py .....................xxx.....xxx.....xxx.. [ 59%]
.xx......................                                                [ 59%]
tests/infer/test_initialization.py .                                     [ 59%]
tests/infer/test_jit.py ...xx..X........................................ [ 59%]
...............................                                          [ 59%]
tests/infer/test_multi_sample_elbos.py ..                                [ 59%]
tests/infer/test_predictive.py .............                             [ 60%]
tests/infer/test_sampling.py ..                                          [ 60%]
tests/infer/test_smcfilter.py .......................................... [ 60%]
.............................................                            [ 60%]
tests/infer/test_svgd.py ............................                    [ 60%]
tests/infer/test_tmc.py ................................................ [ 60%]
................................................................         [ 61%]
tests/infer/test_util.py .........                                       [ 61%]
tests/infer/test_valid_models.py ....................................... [ 61%]
........................................................................ [ 61%]
........................................................................ [ 62%]
........................................................................ [ 62%]
........................................................................ [ 62%]
........................................................................ [ 63%]
........................................................................ [ 63%]
........                                                                 [ 63%]
tests/infer/mcmc/test_adaptation.py ........                             [ 63%]
tests/infer/mcmc/test_hmc.py ssss....................                    [ 63%]
tests/infer/mcmc/test_mcmc_api.py ...............................        [ 63%]
tests/infer/mcmc/test_mcmc_util.py ..........................            [ 63%]
tests/infer/mcmc/test_nuts.py ssss...............................        [ 64%]
tests/infer/mcmc/test_valid_models.py ........Xxx....................... [ 64%]
.............                                                            [ 64%]
tests/infer/reparam/test_conjugate.py .................................. [ 64%]
..............................                                           [ 64%]
tests/infer/reparam/test_discrete_cosine.py ............................ [ 64%]
....                                                                     [ 64%]
tests/infer/reparam/test_haar.py ................                        [ 64%]
tests/infer/reparam/test_hmm.py ........................................ [ 65%]
........................................................................ [ 65%]
........................................................................ [ 65%]
........................................................................ [ 66%]
........................................................................ [ 66%]
........................................................................ [ 66%]
........................................................................ [ 67%]
........................................................................ [ 67%]
........                                                                 [ 67%]
tests/infer/reparam/test_loc_scale.py ..............................     [ 67%]
tests/infer/reparam/test_neutra.py .X..                                  [ 67%]
tests/infer/reparam/test_split.py ............                           [ 67%]
tests/infer/reparam/test_stable.py ..................................... [ 67%]
............s...sss.ssss.ssss.ssss.ssss.ssss.ssss.ssss.sssssss.......... [ 68%]
..............................s...s......                                [ 68%]
tests/infer/reparam/test_studentt.py .................................   [ 68%]
tests/infer/reparam/test_transform.py ...                                [ 68%]
tests/infer/reparam/test_unit_jacobian.py ...                            [ 68%]
tests/integration_tests/test_conjugate_gaussian_models.py ............   [ 68%]
tests/integration_tests/test_tracegraph_elbo.py ............             [ 68%]
tests/nn/test_autoregressive.py ..                                       [ 68%]
tests/nn/test_module.py ...........................................F.    [ 69%]
tests/ops/test_arrowhead.py ...                                          [ 69%]
tests/ops/test_contract.py ............................................. [ 69%]
...............................s..........................s............. [ 69%]
.............s.......................................................... [ 69%]
.......................................................................  [ 70%]
tests/ops/test_gamma_gaussian.py ....................................... [ 70%]
........................................................................ [ 70%]
........................................................................ [ 71%]
........................................................................ [ 71%]
........................................................................ [ 71%]
........................................................................ [ 72%]
........................................................................ [ 72%]
..................................................sssss................. [ 72%]
........................................................................ [ 73%]
..............................................                           [ 73%]
tests/ops/test_gaussian.py ............................................. [ 73%]
........................................................................ [ 74%]
........................................................................ [ 74%]
........................................................................ [ 74%]
........................................................................ [ 75%]
........................................................................ [ 75%]
........................................................................ [ 75%]
........................................................................ [ 76%]
........................................................................ [ 76%]
...........................sss.......................................... [ 76%]
........................................................................ [ 77%]
......................                                                   [ 77%]
tests/ops/test_indexing.py ............................................. [ 77%]
........................................................................ [ 77%]
........................................................................ [ 78%]
........................................................................ [ 78%]
........................................................................ [ 78%]
...                                                                      [ 78%]
tests/ops/test_integrator.py .........                                   [ 78%]
tests/ops/test_jit.py ...                                                [ 78%]
tests/ops/test_linalg.py ............                                    [ 79%]
tests/ops/test_newton.py ............................................... [ 79%]
...............................                                          [ 79%]
tests/ops/test_packed.py .......................................         [ 79%]
tests/ops/test_special.py .....................                          [ 79%]
tests/ops/test_ssm_gp.py .........                                       [ 79%]
tests/ops/test_stats.py ................................................ [ 79%]
...........                                                              [ 80%]
tests/ops/test_tensor_utils.py ......................................... [ 80%]
........................................................................ [ 80%]
........................................................................ [ 80%]
........................................................................ [ 81%]
........................................................................ [ 81%]
........................................................................ [ 81%]
........................................................................ [ 82%]
........................................................................ [ 82%]
........................................................................ [ 82%]
........................................................................ [ 83%]
........................................................................ [ 83%]
.......................................                                  [ 83%]
tests/ops/test_welford.py ..............                                 [ 83%]
tests/ops/einsum/test_adjoint.py ....................................... [ 84%]
........................................................................ [ 84%]
.................                                                        [ 84%]
tests/ops/einsum/test_torch_log.py ..................................... [ 84%]
...............................................                          [ 84%]
tests/optim/test_multi.py .......                                        [ 85%]
tests/optim/test_optim.py ................................               [ 85%]
tests/params/test_module.py .......                                      [ 85%]
tests/params/test_param.py ..                                            [ 85%]
tests/perf/test_benchmark.py EEEEEEEE                                    [ 85%]
tests/poutine/test_counterfactual.py ....                                [ 85%]
tests/poutine/test_mapdata.py ...............ssss..................      [ 85%]
tests/poutine/test_nesting.py .                                          [ 85%]
tests/poutine/test_poutines.py ......................................... [ 85%]
.................................                                        [ 85%]
tests/poutine/test_properties.py ........................                [ 85%]
tests/poutine/test_trace_struct.py ..................................... [ 86%]
........................................................................ [ 86%]
........................................................................ [ 86%]
........................................................................ [ 87%]
........................................................................ [ 87%]
........................................................................ [ 87%]
........................................................................ [ 88%]
........................................................................ [ 88%]
........................................................................ [ 88%]
........................................................................ [ 89%]
........................................................................ [ 89%]
........................................................................ [ 89%]
........................................................................ [ 90%]
........................................................................ [ 90%]
........................................................................ [ 90%]
........................................................................ [ 91%]
........................................................................ [ 91%]
........................................................................ [ 92%]
........................................................................ [ 92%]
........................................................................ [ 92%]
........................................................................ [ 93%]
........................................................................ [ 93%]
........................................................................ [ 93%]
........................................................................ [ 94%]
........................................................................ [ 94%]
........................................................................ [ 94%]
........................................................................ [ 95%]
........................................................................ [ 95%]
........................................................................ [ 95%]
........................................................................ [ 96%]
........................................................................ [ 96%]
........................................................................ [ 96%]
........................................................................ [ 97%]
........................................................................ [ 97%]
........................................................................ [ 97%]
........................................................................ [ 98%]
........................................................................ [ 98%]
........................................................................ [ 98%]
........................................................................ [ 99%]
........................................................................ [ 99%]
...................................                                      [ 99%]
tests/pyroapi/test_pyroapi.py ..........xx..............xx.....x         [100%]

==================================== ERRORS ====================================
_____ ERROR at setup of test_benchmark[PoissonGamma::reparam=False_Trace] ______
file /home/aryaman4/pyro/tests/perf/test_benchmark.py, line 128
  @pytest.mark.parametrize('model, model_args, id', TEST_MODELS, ids=MODEL_IDS)
  @pytest.mark.benchmark(
      min_rounds=5,
      disable_gc=True,
  )
  @pytest.mark.disable_validation()
  def test_benchmark(benchmark, model, model_args, id):
E       fixture 'benchmark' not found
>       available fixtures: cache, capfd, capfdbinary, caplog, capsys, capsysbinary, doctest_namespace, monkeypatch, pytestconfig, record_property, record_xml_attribute, record_xml_property, recwarn, tmpdir, tmpdir_factory
>       use 'pytest --fixtures [testpath]' for help on them.

/home/aryaman4/pyro/tests/perf/test_benchmark.py:128
___ ERROR at setup of test_benchmark[PoissonGamma::reparam=False_TraceGraph] ___
file /home/aryaman4/pyro/tests/perf/test_benchmark.py, line 128
  @pytest.mark.parametrize('model, model_args, id', TEST_MODELS, ids=MODEL_IDS)
  @pytest.mark.benchmark(
      min_rounds=5,
      disable_gc=True,
  )
  @pytest.mark.disable_validation()
  def test_benchmark(benchmark, model, model_args, id):
E       fixture 'benchmark' not found
>       available fixtures: cache, capfd, capfdbinary, caplog, capsys, capsysbinary, doctest_namespace, monkeypatch, pytestconfig, record_property, record_xml_attribute, record_xml_property, recwarn, tmpdir, tmpdir_factory
>       use 'pytest --fixtures [testpath]' for help on them.

/home/aryaman4/pyro/tests/perf/test_benchmark.py:128
______ ERROR at setup of test_benchmark[PoissonGamma::reparam=True_Trace] ______
file /home/aryaman4/pyro/tests/perf/test_benchmark.py, line 128
  @pytest.mark.parametrize('model, model_args, id', TEST_MODELS, ids=MODEL_IDS)
  @pytest.mark.benchmark(
      min_rounds=5,
      disable_gc=True,
  )
  @pytest.mark.disable_validation()
  def test_benchmark(benchmark, model, model_args, id):
E       fixture 'benchmark' not found
>       available fixtures: cache, capfd, capfdbinary, caplog, capsys, capsysbinary, doctest_namespace, monkeypatch, pytestconfig, record_property, record_xml_attribute, record_xml_property, recwarn, tmpdir, tmpdir_factory
>       use 'pytest --fixtures [testpath]' for help on them.

/home/aryaman4/pyro/tests/perf/test_benchmark.py:128
___ ERROR at setup of test_benchmark[PoissonGamma::reparam=True_TraceGraph] ____
file /home/aryaman4/pyro/tests/perf/test_benchmark.py, line 128
  @pytest.mark.parametrize('model, model_args, id', TEST_MODELS, ids=MODEL_IDS)
  @pytest.mark.benchmark(
      min_rounds=5,
      disable_gc=True,
  )
  @pytest.mark.disable_validation()
  def test_benchmark(benchmark, model, model_args, id):
E       fixture 'benchmark' not found
>       available fixtures: cache, capfd, capfdbinary, caplog, capsys, capsysbinary, doctest_namespace, monkeypatch, pytestconfig, record_property, record_xml_attribute, record_xml_property, recwarn, tmpdir, tmpdir_factory
>       use 'pytest --fixtures [testpath]' for help on them.

/home/aryaman4/pyro/tests/perf/test_benchmark.py:128
_____________ ERROR at setup of test_benchmark[BernoulliBeta::HMC] _____________
file /home/aryaman4/pyro/tests/perf/test_benchmark.py, line 128
  @pytest.mark.parametrize('model, model_args, id', TEST_MODELS, ids=MODEL_IDS)
  @pytest.mark.benchmark(
      min_rounds=5,
      disable_gc=True,
  )
  @pytest.mark.disable_validation()
  def test_benchmark(benchmark, model, model_args, id):
E       fixture 'benchmark' not found
>       available fixtures: cache, capfd, capfdbinary, caplog, capsys, capsysbinary, doctest_namespace, monkeypatch, pytestconfig, record_property, record_xml_attribute, record_xml_property, recwarn, tmpdir, tmpdir_factory
>       use 'pytest --fixtures [testpath]' for help on them.

/home/aryaman4/pyro/tests/perf/test_benchmark.py:128
____________ ERROR at setup of test_benchmark[BernoulliBeta::NUTS] _____________
file /home/aryaman4/pyro/tests/perf/test_benchmark.py, line 128
  @pytest.mark.parametrize('model, model_args, id', TEST_MODELS, ids=MODEL_IDS)
  @pytest.mark.benchmark(
      min_rounds=5,
      disable_gc=True,
  )
  @pytest.mark.disable_validation()
  def test_benchmark(benchmark, model, model_args, id):
E       fixture 'benchmark' not found
>       available fixtures: cache, capfd, capfdbinary, caplog, capsys, capsysbinary, doctest_namespace, monkeypatch, pytestconfig, record_property, record_xml_attribute, record_xml_property, recwarn, tmpdir, tmpdir_factory
>       use 'pytest --fixtures [testpath]' for help on them.

/home/aryaman4/pyro/tests/perf/test_benchmark.py:128
________ ERROR at setup of test_benchmark[VSGP::MultiClass_whiten=True] ________
file /home/aryaman4/pyro/tests/perf/test_benchmark.py, line 128
  @pytest.mark.parametrize('model, model_args, id', TEST_MODELS, ids=MODEL_IDS)
  @pytest.mark.benchmark(
      min_rounds=5,
      disable_gc=True,
  )
  @pytest.mark.disable_validation()
  def test_benchmark(benchmark, model, model_args, id):
E       fixture 'benchmark' not found
>       available fixtures: cache, capfd, capfdbinary, caplog, capsys, capsysbinary, doctest_namespace, monkeypatch, pytestconfig, record_property, record_xml_attribute, record_xml_property, recwarn, tmpdir, tmpdir_factory
>       use 'pytest --fixtures [testpath]' for help on them.

/home/aryaman4/pyro/tests/perf/test_benchmark.py:128
_______ ERROR at setup of test_benchmark[VSGP::MultiClass_whiten=False] ________
file /home/aryaman4/pyro/tests/perf/test_benchmark.py, line 128
  @pytest.mark.parametrize('model, model_args, id', TEST_MODELS, ids=MODEL_IDS)
  @pytest.mark.benchmark(
      min_rounds=5,
      disable_gc=True,
  )
  @pytest.mark.disable_validation()
  def test_benchmark(benchmark, model, model_args, id):
E       fixture 'benchmark' not found
>       available fixtures: cache, capfd, capfdbinary, caplog, capsys, capsysbinary, doctest_namespace, monkeypatch, pytestconfig, record_property, record_xml_attribute, record_xml_property, recwarn, tmpdir, tmpdir_factory
>       use 'pytest --fixtures [testpath]' for help on them.

/home/aryaman4/pyro/tests/perf/test_benchmark.py:128
=================================== FAILURES ===================================
_____________________ test_cpu[air/main.py --num-steps=1] ______________________

example = ['air/main.py', '--num-steps=1']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/air/main.py', '--num-steps=1'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/air/main.py', '--num-steps=1']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/air/main.py', '--num-steps=1']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/air/main.py --num-steps=1
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/air/main.py", line 21, in <module>
    import visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/air/main.py --num-steps=1
______________ test_cpu[air/main.py --num-steps=1 --no-baseline] _______________

example = ['air/main.py', '--num-steps=1', '--no-baseline']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/air/main.py', '--num-steps=1', '--no-baseline'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/air/main.py', '--num-steps=1', '--no-baseline']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/air/main.py', '--num-steps=1', '--no-baseline']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/air/main.py --num-steps=1 --no-baseline
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/air/main.py", line 21, in <module>
    import visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/air/main.py --num-steps=1 --no-baseline
__ test_cpu[baseball.py --num-samples=200 --warmup-steps=100 --num-chains=2] ___

example = ['baseball.py', '--num-samples=200', '--warmup-steps=100', '--num-chains=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/baseball.py', '--num-samples=200', '--warmup-steps=100', '--num-chains=2'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/baseball.py', '--num-samples=200', '--warmup-steps=100', '--num-chains=2']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/baseball.py', '--num-samples=200', '--warmup-steps=100', '--num-chains=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/baseball.py --num-samples=200 --warmup-steps=100 --num-chains=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/baseball.py", line 8, in <module>
    import pandas as pd
ImportError: No module named 'pandas'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/baseball.py --num-samples=200 --warmup-steps=100 --num-chains=2
_______ test_cpu[contrib/forecast/bart.py --num-steps=2 --stride=99999] ________

example = ['contrib/forecast/bart.py', '--num-steps=2', '--stride=99999']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/contrib/forecast/bart.py', '--num-steps=2', '--stride=99999'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/contrib/forecast/bart.py', '--num-steps=2', '--stride=99999']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/contrib/forecast/bart.py', '--num-steps=2', '--stride=99999']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/contrib/forecast/bart.py --num-steps=2 --stride=99999
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/contrib/forecast/bart.py", line 12, in <module>
    from pyro.contrib.examples.bart import load_bart_od
  File "/home/aryaman4/pyro/pyro/contrib/examples/bart.py", line 17, in <module>
    from pyro.contrib.examples.util import get_data_directory, _mkdir_p
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/contrib/forecast/bart.py --num-steps=2 --stride=99999
_ test_cpu[contrib/gp/sv-dkl.py --epochs=1 --num-inducing=4 --batch-size=1000] _

example = ['contrib/gp/sv-dkl.py', '--epochs=1', '--num-inducing=4', '--batch-size=1000']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/contrib/gp/sv-dkl.py', '--epochs=1', '--num-inducing=4', '--batch-size=1000'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/contrib/gp/sv-dkl.py', '--epochs=1', '--num-inducing=4', '--batch-size=1000']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/contrib/gp/sv-dkl.py', '--epochs=1', '--num-inducing=4', '--batch-size=1000']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/contrib/gp/sv-dkl.py --epochs=1 --num-inducing=4 --batch-size=1000
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/contrib/gp/sv-dkl.py", line 37, in <module>
    from torchvision import transforms
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/contrib/gp/sv-dkl.py --epochs=1 --num-inducing=4 --batch-size=1000
_ test_cpu[contrib/gp/sv-dkl.py --binary --epochs=1 --num-inducing=4 --batch-size=1000] _

example = ['contrib/gp/sv-dkl.py', '--binary', '--epochs=1', '--num-inducing=4', '--batch-size=1000']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/contrib/gp/sv-dkl.py', '--binary', '--epochs=1', '--num-inducing=4', '--batch-size=1000'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/contrib/gp/sv-dkl.py', '--binary', '--epochs=1', '--num-inducing=4', '--batch-size=1000']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/contrib/gp/sv-dkl.py', '--binary', '--epochs=1', '--num-inducing=4', '--batch-size=1000']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/contrib/gp/sv-dkl.py --binary --epochs=1 --num-inducing=4 --batch-size=1000
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/contrib/gp/sv-dkl.py", line 37, in <module>
    from torchvision import transforms
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/contrib/gp/sv-dkl.py --binary --epochs=1 --num-inducing=4 --batch-size=1000
_____________________ test_cpu[dmm/dmm.py --num-epochs=1] ______________________

example = ['dmm/dmm.py', '--num-epochs=1']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/dmm/dmm.py --num-epochs=1
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/dmm/dmm.py", line 26, in <module>
    import polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/dmm/dmm.py --num-epochs=1
______ test_cpu[dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=20] ______

example = ['dmm/dmm.py', '--num-epochs=1', '--tmcelbo', '--tmc-num-samples=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--tmcelbo', '--tmc-num-samples=2'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--tmcelbo', '--tmc-num-samples=2']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--tmcelbo', '--tmc-num-samples=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/dmm/dmm.py", line 26, in <module>
    import polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=2
_______________ test_cpu[dmm/dmm.py --num-epochs=1 --num-iafs=1] _______________

example = ['dmm/dmm.py', '--num-epochs=1', '--num-iafs=1']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--num-iafs=1'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--num-iafs=1']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--num-iafs=1']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/dmm/dmm.py --num-epochs=1 --num-iafs=1
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/dmm/dmm.py", line 26, in <module>
    import polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/dmm/dmm.py --num-epochs=1 --num-iafs=1
________ test_cpu[dmm/dmm.py --num-epochs=1 --tmc --tmc-num-samples=2] _________

example = ['dmm/dmm.py', '--num-epochs=1', '--tmc', '--tmc-num-samples=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--tmc', '--tmc-num-samples=2'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--tmc', '--tmc-num-samples=2']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--tmc', '--tmc-num-samples=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/dmm/dmm.py --num-epochs=1 --tmc --tmc-num-samples=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/dmm/dmm.py", line 26, in <module>
    import polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/dmm/dmm.py --num-epochs=1 --tmc --tmc-num-samples=2
______ test_cpu[dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=21] ______

example = ['dmm/dmm.py', '--num-epochs=1', '--tmcelbo', '--tmc-num-samples=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--tmcelbo', '--tmc-num-samples=2'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--tmcelbo', '--tmc-num-samples=2']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/dmm/dmm.py', '--num-epochs=1', '--tmcelbo', '--tmc-num-samples=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/dmm/dmm.py", line 26, in <module>
    import polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=2
____________ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=0] ____________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=0']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=0'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=0']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=0']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=0
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=0
____________ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=1] ____________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=1']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=1'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=1']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=1']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=1
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=1
____________ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=2] ____________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=2'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=2']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=2
____________ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=3] ____________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=3']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=3'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=3']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=3']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=3
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=3
____________ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=4] ____________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=4']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=4'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=4']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=4']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=4
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=4
____________ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=5] ____________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=5']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=5'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=5']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=5']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=5
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=5
____________ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6] ____________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=6']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=6'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=6']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=6']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=6
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=6
_ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6 --raftery-parameterization] _

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=6', '--raftery-parameterization']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=6', '--raftery-parameterization'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=6', '--raftery-parameterization']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=6', '--raftery-parameterization']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=6 --raftery-parameterization
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=6 --raftery-parameterization
____________ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=7] ____________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=7']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=7'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=7']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=7']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=7
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=7
_ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=0 --tmc --tmc-num-samples=2] _

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=0', '--tmc', '--tmc-num-samples=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=0', '--tmc', ...],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=0', '--tmc', ...]

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=0', '--tmc', '--tmc-num-samples=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=0 --tmc --tmc-num-samples=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=0 --tmc --tmc-num-samples=2
_ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=1 --tmc --tmc-num-samples=2] _

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=1', '--tmc', '--tmc-num-samples=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=1', '--tmc', ...],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=1', '--tmc', ...]

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=1', '--tmc', '--tmc-num-samples=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=1 --tmc --tmc-num-samples=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=1 --tmc --tmc-num-samples=2
_ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=2 --tmc --tmc-num-samples=2] _

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=2', '--tmc', '--tmc-num-samples=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=2', '--tmc', ...],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=2', '--tmc', ...]

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=2', '--tmc', '--tmc-num-samples=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=2 --tmc --tmc-num-samples=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=2 --tmc --tmc-num-samples=2
_ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=3 --tmc --tmc-num-samples=2] _

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=3', '--tmc', '--tmc-num-samples=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=3', '--tmc', ...],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=3', '--tmc', ...]

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=3', '--tmc', '--tmc-num-samples=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=3 --tmc --tmc-num-samples=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=3 --tmc --tmc-num-samples=2
_ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=4 --tmc --tmc-num-samples=2] _

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=4', '--tmc', '--tmc-num-samples=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=4', '--tmc', ...],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=4', '--tmc', ...]

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=4', '--tmc', '--tmc-num-samples=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=4 --tmc --tmc-num-samples=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=4 --tmc --tmc-num-samples=2
_ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=5 --tmc --tmc-num-samples=2] _

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=5', '--tmc', '--tmc-num-samples=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=5', '--tmc', ...],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=5', '--tmc', ...]

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=5', '--tmc', '--tmc-num-samples=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=5 --tmc --tmc-num-samples=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=5 --tmc --tmc-num-samples=2
_ test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6 --tmc --tmc-num-samples=2] _

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=6', '--tmc', '--tmc-num-samples=2']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=6', '--tmc', ...],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=6', '--tmc', ...]

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=6', '--tmc', '--tmc-num-samples=2']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=6 --tmc --tmc-num-samples=2
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=6 --tmc --tmc-num-samples=2
_______________ test_cpu[mixed_hmm/experiment.py --timesteps=1] ________________

example = ['mixed_hmm/experiment.py', '--timesteps=1']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/mixed_hmm/experiment.py', '--timesteps=1'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/mixed_hmm/experiment.py', '--timesteps=1']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/mixed_hmm/experiment.py', '--timesteps=1']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/mixed_hmm/experiment.py --timesteps=1
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/mixed_hmm/experiment.py", line 17, in <module>
    from seal_data import prepare_seal
  File "/home/aryaman4/pyro/examples/mixed_hmm/seal_data.py", line 7, in <module>
    import pandas as pd
ImportError: No module named 'pandas'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/mixed_hmm/experiment.py --timesteps=1
__________ test_cpu[neutra.py -n 10 --num-warmup 10 --num-samples 10] __________

example = ['neutra.py', '-n', '10', '--num-warmup', '10', '--num-samples', ...]

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/neutra.py', '-n', '10', '--num-warmup', '10', ...],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/neutra.py', '-n', '10', '--num-warmup', '10', ...]

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/neutra.py', '-n', '10', '--num-warmup', '10', '--num-samples', '10']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/neutra.py -n 10 --num-warmup 10 --num-samples 10
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/neutra.py", line 28, in <module>
    import matplotlib.pyplot as plt
ImportError: No module named 'matplotlib'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/neutra.py -n 10 --num-warmup 10 --num-samples 10
_ test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide custom] _

example = ['sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', 'custom']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', ...],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', ...]

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', 'custom']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide custom
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/sparse_gamma_def.py", line 27, in <module>
    import wget
ImportError: No module named 'wget'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide custom
_ test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide auto] _

example = ['sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', 'auto']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', ...],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', ...]

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', 'auto']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide auto
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/sparse_gamma_def.py", line 27, in <module>
    import wget
ImportError: No module named 'wget'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide auto
_ test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide easy] _

example = ['sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', 'easy']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', ...],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', ...]

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/sparse_gamma_def.py', '--num-epochs=2', '--eval-particles=2', '--eval-frequency=1', '--guide', 'easy']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide easy
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/sparse_gamma_def.py", line 27, in <module>
    import wget
ImportError: No module named 'wget'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide easy
______ test_cpu[toy_mixture_model_discrete_enumeration.py  --num-steps=1] ______

example = ['toy_mixture_model_discrete_enumeration.py', '--num-steps=1']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/toy_mixture_model_discrete_enumeration.py', '--num-steps=1'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/toy_mixture_model_discrete_enumeration.py', '--num-steps=1']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/toy_mixture_model_discrete_enumeration.py', '--num-steps=1']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/toy_mixture_model_discrete_enumeration.py  --num-steps=1
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/toy_mixture_model_discrete_enumeration.py", line 25, in <module>
    import matplotlib.pyplot as plt
ImportError: No module named 'matplotlib'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/toy_mixture_model_discrete_enumeration.py  --num-steps=1
__________________ test_cpu[vae/ss_vae_M2.py --num-epochs=1] ___________________

example = ['vae/ss_vae_M2.py', '--num-epochs=1']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/vae/ss_vae_M2.py --num-epochs=1
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/vae/ss_vae_M2.py", line 8, in <module>
    from visdom import Visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/vae/ss_vae_M2.py --num-epochs=1
_____________ test_cpu[vae/ss_vae_M2.py --num-epochs=1 --aux-loss] _____________

example = ['vae/ss_vae_M2.py', '--num-epochs=1', '--aux-loss']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--aux-loss'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--aux-loss']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--aux-loss']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --aux-loss
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/vae/ss_vae_M2.py", line 8, in <module>
    from visdom import Visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --aux-loss
______ test_cpu[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel] ______

example = ['vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=parallel']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=parallel'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=parallel']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=parallel']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/vae/ss_vae_M2.py", line 8, in <module>
    from visdom import Visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel
_____ test_cpu[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential] _____

example = ['vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=sequential']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=sequential'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=sequential']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=sequential']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/vae/ss_vae_M2.py", line 8, in <module>
    from visdom import Visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential
_____________________ test_cpu[vae/vae.py --num-epochs=1] ______________________

example = ['vae/vae.py', '--num-epochs=1']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae.py', '--num-epochs=1'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae.py', '--num-epochs=1']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae.py', '--num-epochs=1']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/vae/vae.py --num-epochs=1
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/vae/vae.py", line 9, in <module>
    import visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/vae/vae.py --num-epochs=1
________________ test_cpu[vae/vae_comparison.py --num-epochs=1] ________________

example = ['vae/vae_comparison.py', '--num-epochs=1']

    @pytest.mark.parametrize('example', CPU_EXAMPLES)
    def test_cpu(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:226: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae_comparison.py', '--num-epochs=1'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae_comparison.py', '--num-epochs=1']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae_comparison.py', '--num-epochs=1']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/vae/vae_comparison.py --num-epochs=1
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/vae/vae_comparison.py", line 12, in <module>
    from torchvision.utils import save_image
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           222 INFO     Running:
python examples/vae/vae_comparison.py --num-epochs=1
__________________ test_jit[air/main.py --num-steps=1 --jit] ___________________

example = ['air/main.py', '--num-steps=1', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/air/main.py', '--num-steps=1', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/air/main.py', '--num-steps=1', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/air/main.py', '--num-steps=1', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/air/main.py --num-steps=1 --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/air/main.py", line 21, in <module>
    import visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/air/main.py --num-steps=1 --jit
_______ test_jit[baseball.py --num-samples=200 --warmup-steps=100 --jit] _______

example = ['baseball.py', '--num-samples=200', '--warmup-steps=100', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/baseball.py', '--num-samples=200', '--warmup-steps=100', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/baseball.py', '--num-samples=200', '--warmup-steps=100', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/baseball.py', '--num-samples=200', '--warmup-steps=100', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/baseball.py --num-samples=200 --warmup-steps=100 --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/baseball.py", line 8, in <module>
    import pandas as pd
ImportError: No module named 'pandas'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/baseball.py --num-samples=200 --warmup-steps=100 --jit
_________ test_jit[hmm.py --num-steps=1 --truncate=10 --model=1 --jit] _________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=1', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=1', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=1', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=1', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=1 --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=1 --jit
_________ test_jit[hmm.py --num-steps=1 --truncate=10 --model=2 --jit] _________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=2', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=2', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=2', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=2', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=2 --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=2 --jit
_________ test_jit[hmm.py --num-steps=1 --truncate=10 --model=3 --jit] _________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=3', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=3', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=3', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=3', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=3 --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=3 --jit
_________ test_jit[hmm.py --num-steps=1 --truncate=10 --model=4 --jit] _________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=4', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=4', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=4', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=4', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=4 --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=4 --jit
_________ test_jit[hmm.py --num-steps=1 --truncate=10 --model=5 --jit] _________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=5', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=5', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=5', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=5', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=5 --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=5 --jit
_________ test_jit[hmm.py --num-steps=1 --truncate=10 --model=7 --jit] _________

example = ['hmm.py', '--num-steps=1', '--truncate=10', '--model=7', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=7', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=7', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/hmm.py', '--num-steps=1', '--truncate=10', '--model=7', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=7 --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/hmm.py", line 46, in <module>
    import dmm.polyphonic_data_loader as poly
  File "/home/aryaman4/pyro/examples/dmm/polyphonic_data_loader.py", line 28, in <module>
    from pyro.contrib.examples.util import get_data_directory
  File "/home/aryaman4/pyro/pyro/contrib/examples/util.py", line 7, in <module>
    import torchvision
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/hmm.py --num-steps=1 --truncate=10 --model=7 --jit
___ test_jit[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel --jit] ___

example = ['vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=parallel', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=parallel', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=parallel', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=parallel', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/vae/ss_vae_M2.py", line 8, in <module>
    from visdom import Visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel --jit
__ test_jit[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential --jit] __

example = ['vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=sequential', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=sequential', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=sequential', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--enum-discrete=sequential', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/vae/ss_vae_M2.py", line 8, in <module>
    from visdom import Visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential --jit
_______________ test_jit[vae/ss_vae_M2.py --num-epochs=1 --jit] ________________

example = ['vae/ss_vae_M2.py', '--num-epochs=1', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/ss_vae_M2.py', '--num-epochs=1', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/vae/ss_vae_M2.py", line 8, in <module>
    from visdom import Visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/vae/ss_vae_M2.py --num-epochs=1 --jit
__________________ test_jit[vae/vae.py --num-epochs=1 --jit] ___________________

example = ['vae/vae.py', '--num-epochs=1', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae.py', '--num-epochs=1', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae.py', '--num-epochs=1', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae.py', '--num-epochs=1', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/vae/vae.py --num-epochs=1 --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/vae/vae.py", line 9, in <module>
    import visdom
ImportError: No module named 'visdom'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/vae/vae.py --num-epochs=1 --jit
_____________ test_jit[vae/vae_comparison.py --num-epochs=1 --jit] _____________

example = ['vae/vae_comparison.py', '--num-epochs=1', '--jit']

    @pytest.mark.parametrize('example', JIT_EXAMPLES)
    def test_jit(example):
        logger.info('Running:\npython examples/{}'.format(example))
        example = example.split()
        filename, args = example[0], example[1:]
        filename = os.path.join(EXAMPLES_DIR, filename)
>       check_call([sys.executable, filename] + args)

tests/test_examples.py:245: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

popenargs = (['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae_comparison.py', '--num-epochs=1', '--jit'],)
kwargs = {}, retcode = 1
cmd = ['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae_comparison.py', '--num-epochs=1', '--jit']

    def check_call(*popenargs, **kwargs):
        """Run command with arguments.  Wait for command to complete.  If
        the exit code was zero then return, otherwise raise
        CalledProcessError.  The CalledProcessError object will have the
        return code in the returncode attribute.
    
        The arguments are the same as for the call function.  Example:
    
        check_call(["ls", "-l"])
        """
        retcode = call(*popenargs, **kwargs)
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
>           raise CalledProcessError(retcode, cmd)
E           subprocess.CalledProcessError: Command '['/home/aryaman4/anaconda3/envs/pyro/bin/python', '/home/aryaman4/pyro/examples/vae/vae_comparison.py', '--num-epochs=1', '--jit']' returned non-zero exit status 1

../anaconda3/envs/pyro/lib/python3.5/subprocess.py:271: CalledProcessError
----------------------------- Captured stderr call -----------------------------
I 	 Running:
python examples/vae/vae_comparison.py --num-epochs=1 --jit
Traceback (most recent call last):
  File "/home/aryaman4/pyro/examples/vae/vae_comparison.py", line 12, in <module>
    from torchvision.utils import save_image
ImportError: No module named 'torchvision'
------------------------------ Captured log call -------------------------------
test_examples.py           241 INFO     Running:
python examples/vae/vae_comparison.py --num-epochs=1 --jit
________________ test_make_complete_graph[cpp-2-expected_grid0] ________________

num_vertices = 2, expected_grid = tensor([[0],
        [1]]), backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_vertices,expected_grid', [
        (2, [[0], [1]]),
        (3, [[0, 0, 1], [1, 2, 2]]),
        (4, [[0, 0, 1, 0, 1, 2], [1, 2, 2, 3, 3, 3]]),
    ])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_make_complete_graph(num_vertices, expected_grid, backend):
        V = num_vertices
        K = V * (V - 1) // 2
        expected_grid = torch.tensor(expected_grid, dtype=torch.long).reshape(2, K)
    
>       grid = make_complete_graph(V, backend=backend)

tests/distributions/test_spanning_tree.py:30: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:204: in make_complete_graph
    return _get_cpp_module().make_complete_graph(num_vertices)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1109: in _jit_compile
    with_cuda=with_cuda)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1176: in _write_ninja_file_and_build_library
    verify_ninja_availability()
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

    def verify_ninja_availability():
        '''
        Returns ``True`` if the `ninja <https://ninja-build.org/>`_ build system is
        available on the system.
        '''
        if not _is_ninja_available():
>           raise RuntimeError("Ninja is required to load C++ extensions")
E           RuntimeError: Ninja is required to load C++ extensions

../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1228: RuntimeError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
Creating extension directory /tmp/torch_extensions/cpp_spanning_tree...
________________ test_make_complete_graph[cpp-3-expected_grid1] ________________

num_vertices = 3, expected_grid = tensor([[0, 0, 1],
        [1, 2, 2]])
backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_vertices,expected_grid', [
        (2, [[0], [1]]),
        (3, [[0, 0, 1], [1, 2, 2]]),
        (4, [[0, 0, 1, 0, 1, 2], [1, 2, 2, 3, 3, 3]]),
    ])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_make_complete_graph(num_vertices, expected_grid, backend):
        V = num_vertices
        K = V * (V - 1) // 2
        expected_grid = torch.tensor(expected_grid, dtype=torch.long).reshape(2, K)
    
>       grid = make_complete_graph(V, backend=backend)

tests/distributions/test_spanning_tree.py:30: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:204: in make_complete_graph
    return _get_cpp_module().make_complete_graph(num_vertices)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
________________ test_make_complete_graph[cpp-4-expected_grid2] ________________

num_vertices = 4
expected_grid = tensor([[0, 0, 1, 0, 1, 2],
        [1, 2, 2, 3, 3, 3]])
backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_vertices,expected_grid', [
        (2, [[0], [1]]),
        (3, [[0, 0, 1], [1, 2, 2]]),
        (4, [[0, 0, 1, 0, 1, 2], [1, 2, 2, 3, 3, 3]]),
    ])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_make_complete_graph(num_vertices, expected_grid, backend):
        V = num_vertices
        K = V * (V - 1) // 2
        expected_grid = torch.tensor(expected_grid, dtype=torch.long).reshape(2, K)
    
>       grid = make_complete_graph(V, backend=backend)

tests/distributions/test_spanning_tree.py:30: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:204: in make_complete_graph
    return _get_cpp_module().make_complete_graph(num_vertices)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
______________________ test_sample_tree_mcmc_smoke[cpp-1] ______________________

num_edges = 1, backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_edges', [1, 3, 10, 30, 100])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_sample_tree_mcmc_smoke(num_edges, backend):
        pyro.set_rng_seed(num_edges)
        E = num_edges
        V = 1 + E
        K = V * (V - 1) // 2
        edge_logits = torch.rand(K)
        edges = torch.tensor([(v, v + 1) for v in range(V - 1)], dtype=torch.long)
        for _ in range(10 if backend == "cpp" or num_edges <= 30 else 1):
>           edges = sample_tree(edge_logits, edges, backend=backend)

tests/distributions/test_spanning_tree.py:45: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:426: in sample_tree
    edges = sample_tree_mcmc(edge_logits, edges, backend=backend)
pyro/distributions/spanning_tree.py:357: in sample_tree_mcmc
    return _get_cpp_module().sample_tree_mcmc(edge_logits, edges)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
______________________ test_sample_tree_mcmc_smoke[cpp-3] ______________________

num_edges = 3, backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_edges', [1, 3, 10, 30, 100])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_sample_tree_mcmc_smoke(num_edges, backend):
        pyro.set_rng_seed(num_edges)
        E = num_edges
        V = 1 + E
        K = V * (V - 1) // 2
        edge_logits = torch.rand(K)
        edges = torch.tensor([(v, v + 1) for v in range(V - 1)], dtype=torch.long)
        for _ in range(10 if backend == "cpp" or num_edges <= 30 else 1):
>           edges = sample_tree(edge_logits, edges, backend=backend)

tests/distributions/test_spanning_tree.py:45: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:426: in sample_tree
    edges = sample_tree_mcmc(edge_logits, edges, backend=backend)
pyro/distributions/spanning_tree.py:357: in sample_tree_mcmc
    return _get_cpp_module().sample_tree_mcmc(edge_logits, edges)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
_____________________ test_sample_tree_mcmc_smoke[cpp-10] ______________________

num_edges = 10, backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_edges', [1, 3, 10, 30, 100])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_sample_tree_mcmc_smoke(num_edges, backend):
        pyro.set_rng_seed(num_edges)
        E = num_edges
        V = 1 + E
        K = V * (V - 1) // 2
        edge_logits = torch.rand(K)
        edges = torch.tensor([(v, v + 1) for v in range(V - 1)], dtype=torch.long)
        for _ in range(10 if backend == "cpp" or num_edges <= 30 else 1):
>           edges = sample_tree(edge_logits, edges, backend=backend)

tests/distributions/test_spanning_tree.py:45: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:426: in sample_tree
    edges = sample_tree_mcmc(edge_logits, edges, backend=backend)
pyro/distributions/spanning_tree.py:357: in sample_tree_mcmc
    return _get_cpp_module().sample_tree_mcmc(edge_logits, edges)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
_____________________ test_sample_tree_mcmc_smoke[cpp-30] ______________________

num_edges = 30, backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_edges', [1, 3, 10, 30, 100])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_sample_tree_mcmc_smoke(num_edges, backend):
        pyro.set_rng_seed(num_edges)
        E = num_edges
        V = 1 + E
        K = V * (V - 1) // 2
        edge_logits = torch.rand(K)
        edges = torch.tensor([(v, v + 1) for v in range(V - 1)], dtype=torch.long)
        for _ in range(10 if backend == "cpp" or num_edges <= 30 else 1):
>           edges = sample_tree(edge_logits, edges, backend=backend)

tests/distributions/test_spanning_tree.py:45: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:426: in sample_tree
    edges = sample_tree_mcmc(edge_logits, edges, backend=backend)
pyro/distributions/spanning_tree.py:357: in sample_tree_mcmc
    return _get_cpp_module().sample_tree_mcmc(edge_logits, edges)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
_____________________ test_sample_tree_mcmc_smoke[cpp-100] _____________________

num_edges = 100, backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_edges', [1, 3, 10, 30, 100])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_sample_tree_mcmc_smoke(num_edges, backend):
        pyro.set_rng_seed(num_edges)
        E = num_edges
        V = 1 + E
        K = V * (V - 1) // 2
        edge_logits = torch.rand(K)
        edges = torch.tensor([(v, v + 1) for v in range(V - 1)], dtype=torch.long)
        for _ in range(10 if backend == "cpp" or num_edges <= 30 else 1):
>           edges = sample_tree(edge_logits, edges, backend=backend)

tests/distributions/test_spanning_tree.py:45: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:426: in sample_tree
    edges = sample_tree_mcmc(edge_logits, edges, backend=backend)
pyro/distributions/spanning_tree.py:357: in sample_tree_mcmc
    return _get_cpp_module().sample_tree_mcmc(edge_logits, edges)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
_____________________ test_sample_tree_approx_smoke[cpp-1] _____________________

num_edges = 1, backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_edges', [1, 3, 10, 30, 100])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_sample_tree_approx_smoke(num_edges, backend):
        pyro.set_rng_seed(num_edges)
        E = num_edges
        V = 1 + E
        K = V * (V - 1) // 2
        edge_logits = torch.rand(K)
        for _ in range(10 if backend == "cpp" or num_edges <= 30 else 1):
>           sample_tree(edge_logits, backend=backend)

tests/distributions/test_spanning_tree.py:58: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:424: in sample_tree
    edges = sample_tree_approx(edge_logits, backend=backend)
pyro/distributions/spanning_tree.py:416: in sample_tree_approx
    return _get_cpp_module().sample_tree_approx(edge_logits)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
_____________________ test_sample_tree_approx_smoke[cpp-3] _____________________

num_edges = 3, backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_edges', [1, 3, 10, 30, 100])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_sample_tree_approx_smoke(num_edges, backend):
        pyro.set_rng_seed(num_edges)
        E = num_edges
        V = 1 + E
        K = V * (V - 1) // 2
        edge_logits = torch.rand(K)
        for _ in range(10 if backend == "cpp" or num_edges <= 30 else 1):
>           sample_tree(edge_logits, backend=backend)

tests/distributions/test_spanning_tree.py:58: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:424: in sample_tree
    edges = sample_tree_approx(edge_logits, backend=backend)
pyro/distributions/spanning_tree.py:416: in sample_tree_approx
    return _get_cpp_module().sample_tree_approx(edge_logits)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
____________________ test_sample_tree_approx_smoke[cpp-10] _____________________

num_edges = 10, backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_edges', [1, 3, 10, 30, 100])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_sample_tree_approx_smoke(num_edges, backend):
        pyro.set_rng_seed(num_edges)
        E = num_edges
        V = 1 + E
        K = V * (V - 1) // 2
        edge_logits = torch.rand(K)
        for _ in range(10 if backend == "cpp" or num_edges <= 30 else 1):
>           sample_tree(edge_logits, backend=backend)

tests/distributions/test_spanning_tree.py:58: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:424: in sample_tree
    edges = sample_tree_approx(edge_logits, backend=backend)
pyro/distributions/spanning_tree.py:416: in sample_tree_approx
    return _get_cpp_module().sample_tree_approx(edge_logits)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
____________________ test_sample_tree_approx_smoke[cpp-30] _____________________

num_edges = 30, backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_edges', [1, 3, 10, 30, 100])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_sample_tree_approx_smoke(num_edges, backend):
        pyro.set_rng_seed(num_edges)
        E = num_edges
        V = 1 + E
        K = V * (V - 1) // 2
        edge_logits = torch.rand(K)
        for _ in range(10 if backend == "cpp" or num_edges <= 30 else 1):
>           sample_tree(edge_logits, backend=backend)

tests/distributions/test_spanning_tree.py:58: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:424: in sample_tree
    edges = sample_tree_approx(edge_logits, backend=backend)
pyro/distributions/spanning_tree.py:416: in sample_tree_approx
    return _get_cpp_module().sample_tree_approx(edge_logits)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
____________________ test_sample_tree_approx_smoke[cpp-100] ____________________

num_edges = 100, backend = 'cpp'

    @pytest.mark.filterwarnings("always")
    @pytest.mark.parametrize('num_edges', [1, 3, 10, 30, 100])
    @pytest.mark.parametrize('backend', ["python", "cpp"])
    def test_sample_tree_approx_smoke(num_edges, backend):
        pyro.set_rng_seed(num_edges)
        E = num_edges
        V = 1 + E
        K = V * (V - 1) // 2
        edge_logits = torch.rand(K)
        for _ in range(10 if backend == "cpp" or num_edges <= 30 else 1):
>           sample_tree(edge_logits, backend=backend)

tests/distributions/test_spanning_tree.py:58: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
pyro/distributions/spanning_tree.py:424: in sample_tree
    edges = sample_tree_approx(edge_logits, backend=backend)
pyro/distributions/spanning_tree.py:416: in sample_tree_approx
    return _get_cpp_module().sample_tree_approx(edge_logits)
pyro/distributions/spanning_tree.py:188: in _get_cpp_module
    verbose=True)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:921: in load
    is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1120: in _jit_compile
    return _import_module_from_library(name, build_directory, is_python_module)
../anaconda3/envs/pyro/lib/python3.5/site-packages/torch/utils/cpp_extension.py:1441: in _import_module_from_library
    file, path, description = imp.find_module(module_name, [path])
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

name = 'cpp_spanning_tree', path = ['/tmp/torch_extensions/cpp_spanning_tree']

    def find_module(name, path=None):
        """**DEPRECATED**
    
        Search for a module.
    
        If path is omitted or None, search for a built-in, frozen or special
        module and continue search in sys.path. The module name cannot
        contain '.'; to search for a submodule of a package, pass the
        submodule name and the package's __path__.
    
        """
        if not isinstance(name, str):
            raise TypeError("'name' must be a str, not {}".format(type(name)))
        elif not isinstance(path, (type(None), list)):
            # Backwards-compatibility
            raise RuntimeError("'path' must be None or a list, "
                               "not {}".format(type(path)))
    
        if path is None:
            if is_builtin(name):
                return None, None, ('', '', C_BUILTIN)
            elif is_frozen(name):
                return None, None, ('', '', PY_FROZEN)
            else:
                path = sys.path
    
        for entry in path:
            package_directory = os.path.join(entry, name)
            for suffix in ['.py', machinery.BYTECODE_SUFFIXES[0]]:
                package_file_name = '__init__' + suffix
                file_path = os.path.join(package_directory, package_file_name)
                if os.path.isfile(file_path):
                    return None, package_directory, ('', '', PKG_DIRECTORY)
            for suffix, mode, type_ in get_suffixes():
                file_name = name + suffix
                file_path = os.path.join(entry, file_name)
                if os.path.isfile(file_path):
                    break
            else:
                continue
            break  # Break out of outer loop when breaking out of inner loop.
        else:
>           raise ImportError(_ERR_MSG.format(name), name=name)
E           ImportError: No module named 'cpp_spanning_tree'

../anaconda3/envs/pyro/lib/python3.5/imp.py:297: ImportError
----------------------------- Captured stdout call -----------------------------
Using /tmp/torch_extensions as PyTorch extensions root...
No modifications detected for re-loaded extension module cpp_spanning_tree, skipping build step...
Loading extension module cpp_spanning_tree...
_____________________________ test_pyro_serialize ______________________________

    def test_pyro_serialize():
        class MyModule(PyroModule):
            def __init__(self):
                super().__init__()
                self.x = PyroParam(torch.tensor(1.234), constraints.positive)
                self.y = nn.Parameter(torch.randn(3))
    
            def forward(self):
                return self.x, self.y
    
        module = MyModule()
        assert len(pyro.get_param_store()) == 0
    
        assert isinstance(module.x, torch.Tensor)
        assert len(pyro.get_param_store()) == 0
    
        actual = module()  # triggers saving in param store
        assert_equal(actual[0], module.x)
        assert_equal(actual[1], module.y)
        assert set(pyro.get_param_store().keys()) == {"x", "y"}
        assert_equal(pyro.param("x").detach(), module.x.detach())
        assert_equal(pyro.param("y").detach(), module.y.detach())
    
>       pyro.get_param_store().save("/tmp/pyro_module.pt")

tests/nn/test_module.py:583: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <pyro.params.param_store.ParamStoreDict object at 0x7f6d86dba208>
filename = '/tmp/pyro_module.pt'

    def save(self, filename):
        """
            Save parameters to disk
    
            :param filename: file name to save to
            :type filename: str
            """
>       with open(filename, "wb") as output_file:
E       PermissionError: [Errno 13] Permission denied: '/tmp/pyro_module.pt'

pyro/params/param_store.py:267: PermissionError
============================ slowest test durations ============================
544.51s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_reparameterized_four_layers
113.17s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_three_layers
80.95s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_three_layers_model_permuted
56.22s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_reparameterized_N_is_17
45.73s call     tests/contrib/autoguide/test_inference.py::AutoGaussianChain::test_multivariatate_normal_auto
44.60s call     tests/infer/test_inference.py::test_exponential_gamma[nonreparam-RenyiELBO]
43.15s call     tests/infer/test_inference.py::PoissonGammaTests::test_elbo_nonreparameterized
40.81s call     tests/infer/test_inference.py::PoissonGammaTests::test_renyi_nonreparameterized
35.73s call     tests/infer/mcmc/test_nuts.py::test_beta_binomial[True]
33.70s call     tests/infer/test_inference.py::test_exponential_gamma[nonreparam-TraceEnum_ELBO]
30.27s call     tests/infer/test_inference.py::test_exponential_gamma[rsvi-TraceEnum_ELBO]
28.32s call     tests/infer/test_inference.py::PoissonGammaTests::test_rws_nonreparameterized
27.59s call     tests/infer/test_inference.py::NormalNormalTests::test_elbo_nonreparameterized
25.80s call     tests/test_examples.py::test_cpu[contrib/oed/ab_test.py --num-vi-steps=10 --num-bo-steps=2]
25.56s call     tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_reparameterized
25.26s call     tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoLowRankMultivariateNormal]
25.15s call     tests/infer/test_inference.py::test_exponential_gamma[reparam-RenyiELBO]
24.04s call     tests/integration_tests/test_tracegraph_elbo.py::RaoBlackwellizationTests::test_plate_in_elbo_with_superfluous_rvs
23.80s call     tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoLowRankMultivariateNormal]
23.33s call     tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_reparameterized
23.08s call     tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[iplate-3-7000-0.0008]
22.63s call     tests/infer/test_inference.py::test_exponential_gamma[rsvi-Trace_ELBO]
22.51s call     tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoMultivariateNormal]
21.98s call     tests/infer/test_inference.py::test_exponential_gamma[nonreparam-ReweightedWakeSleep]
21.40s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_reparameterized_three_layers
21.02s call     tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoMultivariateNormal]
20.90s call     tests/integration_tests/test_tracegraph_elbo.py::RaoBlackwellizationTests::test_nested_iplate_in_elbo
20.59s call     tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_nonreparameterized
20.49s call     tests/infer/test_inference.py::PoissonGammaTests::test_elbo_reparameterized
20.27s call     tests/infer/test_inference.py::test_exponential_gamma[rsvi-TraceGraph_ELBO]
19.51s call     tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_nonreparameterized
18.55s call     tests/infer/test_inference.py::PoissonGammaTests::test_renyi_reparameterized
18.02s call     tests/infer/mcmc/test_nuts.py::test_gaussian_hmm[30]
17.72s call     tests/infer/test_inference.py::NormalNormalTests::test_rws_nonreparameterized
17.70s call     tests/infer/test_inference.py::NormalNormalTests::test_renyi_nonreparameterized
17.42s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_nonreparameterized_N_is_7
17.18s call     tests/infer/test_inference.py::test_exponential_gamma[nonreparam-TraceGraph_ELBO]
16.93s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-<lambda>]
16.83s call     tests/test_examples.py::test_cpu[eight_schools/mcmc.py --num-samples=500 --warmup-steps=100]
16.75s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-<lambda>]
16.47s call     tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[plate-3-7000-0.0008]
16.35s call     tests/infer/test_inference.py::test_exponential_gamma[nonreparam-Trace_ELBO]
16.32s call     tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_nonreparameterized_nn_baseline
16.26s call     tests/integration_tests/test_tracegraph_elbo.py::ExponentialGammaTests::test_elbo_reparameterized
16.23s call     tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[plate-8-7000-0.0008]
16.20s call     tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[plate-None-7000-0.0008]
15.76s call     tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_nonreparameterized_decaying_baseline
15.70s call     tests/infer/test_inference.py::NormalNormalTests::test_mmd_nonvectorized
15.09s call     tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_reparameterized_vectorized
15.00s call     tests/integration_tests/test_tracegraph_elbo.py::ExponentialGammaTests::test_elbo_nonreparameterized
14.86s call     tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_reparameterized_vectorized
14.17s call     tests/infer/test_inference.py::test_exponential_gamma[reparam-TraceEnum_ELBO]
14.14s call     tests/integration_tests/test_tracegraph_elbo.py::NormalNormalTests::test_elbo_nonreparameterized
14.03s call     tests/infer/test_inference.py::BernoulliBetaTests::test_mmd_vectorized
13.99s call     tests/infer/mcmc/test_nuts.py::test_gaussian_mixture_model[JIT=False]
13.79s call     tests/infer/test_predictive.py::test_posterior_predictive_svi_manual_guide[False]
13.62s call     tests/infer/test_inference.py::NormalNormalTests::test_renyi_reparameterized
13.51s call     tests/distributions/test_relaxed_straight_through.py::test_onehot_svi_usage
13.38s call     tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoLowRankMultivariateNormal]
13.36s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-subsample-reparam]
13.34s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-full-reparam]
13.24s call     tests/infer/reparam/test_neutra.py::test_neals_funnel_smoke[False]
12.99s call     tests/infer/test_inference.py::BernoulliBetaTests::test_rws_nonreparameterized_vectorized
12.95s call     tests/infer/test_inference.py::BernoulliBetaTests::test_rws_reparameterized_vectorized
12.95s call     tests/infer/test_inference.py::BernoulliBetaTests::test_rws_reparameterized
12.94s call     tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_nonreparameterized_vectorized
12.94s call     tests/infer/test_inference.py::BernoulliBetaTests::test_rws_nonreparameterized
12.72s call     tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_delta_guide[False]
12.72s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-full-nonreparam]
12.68s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-subsample-nonreparam]
12.62s call     tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoLaplaceApproximation]
12.41s call     tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_nonreparameterized_vectorized
12.39s call     tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoLaplaceApproximation]
12.20s call     tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_nonreparameterized_both_baselines
11.40s call     tests/infer/test_inference.py::test_energy_distance_multivariate[1]
11.32s call     tests/infer/test_inference.py::PoissonGammaTests::test_rws_reparameterized
11.27s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_reparameterized_N_is_8
11.18s call     tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoMultivariateNormal]
11.08s call     tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_reparameterized
11.07s call     tests/infer/test_predictive.py::test_posterior_predictive_svi_one_hot
11.02s call     tests/test_examples.py::test_jit[eight_schools/mcmc.py --num-samples=500 --warmup-steps=100 --jit]
10.88s call     tests/infer/test_inference.py::test_exponential_gamma[reparam-ReweightedWakeSleep]
10.77s call     tests/infer/test_inference.py::NormalNormalTests::test_elbo_reparameterized
10.76s call     tests/test_examples.py::test_cpu[lkj.py --n=50 --num-chains=1 --warmup-steps=100 --num-samples=200]
10.53s call     tests/infer/test_inference.py::test_energy_distance_multivariate[0]
10.16s call     tests/infer/mcmc/test_nuts.py::test_bernoulli_latent_model[JIT=False]
10.03s call     tests/infer/test_inference.py::test_exponential_gamma[reparam-TraceGraph_ELBO]
9.85s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-subsample-reparam]
9.76s call     tests/infer/test_inference.py::test_exponential_gamma[reparam-Trace_ELBO]
9.71s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-subsample-reparam]
9.67s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-full-reparam]
9.67s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoLowRankMultivariateNormal]
9.56s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-full-reparam]
9.36s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoLowRankMultivariateNormal]
9.34s call     tests/infer/mcmc/test_nuts.py::test_arrowhead_mass
9.16s call     tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoDiagonalNormal]
9.06s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-subsample-reparam]
9.03s call     tests/infer/test_sampling.py::ImportanceTest::test_importance_prior
8.93s call     tests/infer/mcmc/test_hmc.py::test_logistic_regression[None-1-None-True-False-False]
8.87s call     tests/infer/mcmc/test_nuts.py::test_gaussian_mixture_model[JIT=True]
8.82s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-full-reparam]
8.68s call     tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoDiagonalNormal]
8.61s call     tests/infer/reparam/test_neutra.py::test_neals_funnel_smoke[True]
8.59s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-subsample-nonreparam]
8.49s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-full-nonreparam]
8.34s call     tests/infer/test_inference.py::test_energy_distance_univariate[0.0001]
8.26s call     tests/contrib/gp/test_models.py::test_mean_function_VGP
8.26s call     tests/infer/mcmc/test_nuts.py::test_gamma_beta[JIT=False]
8.24s call     tests/infer/mcmc/test_nuts.py::test_structured_mass
8.14s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-subsample-nonreparam]
8.11s call     tests/integration_tests/test_tracegraph_elbo.py::BernoulliBetaTests::test_elbo_reparameterized
8.09s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoMultivariateNormal]
8.03s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-full-nonreparam]
7.99s call     tests/contrib/gp/test_models.py::test_mean_function_VGP_whiten
7.83s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_nonreparameterized_N_is_3
7.75s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoIAFNormal]
7.72s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoMultivariateNormal]
7.72s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoIAFNormal]
7.63s call     tests/infer/test_inference.py::NormalNormalTests::test_elbo_tail_adaptive
7.56s call     tests/contrib/gp/test_models.py::test_inference_whiten_vsgp
7.50s call     tests/contrib/gp/test_models.py::test_inference_vsgp
7.35s call     tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoDiagonalNormal]
7.34s call     tests/contrib/gp/test_models.py::test_mean_function_VSGP
7.06s call     tests/integration_tests/test_tracegraph_elbo.py::BernoulliBetaTests::test_elbo_nonreparameterized
7.02s call     tests/infer/test_inference.py::test_energy_distance_univariate[0]
7.02s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_nonreparameterized_N_is_5
6.98s call     tests/contrib/gp/test_models.py::test_mean_function_VSGP_whiten
6.43s call     tests/infer/mcmc/test_nuts.py::test_gamma_poisson[True]
6.42s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoDiagonalNormal]
6.33s call     tests/infer/test_inference.py::NormalNormalTests::test_elbo_analytic_kl
6.33s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoDiagonalNormal]
6.11s call     tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoLaplaceApproximation]
5.94s call     tests/infer/test_inference.py::NormalNormalTests::test_rws_reparameterized
5.58s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoLaplaceApproximation]
5.57s call     tests/contrib/gp/test_models.py::test_mean_function_SGPR
5.55s call     tests/infer/mcmc/test_nuts.py::test_logistic_regression[True-JIT=False]
5.53s call     tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoLaplaceApproximation]
5.48s call     tests/contrib/gp/test_models.py::test_inference[SparseGPRegression_y1D]
5.34s call     tests/test_generic.py::test_mcmc_interface[pyro-beta_binomial]
5.28s call     tests/contrib/gp/test_models.py::test_inference[SparseGPRegression_y2D]
5.21s call     tests/infer/mcmc/test_nuts.py::test_gamma_beta[JIT=True]
5.20s call     tests/contrib/gp/test_models.py::test_inference_sgpr
5.19s call     tests/infer/mcmc/test_nuts.py::test_bernoulli_latent_model[JIT=True]
5.09s call     tests/contrib/gp/test_models.py::test_mean_function_SGPR_DTC
5.04s call     tests/integration_tests/test_tracegraph_elbo.py::NormalNormalTests::test_elbo_reparameterized
5.01s call     tests/contrib/gp/test_models.py::test_mean_function_SGPR_FITC
4.95s call     tests/distributions/test_categorical.py::TestCategorical::test_mean_and_var
4.92s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_reparameterized_N_is_3
4.89s call     tests/distributions/test_transforms.py::TransformTests::test_generalized_channel_permute
4.75s call     tests/infer/test_inference.py::NormalNormalTests::test_mmd_vectorized
4.68s call     tests/infer/mcmc/test_hmc.py::test_beta_bernoulli[JIT=False]
4.62s call     tests/distributions/test_transforms.py::TransformTests::test_conditional_generalized_channel_permute
4.38s call     tests/infer/test_abstract_infer.py::test_information_criterion
4.36s call     tests/contrib/gp/test_models.py::test_mean_function_GPR
4.30s call     tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=True-HMC]
4.26s call     tests/infer/test_jit.py::test_svi[JitTraceMeanField_ELBO-10]
4.25s call     tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=False-HMC]
4.24s call     tests/infer/test_sampling.py::ImportanceTest::test_importance_guide
4.20s call     tests/infer/test_inference.py::PoissonGammaTests::test_mmd_vectorized
4.17s call     tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[None-True-True-False]
4.07s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_two_layers_model_permuted
4.06s call     tests/infer/mcmc/test_hmc.py::test_bernoulli_latent_model[JIT=False]
4.06s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-sequential-sequential]
4.05s call     tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=True-NUTS]
4.04s call     tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_diag_normal_guide[False]
4.03s call     tests/infer/mcmc/test_nuts.py::test_logistic_regression[True-JIT=True]
4.03s call     tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[0.5-False-True-False]
4.03s call     tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=False-NUTS]
3.98s call     tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_diag_normal_guide[True]
3.72s call     tests/infer/mcmc/test_nuts.py::test_logistic_regression[False-JIT=False]
3.72s call     tests/contrib/oed/test_linear_models_eig.py::test_dv_linear_model
3.63s call     tests/infer/test_predictive.py::test_posterior_predictive_svi_manual_guide[True]
3.60s call     tests/infer/mcmc/test_hmc.py::test_dirichlet_categorical[JIT=False]
3.54s call     tests/infer/mcmc/test_hmc.py::test_logistic_regression[None-1-None-True-True-False]
3.52s call     tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[None-True-True-True]
3.47s call     tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-100]
3.45s call     tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-100]
3.34s call     tests/contrib/oed/test_linear_models_eig.py::test_lfire_linear_model
3.26s call     tests/contrib/tracking/test_em.py::test_em_nested_in_svi[True]
3.21s call     tests/contrib/gp/test_models.py::test_inference[GPRegression_y2D]
3.15s call     tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_delta_guide[True]
3.10s call     tests/infer/mcmc/test_mcmc_api.py::test_sequential_consistent
3.09s call     tests/test_generic.py::test_mcmc_interface[pyro-eight_schools]
3.08s call     tests/infer/mcmc/test_hmc.py::test_logistic_regression[None-1-None-True-True-True]
3.07s call     tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[None-True-False-False]
3.06s call     tests/contrib/gp/test_models.py::test_inference[GPRegression_y1D]
2.98s call     tests/infer/mcmc/test_hmc.py::test_beta_bernoulli[JIT=True]
2.92s call     tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[0.1-False-False-False]
2.91s call     tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_two_layers
2.85s call     tests/infer/test_svgd.py::test_conjugate[RBFSteinKernel-univariate]
2.83s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-True-1800]
2.81s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-True-2200]
2.81s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-False-1800]
2.80s call     tests/test_examples.py::test_cpu[rsa/generics.py --num-samples=10]
2.80s call     tests/contrib/oed/test_linear_models_eig.py::test_vnmc_linear_model
2.78s call     tests/test_examples.py::test_cpu[einsum.py]
2.75s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-False-None]
2.75s call     tests/infer/mcmc/test_nuts.py::test_logistic_regression[False-JIT=True]
2.74s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-False-2200]
2.70s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-True-None]
2.67s call     tests/infer/mcmc/test_valid_models.py::test_reparam_stable[HMC-kwargs0]
2.67s call     tests/infer/test_svgd.py::test_conjugate[RBFSteinKernel-multivariate]
2.64s call     tests/infer/test_svgd.py::test_conjugate[IMQSteinKernel-multivariate]
2.60s call     tests/infer/mcmc/test_hmc.py::test_logistic_regression[0.0855-None-4-False-True-False]
2.60s call     tests/infer/mcmc/test_hmc.py::test_logistic_regression[0.0855-None-4-False-False-False]
2.59s call     tests/infer/mcmc/test_mcmc_util.py::test_predictive[False-200]
2.59s call     tests/infer/mcmc/test_mcmc_util.py::test_predictive[False-100]
2.59s call     tests/infer/mcmc/test_mcmc_util.py::test_predictive[False-None]
2.57s call     tests/infer/test_svgd.py::test_conjugate[IMQSteinKernel-univariate]
2.53s call     tests/infer/mcmc/test_mcmc_util.py::test_predictive[True-100]
2.53s call     tests/infer/mcmc/test_mcmc_util.py::test_predictive[True-None]
2.53s call     tests/infer/mcmc/test_mcmc_util.py::test_predictive[True-200]
2.29s call     tests/infer/test_jit.py::test_beta_bernoulli[TraceEnum_ELBO-False]
2.28s call     tests/infer/mcmc/test_hmc.py::test_dirichlet_categorical[JIT=True]
2.21s call     tests/infer/mcmc/test_hmc.py::test_gamma_normal
2.20s call     tests/test_examples.py::test_cpu[sir_hmc.py -t=2 -w=2 -n=4 -d=100 -p=10000 -f 2]
2.18s call     tests/contrib/oed/test_linear_models_eig.py::test_marginal_likelihood_linear_model
2.15s call     tests/infer/test_jit.py::test_dirichlet_bernoulli[TraceEnum_ELBO-False]
2.07s call     tests/infer/test_enum.py::test_elbo_dbn_growth
2.07s call     tests/infer/mcmc/test_nuts.py::test_gaussian_hmm[3]
1.96s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-None-sequential]
1.94s call     tests/infer/mcmc/test_nuts.py::test_dirichlet_categorical_grad_adapt
1.93s call     tests/test_examples.py::test_cpu[minipyro.py --backend=pyro]
1.93s call     tests/distributions/test_von_mises.py::test_sample[0.01--1.5707963267948966]
1.92s call     tests/distributions/test_von_mises.py::test_sample[0.01-1.5707963267948966]
1.92s call     tests/distributions/test_von_mises.py::test_sample[0.01-0.0]
1.91s call     tests/contrib/oed/test_finite_spaces_eig.py::test_lfire_finite_space_model
1.83s call     tests/test_examples.py::test_cpu[rsa/hyperbole.py --price=10000]
1.79s call     tests/infer/mcmc/test_nuts.py::test_gamma_normal[False-JIT=False]
1.72s call     tests/infer/mcmc/test_nuts.py::test_dirichlet_categorical[JIT=False]
1.71s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-sequential-sequential]
1.65s call     tests/infer/mcmc/test_nuts.py::test_gamma_normal[True-JIT=False]
1.59s call     tests/infer/test_predictive.py::test_deterministic[event_shape1-True]
1.56s call     tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -c=2]
1.56s call     tests/contrib/oed/test_linear_models_eig.py::test_marginal_linear_model
1.55s call     tests/test_examples.py::test_cpu[minipyro.py]
1.52s call     tests/infer/mcmc/test_hmc.py::test_bernoulli_latent_model[JIT=True]
1.52s call     tests/infer/test_predictive.py::test_deterministic[event_shape0-True]
1.52s call     tests/distributions/test_von_mises.py::test_sample[30.0-1.5707963267948966]
1.51s call     tests/distributions/test_von_mises.py::test_sample[30.0-0.0]
1.51s call     tests/distributions/test_von_mises.py::test_sample[30.0--1.5707963267948966]
1.50s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-parallel-sequential]
1.50s call     tests/distributions/test_von_mises.py::test_sample[10.0--1.5707963267948966]
1.49s call     tests/distributions/test_von_mises.py::test_sample[10.0-0.0]
1.49s call     tests/distributions/test_von_mises.py::test_sample[10.0-1.5707963267948966]
1.48s call     tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 -nb=4]
1.47s call     tests/infer/mcmc/test_nuts.py::test_gaussian_hmm[2]
1.47s call     tests/contrib/oed/test_finite_spaces_eig.py::test_vnmc_finite_space_model
1.43s call     tests/test_examples.py::test_jit[minipyro.py --backend=pyro --jit]
1.42s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-2-3]
1.42s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-2-3]
1.40s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-0-7]
1.38s call     tests/infer/test_jit.py::test_svi[TraceEnum_ELBO-10]
1.34s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-7-7]
1.32s call     tests/test_examples.py::test_cpu[rsa/semantic_parsing.py --num-samples=10]
1.32s call     tests/infer/reparam/test_studentt.py::test_moments[(2, 3)]
1.31s call     tests/infer/test_predictive.py::test_deterministic[event_shape1-False]
1.29s call     tests/contrib/oed/test_linear_models_eig.py::test_posterior_linear_model
1.27s call     tests/infer/reparam/test_stable.py::test_symmetric_stable[(2, 3)]
1.26s call     tests/infer/mcmc/test_nuts.py::test_gamma_normal[False-JIT=True]
1.26s call     tests/infer/mcmc/test_nuts.py::test_dirichlet_categorical[JIT=True]
1.25s call     tests/infer/test_valid_models.py::test_enum_recycling_mutual_recursion
1.25s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-1-3]
1.24s call     tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -nb=4]
1.24s call     tests/test_examples.py::test_jit[minipyro.py --jit]
1.24s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-None-sequential]
1.23s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-1-3]
1.23s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-2-3]
1.23s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-2-3]
1.23s call     tests/infer/test_predictive.py::test_deterministic[event_shape0-False]
1.23s call     tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist0]
1.22s call     tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist3]
1.22s call     tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist0]
1.22s call     tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist0]
1.22s call     tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist3]
1.21s call     tests/optim/test_multi.py::test_optimizers[<lambda>2]
1.21s call     tests/contrib/tracking/test_em.py::test_svi_multi
1.19s call     tests/contrib/tracking/test_em.py::test_em_nested_in_svi[False]
1.19s call     tests/contrib/gp/test_models.py::test_hmc[VariationalGP_y2D]
1.18s call     tests/infer/test_jit.py::test_beta_bernoulli[JitTraceEnum_ELBO-False]
1.17s call     tests/infer/mcmc/test_nuts.py::test_gamma_normal[True-JIT=True]
1.16s call     tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-30]
1.16s call     tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist0]
1.15s call     tests/contrib/oed/test_ewma.py::test_ewma[0.5]
1.15s call     tests/contrib/oed/test_ewma.py::test_ewma[0.99]
1.15s call     tests/contrib/oed/test_ewma.py::test_ewma[0.9]
1.14s call     tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist1]
1.14s call     tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 4]
1.13s call     tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-30]
1.13s call     tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist3]
1.13s call     tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[sequential-True-None]
1.12s call     tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[sequential-False-None]
1.12s call     tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist3]
1.12s call     tests/infer/test_valid_models.py::test_enum_recycling_dbn[True-False]
1.11s call     tests/infer/test_jit.py::test_dirichlet_bernoulli[JitTraceEnum_ELBO-False]
1.11s call     tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist1]
1.11s call     tests/infer/test_valid_models.py::test_enum_recycling_dbn[True-True]
1.11s call     tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist2]
1.10s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-None-sequential]
1.09s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-1-3]
1.09s call     tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_rsample[True-relu]
1.09s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-1-3]
1.09s call     tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 1 --tmc --tmc-num-samples=2]
1.08s call     tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 5 --tmc --tmc-num-samples=2]
1.07s call     tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist2]
1.07s call     tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 2]
1.07s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-7-3]
1.07s call     tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist1]
1.06s call     tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 2 --tmc --tmc-num-samples=2]
1.06s call     tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -o=0.2]
1.05s call     tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 3 --tmc --tmc-num-samples=2]
1.05s call     tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 4 --tmc --tmc-num-samples=2]
1.04s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-7-7]
1.04s call     tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 -hfm=3]
1.04s call     tests/infer/test_enum.py::test_elbo_hmm_growth
1.04s call     tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist1]
1.03s call     tests/distributions/test_relaxed_straight_through.py::test_bernoulli_entropy_grad[1.0]
1.03s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-0-7]
1.03s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-0-3]
1.03s call     tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_rsample[False-relu]
1.02s call     tests/infer/test_jit.py::test_svi[TraceMeanField_ELBO-10]
1.02s call     tests/distributions/test_relaxed_straight_through.py::test_bernoulli_entropy_grad[0.5]
1.02s call     tests/infer/test_jit.py::test_svi[Trace_ELBO-10]
1.01s call     tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 5]
1.01s call     tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 3]
1.01s call     tests/distributions/test_conjugate.py::test_gamma_poisson_log_prob[shape2]
1.01s call     tests/optim/test_multi.py::test_optimizers[<lambda>3]
1.01s call     tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist2]
1.01s call     tests/test_examples.py::test_jit[contrib/epidemiology/regional.py --jit -ss=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --svi]
1.00s call     tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist2]
1.00s call     tests/contrib/gp/test_models.py::test_hmc[SparseGPRegression_y2D]
1.00s call     tests/contrib/gp/test_models.py::test_hmc[VariationalSparseGP_y2D]
1.00s call     tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 1]
0.99s call     tests/distributions/test_von_mises.py::test_sample[3.0--1.5707963267948966]
0.99s call     tests/distributions/test_rejector.py::test_exponential_elbo[0.25-2.0]
0.99s call     tests/distributions/test_von_mises.py::test_sample[3.0-1.5707963267948966]
0.99s call     tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --haar]
0.99s call     tests/contrib/oed/test_finite_spaces_eig.py::test_dv_finite_space_model
0.99s call     tests/distributions/test_von_mises.py::test_sample[3.0-0.0]
0.98s call     tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -e=2]
0.98s call     tests/distributions/test_rejector.py::test_exponential_elbo[0.25-1.0]
0.96s call     tests/infer/test_valid_models.py::test_enum_recycling_reentrant_history[2]
0.96s call     tests/optim/test_multi.py::test_optimizers[<lambda>4]
0.95s call     tests/infer/test_valid_models.py::test_enum_recycling_reentrant
0.95s call     tests/contrib/gp/test_models.py::test_hmc[VariationalSparseGP_y1D]
0.95s call     tests/contrib/oed/test_finite_spaces_eig.py::test_marginal_likelihood_finite_space_model
0.94s call     tests/infer/test_valid_models.py::test_enum_recycling_reentrant_history[1]
0.94s call     tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2]
0.94s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-7-7]
0.94s call     tests/test_examples.py::test_jit[contrib/epidemiology/regional.py --jit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2]
0.93s call     tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -e=2 -k=1]
0.93s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-sequential-sequential-None]
0.92s call     tests/test_examples.py::test_jit[contrib/cevae/synthetic.py --num-epochs=1 --jit]
0.92s call     tests/test_examples.py::test_jit[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -p=10000 --sequential --jit]
0.92s call     tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -ss=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --svi]
0.92s call     tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-None-10-None-10-1]
0.92s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-sequential-sequential]
0.92s call     tests/distributions/test_rejector.py::test_exponential_elbo[0.25-0.5]
0.91s call     tests/distributions/test_relaxed_straight_through.py::test_onehot_entropy_grad[0.3]
0.91s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-2-2]
0.91s call     tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -hfm=3]
0.90s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-2-2]
0.90s call     tests/test_examples.py::test_cpu[inclined_plane.py --num-samples=1]
0.90s call     tests/distributions/test_relaxed_straight_through.py::test_onehot_entropy_grad[1.0]
0.90s call     tests/distributions/test_relaxed_straight_through.py::test_onehot_entropy_grad[0.5]
0.90s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-0-7]
0.89s call     tests/infer/reparam/test_studentt.py::test_moments[(4,)]
0.89s call     tests/infer/test_csis.py::test_csis_sampling
0.89s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-7-3]
0.88s call     tests/test_examples.py::test_cpu[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -p=10000 --sequential]
0.88s call     tests/infer/test_jit.py::test_beta_bernoulli[TraceEnum_ELBO-True]
0.88s call     tests/infer/test_inference.py::test_reparam_stable
0.88s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-7-7]
0.88s call     tests/infer/test_autoguide.py::test_predictive[AutoNormal]
0.88s call     tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -k=1]
0.88s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.87s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-7-3]
0.87s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-0-7]
0.86s call     tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 --haar]
0.86s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-7-7]
0.86s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-0-3]
0.86s call     tests/contrib/gp/test_models.py::test_hmc[VariationalGP_y1D]
0.86s call     tests/test_examples.py::test_jit[contrib/epidemiology/sir.py --jit -np=128 -ss=2 -n=4 -d=20 -p=1000 -f 2 --svi]
0.85s call     tests/infer/test_jit.py::test_svi[TraceGraph_ELBO-10]
0.85s call     tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -ss=2 -n=4 -d=20 -p=1000 -f 2 --svi]
0.85s call     tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 4}-7-12]
0.84s call     tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -a]
0.84s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.83s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-0-7]
0.83s call     tests/test_examples.py::test_cpu[contrib/cevae/synthetic.py --num-epochs=1]
0.83s call     tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2]
0.83s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-0-3]
0.83s call     tests/test_examples.py::test_jit[contrib/epidemiology/sir.py --jit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2]
0.82s call     tests/distributions/test_von_mises.py::test_sample[0.03-0.0]
0.82s call     tests/distributions/test_von_mises.py::test_sample[0.03--1.5707963267948966]
0.82s call     tests/distributions/test_von_mises.py::test_sample[0.03-1.5707963267948966]
0.81s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.81s call     tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape0-0.5-0.6-0.4]
0.80s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-7]
0.80s call     tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.80s call     tests/infer/reparam/test_stable.py::test_stable[StableReparam-(2, 3)]
0.80s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-sequential-sequential]
0.80s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-1-2]
0.80s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-1-2]
0.79s call     tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape0-0.5-0.6-1.1]
0.78s call     tests/test_examples.py::test_jit[sir_hmc.py -t=2 -w=2 -n=4 -p=10000 --jit]
0.78s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-7]
0.78s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.78s call     tests/optim/test_optim.py::test_dctadam_param_subsample
0.78s call     tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoMultivariateNormal]
0.77s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-2-2]
0.77s call     tests/infer/test_valid_models.py::test_dim_allocation_ok[TraceEnum_ELBO-False]
0.77s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-2-2]
0.77s call     tests/infer/test_valid_models.py::test_dim_allocation_ok[TraceEnum_ELBO-True]
0.77s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.77s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-sequential-None-None]
0.76s call     tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoNormal]
0.76s call     tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar_full_mass': 3, 'num_quant_bins': 2}-7-12]
0.76s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-centered3-(3, 2)]
0.76s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoNormal]
0.75s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-1.0-(3, 2)]
0.75s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-None-(3, 2)]
0.75s call     tests/infer/reparam/test_stable.py::test_symmetric_stable[(4,)]
0.75s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-4000]
0.75s call     tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[2-False]
0.75s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-4000]
0.74s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-4000]
0.74s call     tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar': True, 'num_quant_bins': 2}-7-12]
0.74s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-4000]
0.74s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-4000]
0.74s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-4000]
0.74s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-4000]
0.74s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-4000]
0.74s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-4000]
0.74s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-3]
0.74s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.6-(3, 2)]
0.74s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.0-(3, 2)]
0.74s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-centered3-(3, 2)]
0.74s call     tests/infer/test_jit.py::test_dirichlet_bernoulli[TraceEnum_ELBO-True]
0.73s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-4000]
0.73s call     tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-14-1-7-1-1]
0.73s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-4000]
0.73s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-4000]
0.73s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-4000]
0.73s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-4000]
0.73s call     tests/infer/test_autoguide.py::test_predictive[AutoLowRankMultivariateNormal]
0.73s call     tests/test_examples.py::test_jit[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -m=1 --enum --jit]
0.72s call     tests/test_examples.py::test_cpu[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -m=1 --enum]
0.72s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.72s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-3]
0.72s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoMultivariateNormal]
0.72s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-None-sequential-None]
0.72s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-sequential-sequential-None]
0.72s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoMultivariateNormal]
0.72s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_diagnostics[2]
0.71s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-0-3]
0.71s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-2-3]
0.71s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-2-3]
0.71s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-None-(3, 2)]
0.70s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.70s call     tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 2}-7-12]
0.70s call     tests/contrib/oed/test_finite_spaces_eig.py::test_posterior_finite_space_model
0.69s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-1-2]
0.69s call     tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[2-True]
0.69s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.6-(3, 2)]
0.69s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoLowRankMultivariateNormal]
0.69s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-1.0-(3, 2)]
0.69s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.0-(3, 2)]
0.69s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-1-2]
0.69s call     tests/distributions/test_conjugate.py::test_variance[dist3]
0.68s call     tests/distributions/test_conjugate.py::test_mean[dist3]
0.68s call     tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.67s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.67s call     tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-2-3]
0.67s call     tests/infer/mcmc/test_valid_models.py::test_reparam_stable[NUTS-kwargs1]
0.67s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-7-7]
0.66s call     tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-2-3]
0.66s call     tests/distributions/test_stable.py::test_sample[0.1--0.5]
0.66s call     tests/distributions/test_stable.py::test_sample[0.1-0.0]
0.66s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-None-sequential-None]
0.66s call     tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'num_quant_bins': 2}-7-8-6]
0.66s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.65s call     tests/test_examples.py::test_cpu[sparse_regression.py --num-steps=2 --num-data=50 --num-dimensions 20]
0.65s call     tests/contrib/oed/test_finite_spaces_eig.py::test_marginal_finite_space_model
0.65s call     tests/distributions/test_stable.py::test_sample[0.1-0.5]
0.64s call     tests/distributions/test_von_mises.py::test_sample[1.0-0.0]
0.64s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_guide_module_callable]
0.64s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.64s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_guide_callable]
0.64s call     tests/distributions/test_von_mises.py::test_sample[1.0-1.5707963267948966]
0.64s call     tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoLowRankMultivariateNormal]
0.64s call     tests/distributions/test_transforms.py::TransformTests::test_conditional_affine_coupling
0.64s call     tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'num_quant_bins': 2}-0-8-6]
0.64s call     tests/optim/test_multi.py::test_optimizers[<lambda>1]
0.64s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-0-7]
0.63s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-7-7]
0.63s call     tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoMultivariateNormal]
0.63s call     tests/distributions/test_von_mises.py::test_sample[1.0--1.5707963267948966]
0.63s call     tests/test_examples.py::test_jit[lda.py --num-steps=2 --num-words=100 --num-docs=100 --num-words-per-doc=8 --jit]
0.63s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-parallel-sequential-None]
0.62s call     tests/optim/test_multi.py::test_optimizers[<lambda>0]
0.62s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-7-3]
0.62s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.62s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-0-3]
0.62s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-7-3]
0.62s call     tests/test_examples.py::test_cpu[contrib/timeseries/gp_models.py -m lcmgp --test --num-steps=2]
0.62s call     tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoMultivariateNormal]
0.62s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.62s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-sequential-parallel-None]
0.61s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-sequential-sequential-None]
0.61s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoMultivariateNormal]
0.61s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-7-7]
0.61s call     tests/test_examples.py::test_cpu[contrib/timeseries/gp_models.py -m imgp --test --num-steps=2]
0.61s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-0-7]
0.61s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.61s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.61s call     tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape0-0.5-0.6-0.4]
0.60s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-7-3]
0.60s call     tests/distributions/test_transforms.py::TransformTests::test_affine_coupling
0.60s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_guide_list_x]
0.60s call     tests/test_examples.py::test_cpu[rsa/schelling_false.py --num-samples=10]
0.60s call     tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape0-0.5-0.6-1.1]
0.59s call     tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoNormal]
0.59s call     tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoDiagonalNormal]
0.59s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoNormal]
0.59s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.59s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.59s call     tests/distributions/test_transforms.py::TransformTests::test_block_autoregressive_jacobians
0.59s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoDiagonalNormal]
0.59s call     tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op0-JIT=True]
0.59s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class11]
0.59s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class12]
0.59s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class9]
0.59s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-0-3]
0.59s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class10]
0.59s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.59s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-sequential-None]
0.59s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=5 --tmc --tmc-num-samples=2]
0.58s call     tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op1-JIT=True]
0.58s call     tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoNormal]
0.58s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-sequential-parallel]
0.58s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-2-3]
0.58s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoNormal]
0.58s call     tests/contrib/gp/test_models.py::test_hmc[SparseGPRegression_y1D]
0.58s call     tests/infer/test_autoguide.py::test_predictive[AutoMultivariateNormal]
0.58s call     tests/contrib/gp/test_models.py::test_hmc[GPRegression_y2D]
0.58s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-0-7]
0.58s call     tests/test_examples.py::test_jit[contrib/autoname/mixture.py --num-epochs=1 --jit]
0.58s call     tests/contrib/epidemiology/test_models.py::test_overdispersed_sir_smoke[{'num_quant_bins': 2}-7-3]
0.58s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-2-3]
0.57s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.57s call     tests/test_examples.py::test_cpu[contrib/forecast/bart.py --num-steps=2 --stride=99999]
0.57s call     tests/infer/test_autoguide.py::test_predictive[auto_class7]
0.57s call     tests/distributions/test_von_mises.py::test_sample[0.3--1.5707963267948966]
0.57s call     tests/infer/test_autoguide.py::test_predictive[auto_class6]
0.57s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.57s call     tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=20]
0.57s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=3]
0.57s call     tests/test_examples.py::test_cpu[lda.py --num-steps=2 --num-words=100 --num-docs=100 --num-words-per-doc=8]
0.57s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=1 --tmc --tmc-num-samples=2]
0.57s call     tests/distributions/test_von_mises.py::test_sample[0.3-0.0]
0.57s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=4]
0.57s call     tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=2 --tmc --tmc-num-samples=2 --jit]
0.56s call     tests/distributions/test_von_mises.py::test_sample[100.0-0.0]
0.56s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-sequential-None-None]
0.56s call     tests/infer/test_autoguide.py::test_predictive[AutoDiagonalNormal]
0.56s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.55s call     tests/test_examples.py::test_cpu[contrib/autoname/tree_data.py --num-epochs=1]
0.55s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-True-1800]
0.55s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-False-1800]
0.55s call     tests/contrib/cevae/test_cevae.py::test_serialization[jit-studentt-2]
0.55s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-True-2200]
0.55s call     tests/distributions/test_von_mises.py::test_sample[0.3-1.5707963267948966]
0.55s call     tests/distributions/test_von_mises.py::test_sample[100.0--1.5707963267948966]
0.55s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-True-None]
0.55s call     tests/distributions/test_von_mises.py::test_sample[100.0-1.5707963267948966]
0.55s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-False-2200]
0.55s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-False-None]
0.55s call     tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[iplate-8-100-0.018]
0.55s call     tests/test_examples.py::test_jit[eight_schools/svi.py --num-epochs=1 --jit]
0.55s call     tests/contrib/cevae/test_cevae.py::test_serialization[jit-studentt-1]
0.54s call     tests/test_examples.py::test_cpu[contrib/autoname/mixture.py --num-epochs=1]
0.54s call     tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[iplate-None-100-0.013]
0.54s call     tests/test_examples.py::test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide auto]
0.54s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-7]
0.54s call     tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoLaplaceApproximation]
0.54s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-7-3]
0.54s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-1-3]
0.54s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-()-()]
0.54s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-1-3]
0.54s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoLaplaceApproximation]
0.54s call     tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=21]
0.54s call     tests/test_examples.py::test_cpu[rsa/schelling.py --num-samples=10]
0.54s call     tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --num-iafs=1]
0.53s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6 --tmc --tmc-num-samples=2]
0.53s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=2 --tmc --tmc-num-samples=2]
0.53s call     tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=3 --tmc --tmc-num-samples=2 --jit]
0.53s call     tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=3 --jit]
0.53s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=3 --tmc --tmc-num-samples=2]
0.53s call     tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=1 --jit]
0.53s call     tests/test_examples.py::test_cpu[contrib/autoname/scoping_mixture.py --num-epochs=1]
0.53s call     tests/test_examples.py::test_jit[dmm/dmm.py --num-epochs=1 --jit]
0.53s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6]
0.53s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=7]
0.53s call     tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=1 --tmc --tmc-num-samples=2 --jit]
0.53s call     tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=2 --jit]
0.53s call     tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=4 --jit]
0.53s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=1]
0.53s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=5]
0.53s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-3]
0.53s call     tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1]
0.53s call     tests/test_examples.py::test_cpu[mixed_hmm/experiment.py --timesteps=1]
0.53s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=2]
0.52s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-100.0]
0.52s call     tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=7 --jit]
0.52s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-100.0]
0.52s call     tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=4 --tmc --tmc-num-samples=2 --jit]
0.52s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-7-7]
0.52s call     tests/test_examples.py::test_cpu[smcfilter.py --num-timesteps=3 --num-particles=10]
0.52s call     tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --tmc --tmc-num-samples=2]
0.52s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6 --raftery-parameterization]
0.52s call     tests/test_examples.py::test_cpu[eight_schools/svi.py --num-epochs=1]
0.52s call     tests/test_examples.py::test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide easy]
0.52s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=0]
0.52s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=4 --tmc --tmc-num-samples=2]
0.52s call     tests/test_examples.py::test_jit[dmm/dmm.py --num-epochs=1 --num-iafs=1 --jit]
0.52s call     tests/contrib/cevae/test_cevae.py::test_serialization[jit-normal-1]
0.52s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-1.0-(4,)]
0.52s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-centered3-(4,)]
0.52s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-100.0]
0.52s call     tests/contrib/cevae/test_cevae.py::test_serialization[jit-laplace-2]
0.51s call     tests/infer/test_jit.py::test_svi[JitTraceMeanField_ELBO-1]
0.51s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-0-3]
0.51s call     tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=5 --jit]
0.51s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-sequential-None-None]
0.51s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-None-sequential-None]
0.51s call     tests/distributions/test_conjugate.py::test_gamma_poisson_log_prob[shape1]
0.51s call     tests/contrib/cevae/test_cevae.py::test_serialization[jit-laplace-1]
0.51s call     tests/contrib/cevae/test_cevae.py::test_serialization[jit-normal-2]
0.51s call     tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=0 --tmc --tmc-num-samples=2]
0.51s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-0-7]
0.51s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-None-(4,)]
0.51s call     tests/test_examples.py::test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide custom]
0.51s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-7-7]
0.51s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_guide_callable]
0.51s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_guide_module_callable]
0.51s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-centered3-(4,)]
0.51s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-7]
0.51s call     tests/infer/reparam/test_stable.py::test_stable[LatentStableReparam-(2, 3)]
0.51s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-parallel-None-None]
0.51s call     tests/distributions/test_mvt.py::test_log_prob_normalization[2]
0.50s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.0-(4,)]
0.50s call     tests/test_examples.py::test_jit[lkj.py --n=50 --num-chains=1 --warmup-steps=100 --num-samples=200 --jit]
0.50s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.6-(4,)]
0.50s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_guide_callable]
0.50s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_guide_module_callable]
0.50s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-7-7]
0.50s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-parallel-sequential]
0.50s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-3-3]
0.50s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-3]
0.50s call     tests/distributions/test_von_mises.py::test_sample[0.1-0.0]
0.49s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-2-2]
0.49s call     tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[16]
0.49s call     tests/infer/reparam/test_stable.py::test_stable[StableReparam-(4,)]
0.49s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.49s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-()-()]
0.49s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-0-7]
0.49s call     tests/distributions/test_von_mises.py::test_sample[0.1--1.5707963267948966]
0.49s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-2-2]
0.49s call     tests/distributions/test_von_mises.py::test_sample[0.1-1.5707963267948966]
0.49s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-None-sequential-None]
0.49s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-parallel-sequential-None]
0.49s call     tests/test_examples.py::test_jit[air/main.py --num-steps=1 --jit]
0.48s call     tests/test_examples.py::test_jit[vae/vae.py --num-epochs=1 --jit]
0.48s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-None-parallel-None]
0.48s call     tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[range-None-100-0.011]
0.48s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-None-(4,)]
0.48s call     tests/contrib/cevae/test_cevae.py::test_serialization[jit-exponential-2]
0.48s call     tests/nn/test_autoregressive.py::AutoRegressiveNNTests::test_jacobians
0.48s call     tests/contrib/cevae/test_cevae.py::test_serialization[jit-exponential-1]
0.48s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-sequential-sequential-None]
0.48s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-7-7]
0.47s call     tests/contrib/cevae/test_cevae.py::test_serialization[jit-bernoulli-2]
0.47s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.47s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-3-2-(3, 2)]
0.47s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_guide_list_x]
0.47s call     tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[range-8-100-0.01]
0.47s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-1.0-(4,)]
0.47s call     tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[range-3-100-0.018]
0.47s call     tests/contrib/cevae/test_cevae.py::test_serialization[jit-bernoulli-1]
0.47s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.6-(4,)]
0.47s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoDiagonalNormal]
0.47s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.0-(4,)]
0.47s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-sequential-parallel-None]
0.47s call     tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoDelta]
0.47s call     tests/infer/test_autoguide.py::test_predictive[AutoLaplaceApproximation]
0.47s call     tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-None-1-None-1-8]
0.47s call     tests/distributions/test_stable.py::test_sample[0.99-0.0]
0.47s call     tests/contrib/oed/test_linear_models_eig.py::test_laplace_linear_model
0.46s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.46s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-7-3]
0.46s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-0-7]
0.46s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-0-7]
0.46s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class10]
0.46s call     tests/distributions/test_stable.py::test_sample_2[2.0-0.5]
0.46s call     tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoDiagonalNormal]
0.46s call     tests/distributions/test_stable.py::test_sample_2[2.0--0.5]
0.46s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_guide_list_x]
0.46s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class12]
0.46s call     tests/distributions/test_stable.py::test_sample_2[2.0-1.0]
0.46s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class11]
0.46s call     tests/distributions/test_stable.py::test_sample_2[2.0--1.0]
0.46s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class9]
0.46s call     tests/test_examples.py::test_cpu[air/main.py --num-steps=1]
0.46s call     tests/distributions/test_stable.py::test_sample[1.01-0.0]
0.46s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-sequential-None-None]
0.46s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class12]
0.46s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.45s call     tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoDiagonalNormal]
0.45s call     tests/test_examples.py::test_cpu[contrib/gp/sv-dkl.py --epochs=1 --num-inducing=4 --batch-size=1000]
0.45s call     tests/test_examples.py::test_cpu[contrib/gp/sv-dkl.py --binary --epochs=1 --num-inducing=4 --batch-size=1000]
0.45s call     tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential --jit]
0.45s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class11]
0.45s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class9]
0.45s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoDiagonalNormal]
0.45s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class10]
0.45s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-1-3]
0.45s call     tests/distributions/test_stable.py::test_sample[0.4--0.5]
0.45s call     tests/test_examples.py::test_jit[vae/vae_comparison.py --num-epochs=1 --jit]
0.45s call     tests/test_examples.py::test_jit[contrib/gp/sv-dkl.py --epochs=1 --num-inducing=4 --jit]
0.45s call     tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar': True, 'num_quant_bins': 2}-7-4]
0.45s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.45s call     tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --jit]
0.45s call     tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel --jit]
0.45s call     tests/test_examples.py::test_cpu[vae/vae.py --num-epochs=1]
0.45s call     tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1]
0.45s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-1-3]
0.45s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-7-3]
0.45s call     tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --aux-loss --jit]
0.45s call     tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar_full_mass': 3, 'num_quant_bins': 2}-7-4]
0.45s call     tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel]
0.44s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.44s call     tests/test_examples.py::test_cpu[air/main.py --num-steps=1 --no-baseline]
0.44s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-0-3]
0.44s call     tests/contrib/epidemiology/test_models.py::test_overdispersed_sir_smoke[{'haar_full_mass': 2}-7-3]
0.44s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-7]
0.44s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-7-3]
0.44s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-parallel-sequential-None]
0.44s call     tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential]
0.44s call     tests/distributions/test_stable.py::test_sample[0.4-0.5]
0.44s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.44s call     tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1 --aux-loss]
0.44s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-None-sequential-None]
0.43s call     tests/distributions/test_conjugate.py::test_variance[dist1]
0.43s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-7]
0.43s call     tests/test_examples.py::test_cpu[vae/vae_comparison.py --num-epochs=1]
0.43s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-7-3]
0.43s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-2-1]
0.43s call     tests/distributions/test_rejector.py::test_exponential_elbo[0.5-0.5]
0.43s call     tests/distributions/test_conjugate.py::test_mean[dist1]
0.43s call     tests/infer/test_jit.py::test_svi_enum[2-sequential-sequential]
0.43s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-sequential-sequential-None]
0.43s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-2-1]
0.43s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.43s call     tests/distributions/test_rejector.py::test_exponential_elbo[0.5-1.0]
0.42s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.42s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-0-3]
0.42s call     tests/distributions/test_stable.py::test_sample[0.4-0.0]
0.42s call     tests/distributions/test_rejector.py::test_exponential_elbo[0.5-2.0]
0.42s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-parallel-parallel-None]
0.42s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.42s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-7-7]
0.42s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoLaplaceApproximation]
0.42s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-sequential-None]
0.42s call     tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoLaplaceApproximation]
0.42s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-sequential-sequential-None]
0.41s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-None-None-None]
0.41s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-0-3]
0.41s call     tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoLaplaceApproximation]
0.41s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoLaplaceApproximation]
0.41s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-0-3]
0.41s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-7-7]
0.41s call     tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op0-JIT=False]
0.41s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-7-7]
0.41s call     tests/contrib/epidemiology/test_models.py::test_overdispersed_seir_smoke[{'num_quant_bins': 2}-7-3]
0.41s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-sequential-None-None]
0.40s call     tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op1-JIT=False]
0.40s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-()-()]
0.40s call     tests/contrib/epidemiology/test_models.py::test_overdispersed_sir_smoke[{}-7-3]
0.40s call     tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 2}-7-4]
0.40s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.40s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-()-()]
0.40s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-3-1]
0.40s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-7-7]
0.40s call     tests/ops/test_integrator.py::test_time_reversibility[QuarticOscillator]
0.40s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-()-()]
0.40s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-None-sequential]
0.40s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-2-2]
0.40s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-2-2]
0.40s call     tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-100]
0.39s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.39s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-3-7]
0.39s call     tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 4}-7-4]
0.39s call     tests/contrib/gp/test_parameterized.py::test_inference
0.39s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-7-3]
0.39s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-1-2]
0.39s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-3-7]
0.39s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-sequential-None]
0.39s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-1-2]
0.39s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-()-()]
0.38s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-0-7]
0.38s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-0-7]
0.38s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.38s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-1-7]
0.38s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-1-7]
0.38s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-1-7]
0.38s call     tests/infer/test_jit.py::test_svi[JitTraceEnum_ELBO-10]
0.38s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-None-sequential-None]
0.38s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-3-2-(3, 2)]
0.38s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-3-7]
0.38s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-3-7]
0.38s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-7-3]
0.37s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-0-7]
0.37s call     tests/infer/test_autoguide.py::test_predictive[AutoDelta]
0.37s call     tests/distributions/test_stable.py::test_sample[1.7-0.5]
0.37s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-None-sequential-None]
0.37s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-7-7]
0.37s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-3-2-(3, 2)]
0.37s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-1-1]
0.37s call     tests/distributions/test_stable.py::test_sample[1.7--0.5]
0.37s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-10.0]
0.37s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-1-1]
0.37s call     tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoDelta]
0.37s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-()-()]
0.36s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-1-7]
0.36s call     tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-1-200]
0.36s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-sequential-sequential]
0.36s call     tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[sequential-inf]
0.36s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-1-7]
0.36s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-0-7]
0.36s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.36s call     tests/contrib/cevae/test_cevae.py::test_serialization[python-studentt-2]
0.36s call     tests/contrib/cevae/test_cevae.py::test_serialization[python-studentt-1]
0.36s call     tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[sequential-1]
0.36s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-3-7]
0.36s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-parallel-None-None]
0.36s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-None-parallel-None]
0.36s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-0-3]
0.36s call     tests/distributions/test_stable.py::test_sample[1.7-0.0]
0.36s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-sequential-None]
0.36s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-0-3]
0.35s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-3-2-(4,)]
0.35s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-7-7]
0.35s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-7-7]
0.35s call     tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar_full_mass': 4}-7-8-6]
0.35s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-7-7]
0.35s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-7-3]
0.35s call     tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoDelta]
0.35s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-3-7]
0.35s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-sequential-parallel-None]
0.35s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.35s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-2-1]
0.35s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-2-1]
0.35s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-None-parallel]
0.35s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-0-7]
0.34s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-()-()]
0.34s call     tests/contrib/cevae/test_cevae.py::test_serialization[python-normal-2]
0.34s call     tests/contrib/cevae/test_cevae.py::test_serialization[python-normal-1]
0.34s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-7-3]
0.34s call     tests/contrib/cevae/test_cevae.py::test_serialization[python-laplace-2]
0.34s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-parallel-None-None]
0.34s call     tests/contrib/cevae/test_cevae.py::test_serialization[python-laplace-1]
0.34s call     tests/infer/test_jit.py::test_beta_bernoulli[JitTraceEnum_ELBO-True]
0.34s call     tests/distributions/test_mvt.py::test_log_prob_normalization[1]
0.34s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-7-7]
0.34s call     tests/contrib/gp/test_models.py::test_hmc[GPRegression_y1D]
0.34s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-True-0-1000-False-2]
0.34s call     tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar': True}-7-8-6]
0.34s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-0-3]
0.34s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-7-3]
0.34s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-3-7]
0.34s call     tests/ops/test_contract.py::test_ubersum[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.33s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-None-parallel-None]
0.33s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-sequential-None-None]
0.33s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-7-7]
0.33s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-7-3]
0.33s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-3-2-(4,)]
0.33s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.33s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-2-3]
0.33s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.33s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-None-sequential-None]
0.33s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-None-sequential-None]
0.33s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-0-7]
0.33s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-1-7]
0.33s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-0-7]
0.33s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-0-7]
0.33s call     tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar_full_mass': 4}-0-8-6]
0.33s call     tests/contrib/cevae/test_cevae.py::test_serialization[python-exponential-1]
0.33s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-True-0-1000-False-2]
0.33s call     tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-2-200]
0.33s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-True-0-1000-False-2]
0.33s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-7-3]
0.33s call     tests/test_generic.py::test_mcmc_interface[pyro-neals_funnel]
0.33s call     tests/contrib/cevae/test_cevae.py::test_serialization[python-exponential-2]
0.32s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-0-3]
0.32s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-4-3-MixtureOfDiagNormals]
0.32s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-0-3]
0.32s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-True-0-1000-False-2]
0.32s call     tests/contrib/cevae/test_cevae.py::test_serialization[python-bernoulli-1]
0.32s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-sequential-parallel-None]
0.32s call     tests/contrib/cevae/test_cevae.py::test_serialization[python-bernoulli-2]
0.32s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-3-1]
0.32s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-1-2]
0.32s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-1-2]
0.32s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-7-7]
0.32s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-7-3]
0.32s call     tests/ops/test_contract.py::test_ubersum_jit[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.32s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-parallel-parallel-None]
0.32s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-1-7]
0.32s call     tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar': True}-0-8-6]
0.31s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-0-7]
0.31s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-0-3]
0.31s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-3-7]
0.31s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-3-7]
0.31s call     tests/ops/test_contract.py::test_naive_ubersum[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.31s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-()-()]
0.31s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-None-sequential-None]
0.31s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-False-0-1000-False-2]
0.31s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-0-7]
0.31s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-7-3]
0.31s call     tests/distributions/test_rejector.py::test_rejector[0.25-1.0]
0.31s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-7-3]
0.31s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.31s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-None-parallel-None]
0.31s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-7-3]
0.31s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-1-7]
0.31s call     tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-0-3]
0.31s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-False-0-1000-False-2]
0.31s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-1-7]
0.31s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-3-7]
0.30s call     tests/distributions/test_rejector.py::test_rejector[0.25-2.0]
0.30s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-()-()]
0.30s call     tests/distributions/test_rejector.py::test_rejector[0.25-0.5]
0.30s call     tests/infer/reparam/test_stable.py::test_stable[LatentStableReparam-(4,)]
0.30s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-False-0-1000-False-2]
0.30s call     tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{}-7-8-6]
0.30s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-1-1]
0.30s call     tests/contrib/cevae/test_cevae.py::test_smoke[studentt-1-200]
0.30s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-1-1]
0.30s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-7-7]
0.30s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.30s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-sequential-parallel]
0.30s call     tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-10-1-None-3-5]
0.30s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-False-0-1000-False-2]
0.30s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-()-()]
0.30s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-0-7]
0.30s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-3]
0.29s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-3-1]
0.29s call     tests/distributions/test_stable.py::test_sample[0.1--1.0]
0.29s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-3-1]
0.29s call     tests/distributions/test_stable.py::test_sample[0.1-1.0]
0.29s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-1-7]
0.29s call     tests/infer/test_jit.py::test_infer_discrete[sample-10]
0.29s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-0-3]
0.29s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-0-3]
0.29s call     tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-0-3]
0.29s call     tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-0-3]
0.29s call     tests/infer/test_jit.py::test_dirichlet_bernoulli[JitTraceEnum_ELBO-True]
0.29s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-parallel-sequential]
0.29s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-None-parallel-None]
0.29s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-2-1]
0.28s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-1-1]
0.28s call     tests/infer/test_jit.py::test_svi[JitTrace_ELBO-10]
0.28s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-50-False]
0.28s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-2-1]
0.28s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-sequential-None-None]
0.28s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-3-2-(4,)]
0.28s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-parallel-None-None]
0.28s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-1-1]
0.28s call     tests/infer/reparam/test_unit_jacobian.py::test_normal[(4, 5)]
0.28s call     tests/infer/test_jit.py::test_svi[JitTraceGraph_ELBO-10]
0.28s call     tests/distributions/test_stable.py::test_sample[0.99-0.5]
0.28s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-3-7]
0.28s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-4]
0.28s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-0-7]
0.28s call     tests/distributions/test_mvt.py::test_log_prob[2-(3, 2)]
0.28s call     tests/contrib/cevae/test_cevae.py::test_smoke[normal-2-200]
0.28s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.28s call     tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{}-0-8-6]
0.28s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-7-7]
0.28s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-7-7]
0.27s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-sequential-None-None]
0.27s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-parallel-None-None]
0.27s call     tests/distributions/test_stable.py::test_sample[1.01--0.5]
0.27s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-sequential-parallel-None]
0.27s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-parallel-None]
0.27s call     tests/contrib/cevae/test_cevae.py::test_smoke[studentt-2-200]
0.27s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-parallel-sequential-None]
0.27s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-sequential-parallel-None]
0.27s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-4]
0.27s call     tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-7-1-7-1-7]
0.27s call     tests/distributions/test_stable.py::test_sample[0.99--0.5]
0.27s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-3]
0.27s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-None-None-None]
0.27s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-()-()]
0.27s call     tests/distributions/test_stable.py::test_sample[1.01-0.5]
0.27s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-None-parallel-None]
0.27s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-1-1]
0.27s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-None-sequential-None]
0.27s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-parallel-sequential]
0.27s call     tests/test_generic.py::test_mcmc_interface[pyro-logistic_regression]
0.27s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-3-7]
0.27s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-4-3-MixtureOfDiagNormals]
0.27s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-3-2-(3, 2)]
0.27s call     tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-None-5-10-1-5]
0.26s call     tests/contrib/cevae/test_cevae.py::test_smoke[exponential-2-200]
0.26s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-()-()]
0.26s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-sequential-None]
0.26s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.26s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-7-7]
0.26s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-7-7]
0.26s call     tests/contrib/cevae/test_cevae.py::test_smoke[laplace-2-200]
0.26s call     tests/ops/test_contract.py::test_naive_ubersum[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.26s call     tests/nn/test_autoregressive.py::AutoRegressiveNNTests::test_masks
0.26s call     tests/contrib/cevae/test_cevae.py::test_smoke[normal-1-200]
0.26s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-parallel-sequential-None]
0.26s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-7-7]
0.26s call     tests/ops/test_contract.py::test_ubersum[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.26s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.26s call     tests/contrib/cevae/test_cevae.py::test_smoke[laplace-1-200]
0.26s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-()-()]
0.26s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-None-sequential]
0.26s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-1-7]
0.26s call     tests/distributions/test_stable.py::test_sample[1.7--1.0]
0.26s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-7-7]
0.26s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-3-1]
0.26s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-0-7]
0.26s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-10.0]
0.26s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-10]
0.26s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-1-7]
0.26s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-7-7]
0.25s call     tests/infer/test_util.py::test_psis_diagnostic[5-0.95-krange1-500000]
0.25s call     tests/contrib/cevae/test_cevae.py::test_smoke[exponential-1-200]
0.25s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-7-3]
0.25s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-0-7]
0.25s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-None-None-None]
0.25s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.25]
0.25s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.0]
0.25s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-1-7]
0.25s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.5]
0.25s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.75]
0.25s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-1.0]
0.25s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-7-3]
0.25s call     tests/distributions/test_stable.py::test_sample[1.7-1.0]
0.25s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-1-7]
0.25s call     tests/infer/test_conjugate_gradients.py::ConjugateChainGradientTests::test_gradients
0.25s call     tests/infer/test_util.py::test_psis_diagnostic[5-0.5-krange0-500000]
0.25s call     tests/ops/test_integrator.py::test_time_reversibility[CircularPlanetaryMotion]
0.25s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-()-()]
0.25s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-True-0-1000-False-2]
0.25s call     tests/infer/test_jit.py::test_svi_enum[2-parallel-sequential]
0.25s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-parallel-sequential-None]
0.25s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-3-7]
0.25s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.25s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-None-None-None]
0.25s call     tests/infer/test_util.py::test_psis_diagnostic[5-0.5-krange0-250000]
0.25s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-3-1]
0.24s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-3-7]
0.24s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-3-1]
0.24s call     tests/infer/test_discrete.py::test_hmm_smoke[sample-100]
0.24s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-sequential-None-None]
0.24s call     tests/infer/test_util.py::test_psis_diagnostic[5-0.95-krange1-250000]
0.24s call     tests/contrib/epidemiology/test_models.py::test_overdispersed_seir_smoke[{'haar_full_mass': 2}-7-3]
0.24s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.24s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-0-7]
0.24s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-7-7]
0.24s call     tests/distributions/test_conjugate.py::test_variance[dist2]
0.24s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-3-7]
0.24s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-0-7]
0.24s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-7-3]
0.24s call     tests/infer/test_smcfilter.py::test_likelihood_ratio
0.24s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-parallel-None-None]
0.24s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-()-()]
0.24s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-None-None-None]
0.24s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-0-7]
0.24s call     tests/contrib/easyguide/test_easyguide.py::test_serialize
0.24s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-3-7]
0.24s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-1-1]
0.24s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-3-7]
0.24s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-0-3]
0.24s call     tests/distributions/test_conjugate.py::test_mean[dist2]
0.24s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-3-7]
0.24s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-3-7]
0.24s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-3]
0.24s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-3-1]
0.24s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-2]
0.24s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-10]
0.24s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-3-7]
0.24s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-2]
0.23s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-1]
0.23s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-True-0-1000-False-2]
0.23s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-0-7]
0.23s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-3]
0.23s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-1]
0.23s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-10]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-1-7]
0.23s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-sequential-None-None]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-1-7]
0.23s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-False-0-1000-False-2]
0.23s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-0-3]
0.23s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-7-7]
0.23s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-sequential-None-None]
0.23s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-2-1]
0.23s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-2-1]
0.23s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-parallel-parallel-None]
0.23s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-0-7]
0.23s call     tests/infer/test_valid_models.py::test_enum_recycling_nested
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-1-1]
0.23s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-None-sequential-None]
0.23s call     tests/contrib/epidemiology/test_models.py::test_overdispersed_seir_smoke[{}-7-3]
0.23s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-40-False]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-1-1]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-1-7]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-1-1]
0.23s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-parallel-parallel]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-1-7]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-3-7]
0.23s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-None-parallel-None]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-1-7]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-3-1]
0.23s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-None-parallel]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-1-7]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-1-7]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-3-7]
0.23s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-None-None-None]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-3-7]
0.23s call     tests/distributions/test_stable.py::test_sample[1.3--0.5]
0.23s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.23s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-1-1]
0.23s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-3-7]
0.22s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-7-3]
0.22s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-0-3]
0.22s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-False-0-1000-False-2]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-3-7]
0.22s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-4-3-MixtureOfDiagNormalsSharedCovariance]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-3-1]
0.22s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-7-7]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-3-7]
0.22s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-1-1]
0.22s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-7-3]
0.22s call     tests/distributions/test_stable.py::test_sample[1.3-0.5]
0.22s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-sequential-None]
0.22s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-2-3-MixtureOfDiagNormals]
0.22s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-7-3]
0.22s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-1.0]
0.22s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.75]
0.22s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.5]
0.22s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.0]
0.22s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.25]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-1-1]
0.22s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-3-2-(4,)]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-3-7]
0.22s call     tests/contrib/forecast/test_evaluate.py::test_simple[warm-None-10-None-10-1]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-3-7]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-1-7]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-3-1]
0.22s call     tests/contrib/forecast/test_evaluate.py::test_poisson[svi-None-10-None-10-1]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-3-7]
0.22s call     tests/ops/test_contract.py::test_naive_ubersum[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.22s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.0]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-3-7]
0.22s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.75]
0.22s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.25]
0.22s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.5]
0.22s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-1.0]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-1-1]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-3-7]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-3-7]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-1-7]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-3-7]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-3-1]
0.22s call     tests/contrib/forecast/test_evaluate.py::test_simple[cold-None-10-None-10-1]
0.22s call     tests/distributions/test_stable.py::test_sample[0.8--0.5]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-3-1]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-1-1]
0.22s call     tests/distributions/test_stable.py::test_sample[0.8-0.5]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-3-7]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-1-7]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-3-1]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-3-7]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-1-1]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-1-7]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-3-1]
0.22s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-3-7]
0.22s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-0-7]
0.21s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-0-7]
0.21s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-7-7]
0.21s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-7-3]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-1-7]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-1-7]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-3-1]
0.21s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-None-sequential]
0.21s call     tests/distributions/test_stable.py::test_sample[1.01--1.0]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-3-1]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-3-1]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-1-7]
0.21s call     tests/infer/reparam/test_studentt.py::test_moments[()]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.5]
0.21s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-None-parallel]
0.21s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-1-7]
0.21s call     tests/infer/mcmc/test_nuts.py::test_beta_binomial[False]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-1-7]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-1-7]
0.21s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-()-()]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-1-7]
0.21s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-sequential-None-None]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-1-7]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.75]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.5]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-1.0]
0.21s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-7-3]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.0]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-1-1]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-1-1]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-1.0]
0.21s call     tests/ops/test_contract.py::test_ubersum[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.21s call     tests/distributions/test_stable.py::test_sample[0.99--1.0]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.25]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.25]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.75]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.0]
0.21s call     tests/distributions/test_stable.py::test_sample[0.4--1.0]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-1-1]
0.21s call     tests/infer/test_smcfilter.py::test_smoke[10-7-5-1]
0.21s call     tests/distributions/test_stable.py::test_sample[0.8-0.0]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-3-1]
0.21s call     tests/distributions/test_stable.py::test_sample[2.0-1.0]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-3-1]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-3-7]
0.21s call     tests/infer/test_smcfilter.py::test_smoke[10-7-5-2]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-3-7]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-3-1]
0.21s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-0-3]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-1.0]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-1-7]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.5]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-1-1]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.0]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.75]
0.21s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.25]
0.21s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-7-7]
0.21s call     tests/distributions/test_stable.py::test_sample[2.0--0.5]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-3-7]
0.21s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-1000]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-1-7]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-1-1]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-1-7]
0.21s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-7]
0.21s call     tests/distributions/test_stable.py::test_sample[2.0-0.5]
0.21s call     tests/distributions/test_stable.py::test_sample[2.0-0.0]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-3-7]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-1-7]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-1-1]
0.21s call     tests/distributions/test_stable.py::test_sample[2.0--1.0]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-3-1]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-1-1]
0.21s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-3]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-1-1]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-3-7]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-1-1]
0.21s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-0-3]
0.21s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-3-1]
0.21s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-0-7]
0.21s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-None-None]
0.21s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-()-()]
0.21s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-()-()]
0.21s call     tests/infer/test_smcfilter.py::test_smoke[10-7-2-2]
0.20s call     tests/infer/test_smcfilter.py::test_smoke[10-7-2-1]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.0]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-3-1]
0.20s call     tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{}-7-12]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-3-1]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.75]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.5]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.25]
0.20s call     tests/ops/test_contract.py::test_ubersum[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.20s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-0-3]
0.20s call     tests/distributions/test_stable.py::test_sample[1.3-1.0]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-1.0]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.5]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-3-1]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-1-1]
0.20s call     tests/distributions/test_stable.py::test_sample[0.4-1.0]
0.20s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-1000]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.0]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-1-1]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.75]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.25]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-1.0]
0.20s call     tests/contrib/tracking/test_em.py::test_em[True]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-3-7]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-1-1]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-3-7]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-3-1]
0.20s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-1000]
0.20s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-1000]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-3-7]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.25]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.75]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.0]
0.20s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-1000]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-1-7]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.5]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-1.0]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.5]
0.20s call     tests/distributions/test_mvt.py::test_log_prob[1-(3, 2)]
0.20s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-2]
0.20s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-1000]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-3-7]
0.20s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-1000]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.0]
0.20s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-1000]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.25]
0.20s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-1000]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-1.0]
0.20s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.75]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-3-1]
0.20s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-parallel-parallel-None]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-1-7]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-3-1]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-3-7]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-1-7]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-3-1]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-1-7]
0.20s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-sequential-sequential]
0.20s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-None-parallel-None]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-3-7]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-1-7]
0.20s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-3]
0.20s call     tests/ops/test_integrator.py::test_energy_conservation[QuarticOscillator]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-1-1]
0.20s call     tests/distributions/test_stable.py::test_sample[0.99-1.0]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-1-1]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-3-1]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-1-1]
0.20s call     tests/ops/test_integrator.py::test_trajectory[QuarticOscillator]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-3-1]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-3-1]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-3-1]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-3-7]
0.20s call     tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{}-7-4]
0.20s call     tests/contrib/forecast/test_evaluate.py::test_poisson[svi-14-1-7-1-1]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-3-7]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-3-1]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-1-1]
0.20s call     tests/ops/test_contract.py::test_ubersum_jit[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-1-1]
0.20s call     tests/distributions/test_stable.py::test_sample[1.01-1.0]
0.20s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-7-3]
0.20s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-1-7]
0.19s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-()-()]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-1-1]
0.19s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-0-3]
0.19s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-1]
0.19s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-7-3]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-1-7]
0.19s call     tests/distributions/test_stable.py::test_sample[0.8--1.0]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-3-1]
0.19s call     tests/contrib/forecast/test_evaluate.py::test_simple[cold-14-1-7-1-1]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-3-1]
0.19s call     tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[12]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-1-7]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.5]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-1-1]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-1-7]
0.19s call     tests/infer/test_jit.py::test_traceenum_elbo[10]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-3-1]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-1-1]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-3-1]
0.19s call     tests/distributions/test_stable.py::test_sample[1.3--1.0]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-1-7]
0.19s call     tests/distributions/test_conjugate.py::test_variance[dist0]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-1-7]
0.19s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-()-()]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.0]
0.19s call     tests/distributions/test_conjugate.py::test_mean[dist0]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-1.0]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.75]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.25]
0.19s call     tests/distributions/test_transforms.py::TransformTests::test_spline_coupling
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-3-1]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-3-1]
0.19s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-parallel-None-None]
0.19s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-7]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-3-7]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.0]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.5]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.75]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.25]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-1-1]
0.19s call     tests/infer/test_valid_models.py::test_enum_recycling_chain
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-1.0]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-1-7]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-3-7]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-3-7]
0.19s call     tests/distributions/test_stable.py::test_sample[1.3-0.0]
0.19s call     tests/ops/test_contract.py::test_ubersum[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-1-1]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-1-1]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.5]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-1-1]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-1-1]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.75]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.0]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-1.0]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.25]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-1-1]
0.19s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-0-3]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-3-1]
0.19s call     tests/distributions/test_stable.py::test_sample[0.8-1.0]
0.19s call     tests/infer/test_jit.py::test_infer_discrete[map-10]
0.19s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-parallel-parallel-None]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-3-1]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-3-1]
0.19s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-0-7]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-3-1]
0.19s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-1-1]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.25]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-1-1]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.0]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.75]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.5]
0.19s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-1.0]
0.19s call     tests/distributions/test_mvt.py::test_log_prob[2-(4,)]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-1-1]
0.19s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-1000]
0.19s call     tests/infer/test_smcfilter.py::test_smoke[10-7-1-2]
0.19s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-1-1]
0.19s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-7-3]
0.19s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-1000]
0.19s call     tests/infer/test_smcfilter.py::test_smoke[10-7-1-1]
0.19s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-1000]
0.19s call     tests/ops/test_contract.py::test_naive_ubersum[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.19s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-7-3]
0.19s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-1000]
0.19s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-1000]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-3-1]
0.18s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-None-None-None]
0.18s call     tests/infer/test_discrete.py::test_hmm_smoke[map-100]
0.18s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-1-1]
0.18s call     tests/distributions/test_conjugate.py::test_gamma_poisson_log_prob[shape0]
0.18s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-10.0]
0.18s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-7-7]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-1-7]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-1-1]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-3-1]
0.18s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.75]
0.18s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.25]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-1-7]
0.18s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-1.0]
0.18s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.5]
0.18s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.0]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-1-1]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-1-7]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-3-1]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-3-7]
0.18s call     tests/infer/test_enum.py::test_compute_marginals_hmm[30]
0.18s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-0-3]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-3-1]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-3-7]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-1-1]
0.18s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-0-7]
0.18s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-30-False]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-3-7]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-3-7]
0.18s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-None-parallel-None]
0.18s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-7-7]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-3-7]
0.18s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-4-3-MixtureOfDiagNormalsSharedCovariance]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-1-1]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-3-7]
0.18s call     tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-30-False]
0.18s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-parallel-None]
0.18s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-()-()]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-1-1]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-1-7]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-1-1]
0.18s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-1-7]
0.17s call     tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-0-3]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-3-1]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-1-7]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-3-1]
0.17s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-parallel-None]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-3-1]
0.17s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-None-None-None]
0.17s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-0-3]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-1-7]
0.17s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.75]
0.17s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.0]
0.17s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.25]
0.17s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.5]
0.17s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-1.0]
0.17s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-3]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-1-7]
0.17s call     tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-0-3]
0.17s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-7-7]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-1-7]
0.17s call     tests/ops/test_contract.py::test_naive_ubersum[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-3-1]
0.17s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-parallel-parallel-None]
0.17s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-4]
0.17s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-1-3]
0.17s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-sequential-sequential-None]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-3-1]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-3-1]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-3-1]
0.17s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-parallel-None-None]
0.17s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.25]
0.17s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.0]
0.17s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.5]
0.17s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-1.0]
0.17s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.75]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-1-1]
0.17s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-None-parallel-None]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-1-1]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-1-1]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-3-1]
0.17s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-None-parallel]
0.17s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-3-1]
0.17s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-0-7]
0.16s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-2-3-MixtureOfDiagNormals]
0.16s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-1-1]
0.16s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-1-1-(3, 2)]
0.16s call     tests/ops/test_contract.py::test_ubersum_jit[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.16s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-1-1]
0.16s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-0.2]
0.16s call     tests/infer/mcmc/test_nuts.py::test_gamma_poisson[False]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-1.0]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.25]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.5]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.0]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.75]
0.16s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-1-1]
0.16s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-1-1]
0.16s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-1-1]
0.16s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-2-3-MixtureOfDiagNormalsSharedCovariance]
0.16s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-1-1]
0.16s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-1-1]
0.16s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-1-1]
0.16s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-1-1]
0.16s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-3]
0.16s call     tests/ops/test_newton.py::test_newton_step_converges[3-0.1]
0.16s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-1-7]
0.16s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-3-2-(3, 2)]
0.16s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-parallel-None]
0.16s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-None-parallel-None]
0.16s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-0-7]
0.16s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-3-7]
0.16s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-4-3-MixtureOfDiagNormals]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-1.0]
0.16s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-3-2-()]
0.16s call     tests/contrib/cevae/test_cevae.py::test_smoke[studentt-1-100]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.75]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.0]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.25]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.5]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-1.0]
0.16s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-4-3-MixtureOfDiagNormals]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.75]
0.16s call     tests/infer/test_jit.py::test_svi[TraceEnum_ELBO-1]
0.16s call     tests/infer/test_jit.py::test_svi_enum[1-sequential-sequential]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.5]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.25]
0.16s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.0]
0.16s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.5]
0.16s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.75]
0.16s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.0]
0.16s call     tests/ops/test_contract.py::test_ubersum[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.16s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.25]
0.16s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-1.0]
0.15s call     tests/infer/test_smcfilter.py::test_gaussian_filter
0.15s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-parallel-None-None]
0.15s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-3-2-()]
0.15s call     tests/infer/reparam/test_stable.py::test_stable[StableReparam-()]
0.15s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-None-None]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.25]
0.15s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-3-2-()]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-1.0]
0.15s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-3-7]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.5]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.75]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.0]
0.15s call     tests/distributions/test_rejector.py::test_rejector[0.5-1.0]
0.15s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-4-3-MixtureOfDiagNormalsSharedCovariance]
0.15s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-None-None-None]
0.15s call     tests/contrib/cevae/test_cevae.py::test_smoke[normal-2-100]
0.15s call     tests/distributions/test_rejector.py::test_rejector[0.5-0.5]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.5]
0.15s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-3-7]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.75]
0.15s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-None-parallel-None]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.5]
0.15s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-0-7]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.0]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.25]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-1.0]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.0]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-1.0]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.75]
0.15s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.25]
0.15s call     tests/distributions/test_rejector.py::test_rejector[0.5-2.0]
0.15s call     tests/infer/reparam/test_stable.py::test_symmetric_stable[()]
0.15s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-parallel-None-None]
0.15s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-7-3]
0.15s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-1-7]
0.15s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-0.2]
0.15s call     tests/distributions/test_stable.py::test_sample[1.0--0.5]
0.15s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-0.2]
0.15s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-sequential-sequential-None]
0.15s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-0.2]
0.15s call     tests/infer/mcmc/test_mcmc_util.py::test_model_with_subsample[10]
0.15s call     tests/distributions/test_transforms.py::TransformTests::test_conditional_spline
0.14s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-()-()]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.5]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.0]
0.14s call     tests/infer/reparam/test_transform.py::test_log_normal[(2, 3)]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.75]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.75]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.25]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-1.0]
0.14s call     tests/distributions/test_transforms.py::TransformTests::test_spline
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-1.0]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.25]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.0]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.5]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.75]
0.14s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-7-7]
0.14s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-1-7]
0.14s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-None-None]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.25]
0.14s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-3-7]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.0]
0.14s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-1-1-(3, 2)]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-1.0]
0.14s call     tests/distributions/test_spanning_tree.py::test_log_prob[5]
0.14s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-3-2-(4,)]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.5]
0.14s call     tests/ops/test_stats.py::test_resample[False]
0.14s call     tests/distributions/test_stable.py::test_sample[1.0-0.5]
0.14s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-3-7]
0.14s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-1-7]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.75]
0.14s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-sequential-parallel]
0.14s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-7-7]
0.14s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-7-7]
0.14s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-7-7]
0.14s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-None-None]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.25]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-1.0]
0.14s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-0-3]
0.14s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-0.5]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.0]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.5]
0.14s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-1.0]
0.14s call     tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-1-100]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.5]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-1.0]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.75]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.25]
0.14s call     tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.0]
0.14s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-0.5]
0.14s call     tests/distributions/test_conjugate.py::test_variance[dist5]
0.14s call     tests/ops/test_contract.py::test_naive_ubersum[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.14s call     tests/distributions/test_mvt.py::test_log_prob[1-(4,)]
0.14s call     tests/infer/test_enum.py::test_elbo_rsvi[sequential]
0.14s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-0.2]
0.14s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-0.5]
0.14s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-3-7]
0.14s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-()-()]
0.14s call     tests/distributions/test_conjugate.py::test_mean[dist5]
0.13s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-None-None-None]
0.13s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-7-7]
0.13s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-0.5]
0.13s call     tests/contrib/tracking/test_assignment.py::test_persistent_smoke[enum]
0.13s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-1-7]
0.13s call     tests/infer/test_jit.py::test_svi_enum[2-sequential-parallel]
0.13s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-7-3]
0.13s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-None-None-None]
0.13s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-()-()]
0.13s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-None-None]
0.13s call     tests/ops/test_contract.py::test_ubersum_jit[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.13s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-1-1-(4,)]
0.13s call     tests/infer/test_valid_models.py::test_enum_recycling_dbn[False-True]
0.13s call     tests/infer/test_valid_models.py::test_enum_recycling_dbn[False-False]
0.13s call     tests/ops/test_integrator.py::test_trajectory[CircularPlanetaryMotion]
0.13s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-4]
0.13s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-7-3]
0.13s call     tests/ops/test_contract.py::test_ubersum_jit[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.13s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-(2, 1)-(3,)]
0.13s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-(2, 1)-(3,)]
0.13s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-(2, 1)-(3,)]
0.13s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-0-7]
0.13s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-(2, 1)-(3,)]
0.13s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-3-2]
0.13s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-3-7]
0.13s call     tests/distributions/test_rejector.py::test_gamma_elbo[0.2-1.0]
0.13s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-(2, 1)-(3,)]
0.13s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-1-1-(3, 2)]
0.13s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.6-()]
0.13s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-10]
0.13s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-4-3-MixtureOfDiagNormalsSharedCovariance]
0.13s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-(2, 1)-(3,)]
0.13s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-None-None-None]
0.13s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-(2, 1)-(3,)]
0.13s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-(2, 1)-(3,)]
0.13s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-1-7]
0.13s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-3-7]
0.13s call     tests/ops/test_contract.py::test_ubersum_jit[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.13s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-(2, 3)-(2, 3)]
0.12s call     tests/ops/test_integrator.py::test_energy_conservation[CircularPlanetaryMotion]
0.12s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-0.5]
0.12s call     tests/infer/test_enum.py::test_differentiable_loss[sequential-single]
0.12s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-2-3-MixtureOfDiagNormalsSharedCovariance]
0.12s call     tests/distributions/test_transforms.py::TransformTests::test_neural_autoregressive
0.12s call     tests/distributions/test_transforms.py::TransformTests::test_spline_autoregressive
0.12s call     tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceEnum_ELBO-nonreparam]
0.12s call     tests/contrib/cevae/test_cevae.py::test_smoke[studentt-2-100]
0.12s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-0-7]
0.12s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-0-7]
0.12s call     tests/distributions/test_rejector.py::test_gamma_elbo[2.0-1.0]
0.12s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-0-7]
0.12s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-20-False]
0.12s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-1.0]
0.12s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-0-None]
0.12s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-2.0]
0.12s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.6-()]
0.12s call     tests/distributions/test_mvt.py::test_mean_var[(3, 2)]
0.12s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-None-None]
0.12s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-(2, 3)-(2, 3)]
0.12s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-(2, 3)-(2, 3)]
0.12s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-1-None]
0.12s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-0-2000]
0.12s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-1-2000]
0.12s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-(2, 3)-(2, 3)]
0.12s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-2.0]
0.12s call     tests/infer/test_jit.py::test_svi[TraceGraph_ELBO-1]
0.12s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-2]
0.12s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-2]
0.12s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-3]
0.12s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-10]
0.12s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-3]
0.12s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-1]
0.12s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-()-()]
0.12s call     tests/distributions/test_rejector.py::test_gamma_elbo[0.2-2.0]
0.12s call     tests/infer/test_jit.py::test_svi[TraceMeanField_ELBO-1]
0.12s call     tests/contrib/forecast/test_evaluate.py::test_custom_warm_start
0.12s call     tests/contrib/cevae/test_cevae.py::test_smoke[laplace-2-100]
0.12s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-1]
0.12s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-1.0-()]
0.12s call     tests/infer/test_jit.py::test_svi[Trace_ELBO-1]
0.12s call     tests/distributions/test_rejector.py::test_gamma_elbo[2.0-2.0]
0.12s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-parallel-parallel]
0.12s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-1.0]
0.12s call     tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-20-False]
0.12s call     tests/infer/test_autoguide.py::test_subsample_guide[AutoGuideList-init_to_median]
0.12s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-(2, 3)-(2, 3)]
0.12s call     tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[8]
0.12s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-(2, 3)-(2, 3)]
0.12s call     tests/infer/test_autoguide.py::test_subsample_guide[AutoGuideList-None]
0.12s call     tests/infer/test_autoguide.py::test_subsample_guide[AutoNormal-init_to_mean]
0.12s call     tests/infer/test_autoguide.py::test_subsample_guide[AutoGuideList-init_to_mean]
0.12s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-(2, 3)-(2, 3)]
0.12s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-2-3-MixtureOfDiagNormals]
0.12s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-(2, 3)-(2, 3)]
0.12s call     tests/infer/test_autoguide.py::test_subsample_guide[AutoNormal-None]
0.12s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[sequential-True-None]
0.12s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-0-7]
0.12s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-2.0]
0.12s call     tests/contrib/cevae/test_cevae.py::test_smoke[normal-1-100]
0.12s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-1-7]
0.12s call     tests/infer/test_autoguide.py::test_subsample_guide[AutoNormal-init_to_median]
0.12s call     tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape3]
0.12s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-10]
0.12s call     tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape3]
0.12s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-sequential-True-None]
0.12s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-sequential-False-None]
0.12s call     tests/contrib/cevae/test_cevae.py::test_smoke[laplace-1-100]
0.12s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-sequential-parallel]
0.12s call     tests/distributions/test_rejector.py::test_gamma_elbo[1.0-1.0]
0.11s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[sequential-False-None]
0.11s call     tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-2-100]
0.11s call     tests/contrib/cevae/test_cevae.py::test_smoke[exponential-2-100]
0.11s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_three_layers_model_permuted
0.11s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-3-7]
0.11s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-1.0]
0.11s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-sequential-inf]
0.11s call     tests/infer/test_smcfilter.py::test_smoke[10-3-5-2]
0.11s call     tests/infer/test_smcfilter.py::test_smoke[10-3-5-1]
0.11s call     tests/distributions/test_rejector.py::test_gamma_elbo[5.0-2.0]
0.11s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.11s call     tests/infer/test_smcfilter.py::test_smoke[10-3-2-2]
0.11s call     tests/ops/test_contract.py::test_ubersum_jit[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.11s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-1.0]
0.11s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-0-3]
0.11s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.11s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.11s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.11s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.11s call     tests/contrib/cevae/test_cevae.py::test_smoke[exponential-1-100]
0.11s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.11s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.11s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.11s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-parallel-sequential]
0.11s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-sequential-1]
0.11s call     tests/distributions/test_rejector.py::test_gamma_elbo[0.5-2.0]
0.11s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-1-7]
0.11s call     tests/distributions/test_rejector.py::test_gamma_elbo[5.0-1.0]
0.11s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-2.0]
0.11s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-7-3]
0.11s call     tests/distributions/test_stable.py::test_sample[1.0--1.0]
0.11s call     tests/infer/test_smcfilter.py::test_smoke[10-3-2-1]
0.11s call     tests/infer/test_autoguide.py::test_nested_autoguide[TraceEnum_ELBO]
0.11s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-None-()]
0.11s call     tests/distributions/test_rejector.py::test_standard_gamma_elbo[2.0]
0.11s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-2.0]
0.11s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-3-7]
0.11s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-5.0]
0.11s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-()-()]
0.11s call     tests/infer/mcmc/test_adaptation.py::test_arrowhead_mass_matrix[False]
0.11s call     tests/ops/test_contract.py::test_ubersum[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.11s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-1-1-(4,)]
0.11s call     tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-0-3]
0.11s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-5.0]
0.11s call     tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceEnum_ELBO-reparam]
0.11s call     tests/distributions/test_rejector.py::test_gamma_elbo[0.5-1.0]
0.11s call     tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.11s call     tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceGraph_ELBO-reparam]
0.11s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-7-3]
0.11s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.0-()]
0.11s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-3-7]
0.11s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-3-7]
0.11s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-3-7]
0.11s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-1-7]
0.10s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-parallel-None]
0.10s call     tests/distributions/test_rejector.py::test_gamma_elbo[1.0-5.0]
0.10s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-1-7]
0.10s call     tests/ops/test_contract.py::test_naive_ubersum[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.10s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.0-()]
0.10s call     tests/distributions/test_rejector.py::test_standard_gamma_elbo[1.0]
0.10s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-3]
0.10s call     tests/infer/test_util.py::test_psis_diagnostic[1-0.95-krange1-250000]
0.10s call     tests/distributions/test_rejector.py::test_gamma_elbo[1.0-2.0]
0.10s call     tests/distributions/test_rejector.py::test_gamma_elbo[2.0-5.0]
0.10s call     tests/infer/test_smcfilter.py::test_smoke[10-3-1-2]
0.10s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-None-()]
0.10s call     tests/infer/test_smcfilter.py::test_smoke[10-3-1-1]
0.10s call     tests/ops/test_contract.py::test_ubersum_jit[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.10s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-()-()]
0.10s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-1-7]
0.10s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-5.0]
0.10s call     tests/infer/reparam/test_loc_scale.py::test_normal[Normal-centered3-()]
0.10s call     tests/distributions/test_rejector.py::test_gamma_elbo[0.2-5.0]
0.10s call     tests/ops/test_contract.py::test_ubersum[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.10s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-2]
0.10s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-5.0]
0.10s call     tests/ops/test_contract.py::test_ubersum_jit[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.10s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-3-7]
0.10s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-None-parallel]
0.10s call     tests/distributions/test_rejector.py::test_gamma_elbo[0.5-5.0]
0.10s call     tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-5.0]
0.10s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-3-1]
0.10s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-1-1-(4,)]
0.10s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-sequential-sequential-None]
0.10s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-3-1]
0.10s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-1]
0.10s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-2-3-MixtureOfDiagNormalsSharedCovariance]
0.10s call     tests/distributions/test_rejector.py::test_gamma_elbo[5.0-5.0]
0.10s call     tests/distributions/test_stable.py::test_sample[1.0-1.0]
0.10s call     tests/distributions/test_transforms.py::TransformTests::test_permute
0.10s call     tests/distributions/test_rejector.py::test_standard_gamma_elbo[5.0]
0.10s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-centered3-()]
0.10s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-parallel-None]
0.10s call     tests/ops/test_contract.py::test_ubersum[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.10s call     tests/distributions/test_transforms.py::TransformTests::test_discrete_cosine
0.09s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-5]
0.09s call     tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-1.0-()]
0.09s call     tests/infer/test_valid_models.py::test_enum_recycling_plate
0.09s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-7-3]
0.09s call     tests/infer/test_enum.py::test_compute_marginals_hmm[20]
0.09s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-7-3]
0.09s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-0-3]
0.09s call     tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoMultivariateNormal]
0.09s call     tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.44-0.41]
0.09s call     tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoLowRankMultivariateNormal]
0.09s call     tests/infer/reparam/test_transform.py::test_log_normal[(4,)]
0.09s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-7-3]
0.09s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-sequential-False-None]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-1-7]
0.09s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-sequential-True-None]
0.09s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-7-3]
0.09s call     tests/ops/test_contract.py::test_naive_ubersum[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.09s call     tests/contrib/easyguide/test_easyguide.py::test_amortized_smoke[init_to_median]
0.09s call     tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[Trace_ELBO-reparam]
0.09s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-5]
0.09s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-sequential-sequential-None]
0.09s call     tests/infer/test_util.py::test_psis_diagnostic[1-0.5-krange0-500000]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-3-1]
0.09s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-()-()]
0.09s call     tests/ops/test_contract.py::test_naive_ubersum[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.09s call     tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[DiffTrace_ELBO-nonreparam]
0.09s call     tests/infer/test_abstract_infer.py::test_nesting
0.09s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-None-None]
0.09s call     tests/contrib/easyguide/test_easyguide.py::test_amortized_smoke[None]
0.09s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-3-2-()]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-1-7]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-3-1]
0.09s call     tests/infer/test_jit.py::test_svi_enum[1-parallel-sequential]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-3-1]
0.09s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-(2, 3)-(2, 3)]
0.09s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-()-()]
0.09s call     tests/contrib/easyguide/test_easyguide.py::test_amortized_smoke[init_to_mean]
0.09s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-(2, 3)-(2, 3)]
0.09s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-None-parallel]
0.09s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-(2, 1)-(3,)]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-1-1]
0.09s call     tests/contrib/easyguide/test_easyguide.py::test_subsample_smoke[init_to_median]
0.09s call     tests/infer/test_enum.py::test_elbo_rsvi[parallel]
0.09s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-4]
0.09s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-1-1-(3, 2)]
0.09s call     tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-0-3]
0.09s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-2-3-MixtureOfDiagNormals]
0.09s call     tests/contrib/easyguide/test_easyguide.py::test_subsample_smoke[None]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-1-7]
0.09s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-parallel-sequential-None]
0.09s call     tests/contrib/easyguide/test_easyguide.py::test_subsample_smoke[init_to_mean]
0.09s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-5]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-1-1]
0.09s call     tests/infer/test_util.py::test_psis_diagnostic[1-0.95-krange1-500000]
0.09s call     tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_guide_module_callable]
0.09s call     tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[DiffTrace_ELBO-reparam]
0.09s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-(2, 1)-(3,)]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-3-7]
0.09s call     tests/infer/reparam/test_stable.py::test_stable[LatentStableReparam-()]
0.09s call     tests/infer/mcmc/test_adaptation.py::test_arrowhead_mass_matrix[True]
0.09s call     tests/infer/test_jit.py::test_svi_enum[1-sequential-parallel]
0.09s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-4-3-GaussianScaleMixture]
0.09s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-full-inf]
0.09s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-7-3]
0.09s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-2-2]
0.09s call     tests/infer/test_enum.py::test_elbo_rsvi[None]
0.09s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-full-inf]
0.09s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-sequential-parallel-None]
0.09s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-(2, 1)-(3,)]
0.09s call     tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-3-7]
0.09s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-(2, 1)-(3,)]
0.09s call     tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_guide_callable]
0.09s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-parallel-sequential-None]
0.09s call     tests/infer/test_gradient.py::test_plate[TraceEnum_ELBO-nonreparam]
0.09s call     tests/distributions/test_spanning_tree.py::test_make_complete_graph[cpp-2-expected_grid0]
0.09s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-(2, 1)-(3,)]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-1-7]
0.09s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-3-7]
0.09s call     tests/ops/test_newton.py::test_newton_step_converges[2-0.1]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-3-7]
0.08s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-(2, 3)-(2, 3)]
0.08s call     tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceGraph_ELBO-nonreparam]
0.08s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-full-inf]
0.08s call     tests/ops/test_contract.py::test_ubersum[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.08s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-full-inf]
0.08s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-2-3-(3, 2)]
0.08s call     tests/infer/test_autoguide.py::test_nested_autoguide[TraceGraph_ELBO]
0.08s call     tests/ops/test_contract.py::test_naive_ubersum[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.08s call     tests/infer/test_gradient.py::test_plate[TraceEnum_ELBO-reparam]
0.08s call     tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.17-0.41]
0.08s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-0-3]
0.08s call     tests/infer/test_util.py::test_psis_diagnostic[1-0.5-krange0-250000]
0.08s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-parallel-True-3]
0.08s call     tests/infer/reparam/test_haar.py::test_normal[(2, 5)--1-False]
0.08s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-(2, 3)-(2, 3)]
0.08s call     tests/infer/test_jit.py::test_svi[JitTraceEnum_ELBO-1]
0.08s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-0.0]
0.08s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-(2, 3)-(2, 3)]
0.08s call     tests/infer/test_autoguide.py::test_nested_autoguide[Trace_ELBO]
0.08s call     tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[Trace_ELBO-nonreparam]
0.08s call     tests/infer/reparam/test_haar.py::test_normal[(2, 5)--1-True]
0.08s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-full-0]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-1-7]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-3-7]
0.08s call     tests/distributions/test_mvt.py::test_mean_var[(4,)]
0.08s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-0-3]
0.08s call     tests/contrib/easyguide/test_easyguide.py::test_delta_smoke[init_to_median]
0.08s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-(2, 3)-(2, 3)]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-1-1]
0.08s call     tests/contrib/tracking/test_em.py::test_em[False]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-1-1]
0.08s call     tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_guide_list_x]
0.08s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-(2, 1)-(3,)]
0.08s call     tests/infer/reparam/test_conjugate.py::test_beta_binomial_hmc
0.08s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-sequential-sequential-None]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-3-7]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-3-7]
0.08s call     tests/infer/test_enum.py::test_elbo_iplate[2-None-sequential]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-3-7]
0.08s call     tests/distributions/test_ig.py::test_sample[2.5-3.3]
0.08s call     tests/contrib/forecast/test_evaluate.py::test_simple[cold-None-1-None-1-8]
0.08s call     tests/contrib/easyguide/test_easyguide.py::test_delta_smoke[None]
0.08s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-0.5]
0.08s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-full-0]
0.08s call     tests/contrib/forecast/test_evaluate.py::test_poisson[svi-None-1-None-1-8]
0.08s call     tests/contrib/easyguide/test_easyguide.py::test_delta_smoke[init_to_mean]
0.08s call     tests/contrib/forecast/test_evaluate.py::test_simple[warm-None-1-None-1-8]
0.08s call     tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-1-1]
0.08s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-0-2000]
0.08s call     tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-10]
0.08s call     tests/distributions/test_ig.py::test_sample[3.0-3.3]
0.08s call     tests/distributions/test_ig.py::test_sample[3.0-4.0]
0.08s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-1-2000]
0.08s call     tests/distributions/test_ig.py::test_sample[2.5-4.0]
0.08s call     tests/contrib/cevae/test_cevae.py::test_smoke[normal-2-1]
0.08s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-0-3]
0.08s call     tests/contrib/forecast/test_evaluate.py::test_simple[warm-None-5-10-1-5]
0.08s call     tests/infer/test_valid_models.py::test_markov_history[3]
0.08s call     tests/infer/test_valid_models.py::test_markov_history[2]
0.08s call     tests/contrib/forecast/test_evaluate.py::test_poisson[svi-None-5-10-1-5]
0.08s call     tests/contrib/forecast/test_evaluate.py::test_poisson[svi-10-1-None-3-5]
0.08s call     tests/contrib/cevae/test_cevae.py::test_smoke[laplace-1-1]
0.08s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-0-3]
0.08s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-full-0]
0.08s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-2.0]
0.08s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-(2, 3)-(2, 3)]
0.08s call     tests/contrib/forecast/test_evaluate.py::test_simple[cold-10-1-None-3-5]
0.08s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-3-False]
0.08s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-3-True]
0.08s call     tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoDiagonalNormal]
0.08s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-1.0]
0.08s call     tests/contrib/forecast/test_evaluate.py::test_simple[cold-None-5-10-1-5]
0.08s call     tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.44-0.33]
0.08s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-sequential-parallel]
0.08s call     tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoMultivariateNormal]
0.08s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_two_layers
0.08s call     tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.17-0.41]
0.08s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-full-0]
0.08s call     tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[30]
0.08s call     tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_class9]
0.08s call     tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_class8]
0.08s call     tests/infer/test_autoguide.py::test_subsample_guide[AutoDelta-None]
0.08s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.25]
0.08s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.5]
0.08s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.75]
0.08s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-5]
0.08s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-1.0]
0.08s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.0]
0.08s call     tests/infer/test_autoguide.py::test_subsample_guide[AutoDelta-init_to_mean]
0.08s call     tests/contrib/cevae/test_cevae.py::test_smoke[normal-1-1]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-1-7]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-3-1]
0.08s call     tests/infer/test_autoguide.py::test_subsample_guide[AutoDelta-init_to_median]
0.08s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-3]
0.08s call     tests/ops/test_contract.py::test_ubersum_jit[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.08s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-2-3-(4,)]
0.08s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-()-()]
0.08s call     tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.44-0.41]
0.08s call     tests/contrib/cevae/test_cevae.py::test_smoke[exponential-2-1]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-3-7]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-3-7]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-1-7]
0.08s call     tests/contrib/cevae/test_cevae.py::test_smoke[studentt-2-1]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-1-1]
0.08s call     tests/infer/test_enum.py::test_hmm_enumerate_model[30]
0.08s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-1-7]
0.07s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-4-3-GaussianScaleMixture]
0.07s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[parallel-True-3]
0.07s call     tests/contrib/cevae/test_cevae.py::test_smoke[studentt-1-1]
0.07s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-None-None]
0.07s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-2-3-(3, 2)]
0.07s call     tests/ops/test_contract.py::test_ubersum_jit[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.07s call     tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.17-0.33]
0.07s call     tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-0-3]
0.07s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-2-3-(3, 2)]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-1-7]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-3-7]
0.07s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-(2, 1)-(3,)]
0.07s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-2-3-MixtureOfDiagNormalsSharedCovariance]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-1.7--0.5-0.5]
0.07s call     tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoMultivariateNormal]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-3-1]
0.07s call     tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.44-0.33]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-1-7]
0.07s call     tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoLaplaceApproximation]
0.07s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-0-None]
0.07s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-3]
0.07s call     tests/infer/test_jit.py::test_svi[JitTraceGraph_ELBO-1]
0.07s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-1-None]
0.07s call     tests/infer/reparam/test_unit_jacobian.py::test_normal[(2, 1, 3)]
0.07s call     tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.17-0.33]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-3-7]
0.07s call     tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape4]
0.07s call     tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape5]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-3-1]
0.07s call     tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape5]
0.07s call     tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-3-True]
0.07s call     tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-3-False]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-4-0.37-0.51-0.33]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-3-7]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-3-7]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-3-7]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-3-1]
0.07s call     tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape4]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-1-7]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-4-0.37-0.77-0.33]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-1.7--0.5-1.5]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-4-0.37-0.51-0.33]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-4-0.37-0.51-0.33]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-1.7-0.0-1.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-1.7-0.9-0.5]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-3-0.37-0.51-0.33]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-3-0.37-0.77-0.33]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-3-0.37-0.51-0.33]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-1.7-0.5-0.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S-2.0-0.5-1.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-2.0-0.0-0.5]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-3-0.37-0.77-0.33]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-1.7-0.0-0.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-2.0--0.5-0.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-2.0--0.5-1.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-1.7-0.9-1.5]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-4-0.37-0.77-0.33]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-4-0.37-0.77-0.33]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-3-0.37-0.77-0.33]
0.07s call     tests/infer/reparam/test_unit_jacobian.py::test_normal[(6,)]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-3-0.37-0.51-0.33]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-3-0.37-0.51-0.33]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-4-0.37-0.51-0.33]
0.07s call     tests/distributions/test_stable.py::test_mean[S-2.0-0.5-0.5]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-3-0.37-0.77-0.33]
0.07s call     tests/distributions/test_stable.py::test_mean[S-2.0-0.9-1.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S-2.0-0.9-0.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S-1.7-0.0-1.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S-2.0-0.0-1.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S-1.7-0.0-0.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S-2.0--0.5-0.5]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-1-7]
0.07s call     tests/distributions/test_stable.py::test_mean[S-1.7-0.9-0.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S-1.7-0.5-0.5]
0.07s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-4-0.37-0.77-0.33]
0.07s call     tests/distributions/test_stable.py::test_mean[S-2.0-0.0-0.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S-1.7-0.9-1.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S-2.0--0.5-1.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S-1.7--0.5-1.5]
0.07s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-1-1-()]
0.07s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-1-1-(3, 2)]
0.07s call     tests/distributions/test_stable.py::test_mean[S-1.7--0.5-0.5]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-1.7-0.5-1.5]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-3-7]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-3-1]
0.07s call     tests/distributions/test_stable.py::test_variance[2.0-1.5]
0.07s call     tests/infer/reparam/test_haar.py::test_normal[(4, 2)--2-False]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-3-1]
0.07s call     tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-10]
0.07s call     tests/distributions/test_stable.py::test_variance[2.0-0.5]
0.07s call     tests/infer/test_jit.py::test_svi_enum[2-parallel-parallel]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-3-1]
0.07s call     tests/distributions/test_stable.py::test_mean[S-1.7-0.5-1.5]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-1-7]
0.07s call     tests/infer/reparam/test_haar.py::test_normal[(4, 2)--2-True]
0.07s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-1-1-(4,)]
0.07s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-4]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-3-1]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-3-7]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-3-7]
0.07s call     tests/infer/test_jit.py::test_svi[JitTrace_ELBO-1]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-1-7]
0.07s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_7[1]
0.07s call     tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-False-NUTS-_empty_model]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-3-7]
0.07s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-2-3-(4,)]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.5]
0.07s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-2-3-(4,)]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.75]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.25]
0.07s call     tests/ops/test_contract.py::test_ubersum_jit[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.0]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-1.0]
0.07s call     tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-False-HMC-_empty_model]
0.07s call     tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_guide_module_callable]
0.07s call     tests/distributions/test_stable.py::test_mean[S0-2.0-0.0-1.5]
0.07s call     tests/contrib/cevae/test_cevae.py::test_smoke[laplace-2-1]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-3-1]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-3-7]
0.07s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-4]
0.07s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-()-()]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-1-7]
0.07s call     tests/infer/test_enum.py::test_svi_step_smoke[sequential-single]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-3-7]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-1-7]
0.07s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-(2, 1)-(3,)]
0.07s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-2-3-()]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-3-7]
0.07s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-(2, 1)-(3,)]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.75]
0.07s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-2.0]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.5]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.25]
0.07s call     tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-True-HMC-_empty_model]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-1.0]
0.07s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-3]
0.07s call     tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape1-0.5-0.6-1.1]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.0]
0.07s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-4]
0.07s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-(2, 1)-(3,)]
0.07s call     tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_guide_module_callable]
0.07s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-1.0]
0.07s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-0.5]
0.07s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-0.0]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-3-1]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-3-7]
0.07s call     tests/distributions/test_conjugate.py::test_variance[dist4]
0.07s call     tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape1-0.5-0.6-0.4]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.5]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.0]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.25]
0.07s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-4]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.75]
0.07s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-1-7]
0.07s call     tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_guide_callable]
0.07s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-1.0]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-2-3-(3, 2)]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-1-1]
0.06s call     tests/distributions/test_conjugate.py::test_mean[dist4]
0.06s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_7[10]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-3-1]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-(2, 1)-(3,)]
0.06s call     tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-2-1]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-(2, 1)-(3,)]
0.06s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-4]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-(2, 1)-(3,)]
0.06s call     tests/infer/reparam/test_haar.py::test_normal[(2, 3, 1)--2-True]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-3-7]
0.06s call     tests/infer/reparam/test_haar.py::test_normal[(2, 3, 1)--2-False]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-1-1-(4,)]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-1-1]
0.06s call     tests/infer/test_smcfilter.py::test_smoke[10-1-5-2]
0.06s call     tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_guide_callable]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-(2, 3)-(2, 3)]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-1-1]
0.06s call     tests/nn/test_module.py::test_pyro_serialize
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-(2, 3)-(2, 3)]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-(2, 3)-(2, 3)]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-(2, 3)-(2, 3)]
0.06s call     tests/infer/test_smcfilter.py::test_smoke[10-1-5-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-3-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-3-1]
0.06s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-3-2-()]
0.06s call     tests/infer/test_smcfilter.py::test_smoke[10-1-2-2]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-3-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-1-1]
0.06s call     tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-True-NUTS-_empty_model]
0.06s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-sequential-parallel]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-1-1]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-1-1-(3, 2)]
0.06s call     tests/infer/test_smcfilter.py::test_smoke[10-1-2-1]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-1-1-(3, 2)]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-3-7]
0.06s call     tests/contrib/cevae/test_cevae.py::test_smoke[exponential-1-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-3-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-3-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-3-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-1-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-1-7]
0.06s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-(2, 3)-(2, 3)]
0.06s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-10-False]
0.06s call     tests/infer/reparam/test_conjugate.py::test_beta_binomial_elbo
0.06s call     tests/infer/test_smcfilter.py::test_smoke[10-1-1-1]
0.06s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-1-1-(3, 2)]
0.06s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-(2, 3)-(2, 3)]
0.06s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-(2, 3)-(2, 3)]
0.06s call     tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_guide_list_x]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-3-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-1-1]
0.06s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-parallel-sequential]
0.06s call     tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-10-False]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-1-7]
0.06s call     tests/distributions/test_stable.py::test_mean[S0-2.0-0.5-0.5]
0.06s call     tests/distributions/test_stable.py::test_mean[S0-2.0-0.5-1.5]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-3-7]
0.06s call     tests/distributions/test_stable.py::test_mean[S0-2.0-0.9-1.5]
0.06s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-0.5]
0.06s call     tests/distributions/test_stable.py::test_mean[S0-2.0-0.9-0.5]
0.06s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-2.0]
0.06s call     tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoDelta]
0.06s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-1.0]
0.06s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-0.5]
0.06s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-2.0]
0.06s call     tests/infer/reparam/test_haar.py::test_normal[(6,)--1-False]
0.06s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-1.0]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-3-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-3-7]
0.06s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-0.0]
0.06s call     tests/infer/reparam/test_haar.py::test_normal[(6,)--1-True]
0.06s call     tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-0.0]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-3-7]
0.06s call     tests/infer/test_smcfilter.py::test_smoke[10-1-1-2]
0.06s call     tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_guide_list_x]
0.06s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-(2, 1)-(3,)]
0.06s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-(2, 3)-(2, 3)]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-2-3-()]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-2-3-(4,)]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-1-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-1-7]
0.06s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.5]
0.06s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-3]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-1-7]
0.06s call     tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoDiagonalNormal]
0.06s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.75]
0.06s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-(2, 1)-(3,)]
0.06s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-1.0]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-3-1]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-2-3-()]
0.06s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.25]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-1-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-1-7]
0.06s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.0]
0.06s call     tests/contrib/forecast/test_evaluate.py::test_poisson[svi-7-1-7-1-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-1-7]
0.06s call     tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_class9]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-3-1]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-1-1-(4,)]
0.06s call     tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_class8]
0.06s call     tests/ops/test_welford.py::test_welford_diagonal[1000-7]
0.06s call     tests/ops/test_contract.py::test_ubersum[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-3-1]
0.06s call     tests/infer/test_enum.py::test_elbo_iplate[2-parallel-None]
0.06s call     tests/contrib/forecast/test_evaluate.py::test_simple[cold-7-1-7-1-7]
0.06s call     tests/ops/test_contract.py::test_naive_ubersum[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-3-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-1-7]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-1-1-(4,)]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-3-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-3-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-3-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-3-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-1-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-3-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-1-7]
0.06s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-(2, 3)-(2, 3)]
0.06s call     tests/infer/test_valid_models.py::test_enum_in_model_plate_ok
0.06s call     tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_class9]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-1-1]
0.06s call     tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoDiagonalNormal]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-1-1-(3, 2)]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-1-7]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-1-1-()]
0.06s call     tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_class8]
0.06s call     tests/ops/test_welford.py::test_welford_diagonal[1000-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-1-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-3-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-1-7]
0.06s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-1.0]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-(2, 1)-(3,)]
0.06s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-parallel-sequential-None]
0.06s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-True-sample_shapes1]
0.06s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-2-3-(3, 2)]
0.06s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-4]
0.06s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-parallel-None-None]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-(2, 1)-(3,)]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-(2, 1)-(3,)]
0.06s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-1-1-()]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-(2, 1)-(3,)]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-3-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-3-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-1-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-1-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-3-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-3-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-3-7]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-1-1]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-1-7]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-(2, 3)-(2, 3)]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-1-7]
0.06s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-(2, 1)-(3,)]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-(2, 3)-(2, 3)]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-3-7]
0.06s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-parallel-None-None]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-(2, 3)-(2, 3)]
0.06s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-2-3-GaussianScaleMixture]
0.06s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-1-7]
0.06s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-(2, 3)-(2, 3)]
0.06s call     tests/infer/test_enum.py::test_vectorized_importance[100000]
0.06s call     tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-None-None]
0.06s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-False-sample_shapes1]
0.06s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-(2, 1)-(3,)]
0.06s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-1-7]
0.05s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-parallel-None]
0.05s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-1-7]
0.05s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-sequential-None]
0.05s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-(2, 3)-(2, 3)]
0.05s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-1-1-()]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-1-1]
0.05s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-True-sample_shapes1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-1-1]
0.05s call     tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[parallel-inf]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-3-1]
0.05s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-(2, 3)-(2, 3)]
0.05s call     tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoLaplaceApproximation]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-3-1]
0.05s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-0-2000]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0--1.0--1.0]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_high_obs_dim
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0--1.0--1.0]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0-1.0-1.0]
0.05s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-1-2000]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0--1.0-1.0]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0--1.0-1.0]
0.05s call     tests/ops/test_gaussian.py::test_rsample_distribution[3-(3, 2)]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-3-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-3-7]
0.05s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-1-1-(4,)]
0.05s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-2-3-()]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0-1.0--1.0]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-1-1]
0.05s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-1-1-()]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0-1.0-1.0]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0-1.0--1.0]
0.05s call     tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[20]
0.05s call     tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoLaplaceApproximation]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-3-7]
0.05s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-(2, 1)-(3,)]
0.05s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-2-3-(4,)]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-3-7]
0.05s call     tests/infer/test_discrete.py::test_distribution_3[sample-elbo]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-2]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-1-1]
0.05s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-sequential-None]
0.05s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-sequential-sequential-None]
0.05s call     tests/ops/test_contract.py::test_ubersum_jit[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-1-7]
0.05s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-1-1-(3, 2)]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-3-1]
0.05s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-False-sample_shapes1]
0.05s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-1-1-()]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-3-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-3-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-3-1]
0.05s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-sequential-parallel-None]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-3-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-1-1]
0.05s call     tests/infer/test_enum.py::test_elbo_iplate[2-sequential-sequential]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-3-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-1-1]
0.05s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-()-(3,)]
0.05s call     tests/infer/test_enum.py::test_hmm_enumerate_model[20]
0.05s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-100]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-3-1]
0.05s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-()-(3,)]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-1-7]
0.05s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.5]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-3-1]
0.05s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.0]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-3-7]
0.05s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.25]
0.05s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.75]
0.05s call     tests/distributions/test_transforms.py::TransformTests::test_affine_autoregressive
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-3-7]
0.05s call     tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-1.0]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-1-1]
0.05s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-parallel-sequential-None]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-2]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-3-7]
0.05s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-(3,)-()]
0.05s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-(3,)-()]
0.05s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-100]
0.05s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-(3,)-()]
0.05s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-100]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-1-7]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-3-2]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-1-7]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-3-2]
0.05s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-(3,)-()]
0.05s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-(2, 3)-(2, 3)]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-3-2]
0.05s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-None-None]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-3-2]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-1-7]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-1]
0.05s call     tests/distributions/test_mvt.py::test_log_prob[2-()]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-1-1]
0.05s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-100]
0.05s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-100]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-3-2]
0.05s call     tests/ops/test_contract.py::test_ubersum[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.05s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-parallel-10000]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-1-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-3-2]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-3-2]
0.05s call     tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[1-False]
0.05s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-sequential-None]
0.05s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-parallel-None]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-3-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-3-7]
0.05s call     tests/infer/test_valid_models.py::test_enum_in_model_ok
0.05s call     tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape3]
0.05s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-sequential-parallel-None]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-3-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-1-1]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-4]
0.05s call     tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[parallel-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-1-7]
0.05s call     tests/infer/test_multi_sample_elbos.py::test_inner_outer[RenyiELBO]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-1-1]
0.05s call     tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape3]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-1-7]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-1-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-3-2]
0.05s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-1-1-(4,)]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-1-7]
0.05s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-100]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-3-1]
0.05s call     tests/ops/test_gamma_gaussian.py::test_cat[3-(4, 7, 6)-1-(1, 1, 2, 3)]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-3-2]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-3-2]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-3-2]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-3-2]
0.05s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-(3,)-()]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0-1.0-1.0]
0.05s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-1-1-(4,)]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-3-2]
0.05s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-(3,)-()]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-1-1]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0--1.0--1.0]
0.05s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-(3,)-()]
0.05s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-sequential-parallel-None]
0.05s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-(3,)-()]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0--1.0--1.0]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-3-2]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0-1.0--1.0]
0.05s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-True-sample_shapes1]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0-1.0--1.0]
0.05s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-1-None]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-1-7]
0.05s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-None-None]
0.05s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-0-None]
0.05s call     tests/infer/test_valid_models.py::test_enum_discrete_iplate_single_ok
0.05s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-True-sample_shapes1]
0.05s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-100]
0.05s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-False-sample_shapes1]
0.05s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-True-sample_shapes1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-3-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-3-2]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-3-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-3-1]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0--1.0-1.0]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-3-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-3-1]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0-1.0-1.0]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-1-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-3-2]
0.05s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-3]
0.05s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-1-1-()]
0.05s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-()-(3,)]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0--1.0-1.0]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-3-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-1-7]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-3-1]
0.05s call     tests/ops/test_gamma_gaussian.py::test_cat[3-(4, 7, 6)--1-(2, 1, 3)]
0.05s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-()-(3,)]
0.05s call     tests/ops/test_gamma_gaussian.py::test_cat[3-(4, 7, 6)--2-(1, 1, 2, 3)]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-3-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-3-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-1-7]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-3-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-3-1]
0.05s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-()-(3,)]
0.05s call     tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[None-inf]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-3-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-3-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-1-1]
0.05s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-1-2]
0.05s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-100]
0.05s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-True-sample_shapes1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-3-7]
0.05s call     tests/ops/test_newton.py::test_newton_step[3-None-batch_shape3]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-1-7]
0.05s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-3]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-3-7]
0.05s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-False-sample_shapes1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-1-1]
0.05s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-False-sample_shapes1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-1-1]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-3-1]
0.05s call     tests/infer/test_enum.py::test_backwardsample_posterior_3
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-1-7]
0.05s call     tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape1-0.5-0.6-0.4]
0.05s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-None-None]
0.05s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-2-3-()]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-3-7]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-3-7]
0.05s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-100]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-3-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-1-7]
0.05s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-parallel-parallel-None]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-3-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-3-7]
0.05s call     tests/ops/test_contract.py::test_ubersum_jit[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.05s call     tests/infer/test_valid_models.py::test_enum_discrete_non_enumerated_plate_ok[sequential]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-3-1]
0.05s call     tests/ops/test_contract.py::test_naive_ubersum[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-1-7]
0.05s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-True-sample_shapes1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-3-7]
0.05s call     tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape1-0.5-0.6-1.1]
0.05s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-None-None]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-3-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-3-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-1-7]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-3-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-1-1]
0.05s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-()-(3,)]
0.05s call     tests/infer/test_enum.py::test_elbo_iplate[2-None-parallel]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-3-7]
0.05s call     tests/infer/test_jit.py::test_svi_enum[1-parallel-parallel]
0.05s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-3-2.5]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-1-7]
0.05s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-3-2]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-1-1]
0.05s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-1-7]
0.04s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-3]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-1-1]
0.04s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-2-3-GaussianScaleMixture]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-3-7]
0.04s call     tests/infer/test_smcfilter.py::test_smoke[2-7-5-1]
0.04s call     tests/infer/test_smcfilter.py::test_smoke[2-7-5-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-1-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-3-2]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-3-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-1-7]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-False-sample_shapes1]
0.04s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-nested_auto_guide_callable]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-sequential-None-None]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-3-7]
0.04s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-nested_auto_guide_callable]
0.04s call     tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoDelta]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-3-2]
0.04s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-()-(3,)]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-1-7]
0.04s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_6[1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-3-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-3-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-3-1]
0.04s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-parallel-300]
0.04s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-nested_auto_guide_callable]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-3-7]
0.04s call     tests/infer/test_enum.py::test_elbo_plate[2-sequential-None-None]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-3-1]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-None-sequential-None]
0.04s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-()-(3,)]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-1-1]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-None-sequential-None]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-3-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-3-2]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-3-2]
0.04s call     tests/infer/test_valid_models.py::test_enum_discrete_plates_dependency_ok[None]
0.04s call     tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoDelta]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-1-1]
0.04s call     tests/infer/test_valid_models.py::test_plate_reuse_ok[TraceTMC_ELBO]
0.04s call     tests/infer/test_smcfilter.py::test_smoke[2-7-2-2]
0.04s call     tests/infer/test_valid_models.py::test_enum_discrete_plates_dependency_ok[sequential]
0.04s call     tests/infer/test_smcfilter.py::test_smoke[2-7-2-1]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-False-sample_shapes1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-3-7]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-sequential-sequential-None]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-1-7]
0.04s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-parallel-parallel]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-1-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-3-2]
0.04s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-None-None]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-sequential-None-None]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-3-2]
0.04s call     tests/infer/test_valid_models.py::test_enum_discrete_plates_dependency_ok[parallel]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-1-1]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-sequential-sequential-None]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-3-2]
0.04s call     tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-1-1-()]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-False-sample_shapes1]
0.04s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-4]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-3-1]
0.04s call     tests/infer/test_gradient.py::test_plate[DiffTrace_ELBO-reparam]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-3-2]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-sequential-sequential-None]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-3-1]
0.04s call     tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[None-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-2]
0.04s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-3]
0.04s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-()-(3,)]
0.04s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-subsample-inf]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-3-1]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-sequential-sequential-None]
0.04s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_6[10]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-3-1]
0.04s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-3-1.5]
0.04s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-subsample-inf]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-3-7]
0.04s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-parallel-300]
0.04s call     tests/ops/test_contract.py::test_naive_ubersum[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.04s call     tests/infer/test_gradient.py::test_plate[TraceGraph_ELBO-reparam]
0.04s call     tests/infer/test_discrete.py::test_distribution_2[sample-elbo]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-3-1]
0.04s call     tests/ops/test_contract.py::test_ubersum_jit[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-3-1]
0.04s call     tests/infer/test_gradient.py::test_plate[Trace_ELBO-reparam]
0.04s call     tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-4]
0.04s call     tests/ops/test_gaussian.py::test_cat[3-(4, 7, 6)--1-(2, 1, 3)]
0.04s call     tests/ops/test_gaussian.py::test_cat[3-(4, 7, 6)-1-(1, 1, 2, 3)]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-3-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-1-2]
0.04s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-0-2000]
0.04s call     tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-4]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-3-1]
0.04s call     tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-1]
0.04s call     tests/ops/test_gaussian.py::test_cat[3-(4, 7, 6)--2-(1, 1, 2, 3)]
0.04s call     tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-1-7]
0.04s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-1-2000]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-True-sample_shapes1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-1-7]
0.04s call     tests/infer/test_discrete.py::test_distribution_3[sample]
0.04s call     tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-10]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-1-7]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-None-sequential-None]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-3-7]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-True-sample_shapes1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-3-1]
0.04s call     tests/infer/test_gradient.py::test_plate[Trace_ELBO-nonreparam]
0.04s call     tests/infer/test_valid_models.py::test_iplate_ok[TraceTMC_ELBO-full]
0.04s call     tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-5]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-1-1]
0.04s call     tests/infer/test_gradient.py::test_plate[TraceGraph_ELBO-nonreparam]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-3-7]
0.04s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-sequential-None]
0.04s call     tests/infer/test_jit.py::test_infer_discrete[sample-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-1-1]
0.04s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-3]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-1-1]
0.04s call     tests/ops/test_gaussian.py::test_rsample_distribution[2-(3, 2)]
0.04s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-sequential-None]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-1-1]
0.04s call     tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-5]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-1-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-1-1]
0.04s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-1-7]
0.04s call     tests/infer/test_gradient.py::test_plate[DiffTrace_ELBO-nonreparam]
0.04s call     tests/ops/test_contract.py::test_ubersum[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-3-1]
0.04s call     tests/pyroapi/test_pyroapi.py::test_local_param_ok[pyro-jit]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-3-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-3-1]
0.04s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-subsample-inf]
0.04s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-subsample-inf]
0.04s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-10]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-1-2]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-1-2]
0.04s call     tests/infer/test_valid_models.py::test_plate_reuse_ok[TraceEnum_ELBO]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-1-7]
0.04s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-10]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-3-1]
0.04s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-1-1]
0.04s call     tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[1-True]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-1-7]
0.04s call     tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[4]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-1-7]
0.04s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-3]
0.04s call     tests/distributions/test_mvt.py::test_log_prob[1-()]
0.04s call     tests/infer/test_smcfilter.py::test_smoke[2-7-1-1]
0.04s call     tests/infer/test_smcfilter.py::test_smoke[2-7-1-2]
0.04s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-sequential-inf]
0.04s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-1-7]
0.04s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0--1.0]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-1-1]
0.04s call     tests/infer/test_enum.py::test_vectorized_importance[10000]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-1-2]
0.04s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-True-True]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-3-7]
0.04s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-1]
0.04s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-10]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-1-2]
0.04s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-True-True]
0.04s call     tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-3]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-1-2]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-1-1]
0.04s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-()-(3,)]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-True-sample_shapes1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-3-1]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-False-sample_shapes1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-1-1]
0.04s call     tests/test_examples.py::test_cpu[baseball.py --num-samples=200 --warmup-steps=100 --num-chains=2]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-1-2]
0.04s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-lcmgp-2-2.5]
0.04s call     tests/ops/test_contract.py::test_ubersum_jit[abi,b->,b,ai,abi-i]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-1-1]
0.04s call     tests/test_examples.py::test_cpu[neutra.py -n 10 --num-warmup 10 --num-samples 10]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-1-2]
0.04s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-subsample-0]
0.04s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-subsample-0]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-1-2]
0.04s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-1-1-()]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-1-1]
0.04s call     tests/infer/test_enum.py::test_elbo_iplate[2-sequential-None]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-1-2]
0.04s call     tests/distributions/test_mvt.py::test_rsample[2-9.1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-3-7]
0.04s call     tests/distributions/test_mvt.py::test_rsample[2-3.9]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-False-sample_shapes1]
0.04s call     tests/infer/test_valid_models.py::test_iplate_ok[TraceEnum_ELBO-full]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-3-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-1-1]
0.04s call     tests/infer/test_valid_models.py::test_enum_recycling_grid[True]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-1-1]
0.04s call     tests/test_examples.py::test_jit[baseball.py --num-samples=200 --warmup-steps=100 --jit]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-1-1]
0.04s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-1-2.5]
0.04s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-(3,)-()]
0.04s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-unscaled]
0.04s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-(3,)-()]
0.04s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-(3,)-()]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-1-1]
0.04s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-()-(3,)]
0.04s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-(3,)-()]
0.04s call     tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[pyro-jit]
0.04s call     tests/pyroapi/test_pyroapi.py::test_plate_ok[pyro-jit]
0.04s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-3-2.5]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-1-1]
0.04s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-()-(3,)]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-1-1]
0.04s call     tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-5]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-1-1]
0.04s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-4]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-1-2]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-1-1]
0.04s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-0-None]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-1-1]
0.04s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-1-None]
0.04s call     tests/ops/test_gaussian.py::test_rsample_distribution[3-(4,)]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-3-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-3-7]
0.04s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-4]
0.04s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-None-inf]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-3-7]
0.04s call     tests/ops/test_contract.py::test_ubersum_jit[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-False-sample_shapes1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-1-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-1-7]
0.04s call     tests/infer/test_discrete.py::test_prob[value]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-3-7]
0.04s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-False-True]
0.04s call     tests/infer/test_valid_models.py::test_enum_recycling_grid[False]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-1-1]
0.04s call     tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-None-None]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-3-7]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-parallel-parallel-None]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-3-7]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-sequential-None-None]
0.04s call     tests/distributions/test_von_mises.py::test_von_mises_3d[0.5]
0.04s call     tests/distributions/test_von_mises.py::test_von_mises_3d[0.1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-1-1]
0.04s call     tests/infer/test_enum.py::test_elbo_iplate[1-parallel-None]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-1-2]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-True-sample_shapes1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-1-1]
0.04s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-dmgp-3-1.5]
0.04s call     tests/contrib/tracking/test_assignment.py::test_persistent_smoke[bp]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-1-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-1-1]
0.04s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-3-1]
0.04s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-True-True]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-1-1]
0.04s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-True-True]
0.04s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-subsample-0]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-1-1]
0.04s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-None-None]
0.04s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-(3,)-()]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-1-2]
0.04s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-False-True]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-1-7]
0.04s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-subsample-0]
0.04s call     tests/ops/test_contract.py::test_ubersum[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.04s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-1-7]
0.04s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-()-(3,)]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-1-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-1-2]
0.04s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-lcmgp-3-1.5]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-3-7]
0.04s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-1-1.5]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-1-7]
0.04s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-sequential-inf]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-1-7]
0.04s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-sequential-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-1-1]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-False-sample_shapes1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-3-1]
0.04s call     tests/distributions/test_von_mises.py::test_von_mises_3d[2.0]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-3-7]
0.04s call     tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.02]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-1-1]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-None-None-None]
0.04s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-dmgp-2-1.5]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-1-1]
0.04s call     tests/infer/test_enum.py::test_compute_marginals_hmm[10]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-1-2]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-True-sample_shapes1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-3-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-1-1]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-sequential-None-None]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-3-7]
0.04s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-sequential-1]
0.04s call     tests/distributions/test_von_mises.py::test_von_mises_3d[1.1]
0.04s call     tests/distributions/test_von_mises.py::test_von_mises_3d[10.0]
0.04s call     tests/distributions/test_von_mises.py::test_von_mises_3d[0.9]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-1-2]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-3-1]
0.04s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-3-1.5]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-1-7]
0.04s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-4]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-3-7]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-1-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-1-1]
0.04s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-True-sample_shapes1]
0.04s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-None-sequential-None]
0.04s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-parallel-parallel]
0.04s call     tests/infer/test_jit.py::test_traceenum_elbo[2]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-3-1]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-3-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-1-7]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-1-1]
0.04s call     tests/ops/test_gamma_gaussian.py::test_logsumexp[3-(3, 2)]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-3-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-3-1]
0.04s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-()-(3,)]
0.04s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-1-1]
0.04s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-3-1]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoNormal]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-1-1]
0.03s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-1-1]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-False-sample_shapes1]
0.03s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-1-1]
0.03s call     tests/distributions/test_von_mises.py::test_von_mises_3d[1.0]
0.03s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-2]
0.03s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0--1.0]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoNormal]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-3-7]
0.03s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-1-1]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoNormal]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-3-1]
0.03s call     tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-5]
0.03s call     tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-1-1]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-dmgp-1-1.5]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-True-sample_shapes1]
0.03s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-2]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-1-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-True-unscaled]
0.03s call     tests/infer/test_valid_models.py::test_plate_iplate_ok[TraceTMC_ELBO]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-1-1]
0.03s call     tests/distributions/test_rejector.py::test_exponential_elbo[1.0-0.5]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-3-1]
0.03s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-4]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-3-1]
0.03s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-0-2000]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-1-7]
0.03s call     tests/infer/test_valid_models.py::test_iplate_plate_ok[TraceTMC_ELBO]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-3-7]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_log_prob[1e-08]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-(3,)-()]
0.03s call     tests/infer/test_valid_models.py::test_nested_plate_plate_subsample_param_ok[TraceEnum_ELBO]
0.03s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-1-2000]
0.03s call     tests/distributions/test_transforms.py::TransformTests::test_conditional_neural_autoregressive
0.03s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_7[1]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-None-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-1-7]
0.03s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-2]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-1-1]
0.03s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-3]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-3-7]
0.03s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-4]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-3-1]
0.03s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_2[1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-3-1]
0.03s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-False-True]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-3-7]
0.03s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-False-True]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-1-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-True-scaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-1-7]
0.03s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-2]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-3-1]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-parallel-sequential-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-3-7]
0.03s call     tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-1]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-100]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-True-sample_shapes1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-1-1]
0.03s call     tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape2]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-3-1]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-100]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-1-1]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-100]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-3-7]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoLaplaceApproximation]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-1-7]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-lcmgp-2-2.5]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-100]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-3-1]
0.03s call     tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape2]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-3-7]
0.03s call     tests/ops/test_gaussian.py::test_logsumexp[3-(3, 2)]
0.03s call     tests/test_examples.py::test_cpu[toy_mixture_model_discrete_enumeration.py  --num-steps=1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-3-1]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-(2, 1)-(3,)]
0.03s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-full-0]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-True-sample_shapes1]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-parallel-inf]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-(2, 1)-(3,)]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-100]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_log_prob[1e-06]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-1-1]
0.03s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_in_iplate[9]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-1-1]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-None-parallel-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-3-7]
0.03s call     tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-0]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.0001]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-False-sample_shapes1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-1-7]
0.03s call     tests/ops/test_contract.py::test_einsum_linear[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.03s call     tests/ops/test_contract.py::test_ubersum[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-False-sample_shapes1]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoLaplaceApproximation]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-3-7]
0.03s call     tests/infer/test_jit.py::test_infer_discrete[map-2]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-3-7]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.01]
0.03s call     tests/infer/test_discrete.py::test_prob[grad]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-3-7]
0.03s call     tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-0]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.05]
0.03s call     tests/poutine/test_mapdata.py::test_replay[nested_iplate-20]
0.03s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-full-inf]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-3-7]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-unscaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-1-7]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-sequential-parallel-None]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoLaplaceApproximation]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-sequential-parallel-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-3-7]
0.03s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-full-0]
0.03s call     tests/infer/test_valid_models.py::test_nested_plate_plate_ok[TraceTMC_ELBO]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-1-2.5]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-(2, 1)-(3,)]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-True-unscaled]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-None-parallel-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-1-1]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-(2, 1)-(3,)]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-1-7]
0.03s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-full-inf]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-1-7]
0.03s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-5]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-3-7]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-parallel-sequential-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-3-1]
0.03s call     tests/distributions/test_transforms.py::TransformTests::test_sylvester
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-1-1]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-False-sample_shapes1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-3-7]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-parallel-True-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-1-1]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[None-True-None]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-parallel-False-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-1-7]
0.03s call     tests/infer/test_enum.py::test_elbo_iplate[2-parallel-sequential]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-parallel-True-3]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-ssmgp-2-2.5]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-1-7]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-scaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-1-7]
0.03s call     tests/infer/test_enum.py::test_elbo_plate[2-None-sequential-None]
0.03s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-2-True]
0.03s call     tests/ops/test_contract.py::test_einsum_linear[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-3-1]
0.03s call     tests/ops/test_newton.py::test_newton_step[3-None-batch_shape4]
0.03s call     tests/distributions/test_transforms.py::TransformTests::test_polynomial
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-1-7]
0.03s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-2-False]
0.03s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-4]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-parallel-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-1-1]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[parallel-True-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-1-1]
0.03s call     tests/distributions/test_mvt.py::test_rsample[1-3.9]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-3-1]
0.03s call     tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape4]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-3-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-nonreparam-unscaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-3-1]
0.03s call     tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape4]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[parallel-False-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-3-1]
0.03s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-5]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-3-1]
0.03s call     tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape5]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.10]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-1-1]
0.03s call     tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape5]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-True-sample_shapes1]
0.03s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-5]
0.03s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-parallel-parallel-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-1-1]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.11]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_log_prob[1.0]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-1-7]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-dmgp-3-1.5]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-3-7]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-lcmgp-3-1.5]
0.03s call     tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.2]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-1-1]
0.03s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-True-False]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-1-7]
0.03s call     tests/distributions/test_mvt.py::test_rsample[1-9.1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-3-7]
0.03s call     tests/ops/test_welford.py::test_welford_dense[1000-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-1-7]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-(3,)-()]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-3-7]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-1-1.5]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-(3,)-()]
0.03s call     tests/ops/test_gamma_gaussian.py::test_cat[2-(4, 7, 6)--2-(1, 1, 2, 3)]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-3-1]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-(3,)-()]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-(3,)-()]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-True-sample_shapes1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-3-1]
0.03s call     tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape3]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-True-sample_shapes0]
0.03s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-2]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-3-1]
0.03s call     tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape3]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-ssmgp-3-1.5]
0.03s call     tests/ops/test_gamma_gaussian.py::test_cat[2-(4, 7, 6)--1-(2, 1, 3)]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-3-7]
0.03s call     tests/ops/test_gamma_gaussian.py::test_cat[2-(4, 7, 6)-1-(1, 1, 2, 3)]
0.03s call     tests/infer/test_enum.py::test_elbo_iplate[1-None-parallel]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-3-1]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-dmgp-2-1.5]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-None-None-None]
0.03s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-True-False]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-3-7]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-True-sample_shapes1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-1-7]
0.03s call     tests/ops/test_contract.py::test_naive_ubersum[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-1-1]
0.03s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-0-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-1-7]
0.03s call     tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-1-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-3-7]
0.03s call     tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-3-1]
0.03s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-2-3]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-3-1]
0.03s call     tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-unscaled]
0.03s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-sequential-None]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-None-None]
0.03s call     tests/distributions/test_rejector.py::test_exponential_elbo[1.0-1.0]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-True-sample_shapes1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-1-1]
0.03s call     tests/ops/test_newton.py::test_newton_step[3-None-batch_shape5]
0.03s call     tests/distributions/test_rejector.py::test_exponential_elbo[1.0-2.0]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-3-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-True-scaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-1-1]
0.03s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-sequential-parallel-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-1-1]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-None-inf]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-1-1]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-None-sequential-None]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-False-sample_shapes0]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-3-1]
0.03s call     tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-2-True]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-()-(3,)]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-1-1]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-()-(3,)]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-()-(3,)]
0.03s call     tests/ops/test_contract.py::test_ubersum_jit[,i,i,ia,ia->,i,ia-i]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-3-1]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-()-(3,)]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-None-None-None]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-parallel-parallel-None]
0.03s call     tests/ops/test_contract.py::test_naive_ubersum[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-dmgp-1-1.5]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-(2, 1)-(3,)]
0.03s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_8[iplate-plate-1-1]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-(2, 1)-(3,)]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-(2, 1)-(3,)]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-unscaled]
0.03s call     tests/ops/test_gaussian.py::test_rsample_distribution[2-(4,)]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-True-sample_shapes1]
0.03s call     tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[10]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-False-sample_shapes1]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-(3,)-()]
0.03s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_7[10]
0.03s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-None-None]
0.03s call     tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[parallel-True-3]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-nonreparam-scaled]
0.03s call     tests/infer/test_discrete.py::test_distribution_1[sample]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-False-scaled]
0.03s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-2]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_guide_module_callable]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-3-7]
0.03s call     tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-1-1-()]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-True-sample_shapes1]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-False-sample_shapes1]
0.03s call     tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-2-False]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_guide_module_callable]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-1-1]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-(3,)-()]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-1-7]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-(2, 1)-(3,)]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-1-7]
0.03s call     tests/pyroapi/test_pyroapi.py::test_constraints[pyro-jit]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-(3,)-()]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-1-1]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_guide_module_callable]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-3-7]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-unscaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-1-1]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-(3,)-()]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-3-7]
0.03s call     tests/ops/test_contract.py::test_einsum_linear[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-1-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-unscaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-1-7]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-False-sample_shapes1]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-None-False-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-1-7]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoLowRankMultivariateNormal]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-3-7]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-True-sample_shapes0]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-1-1]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-True-sample_shapes1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-3-1]
0.03s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-sequential-parallel-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-1-1]
0.03s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-parallel-parallel-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-1-1]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-3-7]
0.03s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_6[sequential]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-()-(3,)]
0.03s call     tests/infer/test_discrete.py::test_distribution_2[sample]
0.03s call     tests/ops/test_welford.py::test_welford_dense[1000-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-3-1]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-()-(3,)]
0.03s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-4]
0.03s call     tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[parallel-True-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-3-1]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-()-(3,)]
0.03s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-3]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-1-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-True-unscaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-1-1]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-()-(3,)]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-2-0.5]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-False-scaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-1-1]
0.03s call     tests/ops/test_contract.py::test_ubersum_4[naive_ubersum]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-1-1]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-sequential-sequential-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-3-7]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-sequential-sequential-None]
0.03s call     tests/infer/test_discrete.py::test_distribution_3[map]
0.03s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-2]
0.03s call     tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[parallel-False-None]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-False-sample_shapes1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-False-unscaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-1-7]
0.03s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-None-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-1-7]
0.03s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-False-False]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-1-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-unscaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-3-7]
0.03s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-True-False]
0.03s call     tests/ops/test_special.py::test_safe_log
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-scaled]
0.03s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-False-False]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-3-7]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_guide_list_x]
0.03s call     tests/infer/test_valid_models.py::test_iplate_plate_ok[TraceEnum_ELBO]
0.03s call     tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape2]
0.03s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-4]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-1-7]
0.03s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-True-False]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-True-scaled]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-(3,)-()]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-nonreparam-unscaled]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-False-sample_shapes0]
0.03s call     tests/ops/test_contract.py::test_ubersum_jit[bi,ia->,i,ia,ib,iab-i]
0.03s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-parallel-parallel-None]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-(3,)-()]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-sequential-parallel-None]
0.03s call     tests/ops/test_newton.py::test_newton_step_converges[1-0.1]
0.03s call     tests/infer/test_valid_models.py::test_nested_plate_plate_ok[TraceEnum_ELBO]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-(3,)-()]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-(3,)-()]
0.03s call     tests/infer/test_valid_models.py::test_plate_iplate_ok[TraceEnum_ELBO]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_guide_list_x]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-1-1]
0.03s call     tests/infer/test_valid_models.py::test_vectorized_num_particles[TraceEnum_ELBO]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-3-7]
0.03s call     tests/infer/test_enum.py::test_hmm_enumerate_model[10]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-False-sample_shapes1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-1-1]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-parallel-sequential-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-1-7]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-sequential-parallel-None]
0.03s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-4]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-3-1]
0.03s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-3]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-3-1]
0.03s call     tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape2]
0.03s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-parallel-sequential-None]
0.03s call     tests/ops/test_contract.py::test_naive_ubersum[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.03s call     tests/ops/test_newton.py::test_newton_step[2-None-batch_shape3]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-1-7]
0.03s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_guide_list_x]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-1-1]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-ssmgp-2-2.5]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-1-7]
0.03s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.55-0.33]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-3-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-scaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-3-1]
0.03s call     tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[None-False-None]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-3-1]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-True-sample_shapes1]
0.03s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-2]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-1-1]
0.03s call     tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-parallel-parallel]
0.03s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-4]
0.03s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-True-sample_shapes1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-scaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-3-7]
0.03s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-3]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-1-7]
0.03s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-0-2000]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-1-7]
0.03s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-1-2000]
0.03s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-TraceTMC_ELBO]
0.03s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-5]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-1-1]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-()-(3,)]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-1-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-1-1]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-()-(3,)]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-3-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-3-7]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-3-1]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-()-(3,)]
0.03s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-()-(3,)]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-1-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-True-unscaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-1-1]
0.03s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-3-1]
0.03s call     tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape0-effective_sample_size]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-3-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-nonreparam-scaled]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-1-1]
0.03s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-3-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-True-scaled]
0.03s call     tests/ops/test_newton.py::test_newton_step_converges[3-1.0]
0.03s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-4]
0.03s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-3]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-subsample-reparam]
0.03s call     tests/infer/test_enum.py::test_elbo_iplate[1-None-sequential]
0.03s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-4]
0.03s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-2-3]
0.03s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-2-3]
0.03s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0-0.0]
0.03s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-2-3]
0.03s call     tests/infer/test_smcfilter.py::test_smoke[2-3-5-1]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-True-unscaled]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-True-scaled]
0.03s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-full-inf]
0.03s call     tests/infer/test_smcfilter.py::test_smoke[2-3-5-2]
0.03s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-2-3]
0.03s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-2-3]
0.03s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-parallel-300]
0.03s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-2-3]
0.03s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-True-scaled]
0.03s call     tests/infer/test_enum.py::test_elbo_iplate[2-None-None]
0.03s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-ssmgp-3-1.5]
0.03s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-full-0]
0.03s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-full-0]
0.03s call     tests/infer/test_enum.py::test_elbo_iplate[1-sequential-None]
0.03s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-full-inf]
0.03s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-(3,)-()]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-(3,)-()]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-parallel-False-None]
0.02s call     tests/distributions/test_transforms.py::TransformTests::test_conditional_affine_autoregressive
0.02s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-10]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-2-3]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-2-3]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-False-scaled]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-3]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-True-unscaled]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-None-parallel-None]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-False-unscaled]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-parallel-True-None]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-full-reparam]
0.02s call     tests/ops/test_gaussian.py::test_cat[2-(4, 7, 6)-1-(1, 1, 2, 3)]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-scaled]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0-0.0]
0.02s call     tests/distributions/test_hmm.py::test_discrete_hmm_categorical[9]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-nonreparam-scaled]
0.02s call     tests/ops/test_gaussian.py::test_cat[2-(4, 7, 6)--2-(1, 1, 2, 3)]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-nonreparam-unscaled]
0.02s call     tests/ops/test_gaussian.py::test_cat[2-(4, 7, 6)--1-(2, 1, 3)]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-1-7]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-nonreparam-unscaled]
0.02s call     tests/infer/mcmc/test_valid_models.py::test_enumeration_in_dag[TraceEinsumEvaluator]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-1-7]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-nonreparam-scaled]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-(3,)-()]
0.02s call     tests/infer/test_csis.py::test_csis_validation_batch
0.02s call     tests/infer/test_discrete.py::test_hmm_smoke[sample-10]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[2-3-2-2]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-(3,)-()]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-False-sample_shapes1]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-1-1]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-3-7]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.55-0.33]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-1-7]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-3-7]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-1-7]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-1-7]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-4]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-3-7]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-3-7]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-3-1]
0.02s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-1-0.5]
0.02s call     tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[9]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-1-7]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[2-3-2-1]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-3-1]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-3]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-3-1]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-1-1]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-3-7]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-3-7]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-1-1]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-1-1]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-True-sample_shapes0]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-1-1]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-3-1]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoMultivariateNormal]
0.02s call     tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-None-None]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-False-False]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-1-1]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-False-False]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-3-1]
0.02s call     tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-3-1]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-sequential-None-None]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_median-TraceEnum_ELBO]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-subsample-nonreparam]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-None-None-None]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-True-sample_shapes0]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-sequential-None-None]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_non_enumerated_plate_ok[parallel]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-()-(3,)]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-None-sequential-None]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-()-(3,)]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-False-scaled]
0.02s call     tests/distributions/test_polya_gamma.py::test_polya_gamma[batch_shape1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-2-1]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-()-(3,)]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoMultivariateNormal]
0.02s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-2-0.5]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoMultivariateNormal]
0.02s call     tests/infer/test_autoguide.py::test_factor[AutoNormal-TraceEnum_ELBO]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.39-0.44]
0.02s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-parallel-300]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[1-7-5-2]
0.02s call     tests/distributions/test_rejector.py::test_shape_augmented_beta[0.5-0.5]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-scaled]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[1-7-5-1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-full-nonreparam]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-(2, 3)-(2, 3)]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-False-scaled]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-nonreparam-scaled]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_mean-TraceEnum_ELBO]
0.02s call     tests/ops/test_gamma_gaussian.py::test_logsumexp[3-(4,)]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-True-sample_shapes1]
0.02s call     tests/distributions/test_rejector.py::test_rejector[1.0-2.0]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_sample-TraceEnum_ELBO]
0.02s call     tests/infer/test_enum.py::test_elbo_plate[1-None-sequential-None]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0--0.5]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_feasible-TraceEnum_ELBO]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-(2, 3)-(2, 3)]
0.02s call     tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-None-parallel]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-(2, 3)-(2, 3)]
0.02s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-parallel-300]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-nonreparam-unscaled]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-(2, 3)-(2, 3)]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-(2, 3)-(2, 3)]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-None-inf]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-False-sample_shapes0]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-(2, 3)-(2, 3)]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-None-parallel-None]
0.02s call     tests/infer/mcmc/test_mcmc_api.py::test_mcmc_diagnostics[1]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-(2, 3)-(2, 3)]
0.02s call     tests/distributions/test_spanning_tree.py::test_enumerate_support[5]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-True-scaled]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-3]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[1-7-2-2]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-False-scaled]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-False-sample_shapes0]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[1-7-2-1]
0.02s call     tests/infer/test_enum.py::test_elbo_plate[2-sequential-sequential-None]
0.02s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-glgssm-1-3]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-True-sample_shapes0]
0.02s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-TraceTMC_ELBO]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-False-sample_shapes0]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-False-sample_shapes0]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-False-unscaled]
0.02s call     tests/ops/test_gamma_gaussian.py::test_logsumexp[2-(3, 2)]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-(2, 1)-(3,)]
0.02s call     tests/infer/test_multi_sample_elbos.py::test_outer_inner[RenyiELBO]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.55-0.33]
0.02s call     tests/ops/test_integrator.py::test_time_reversibility[HarmonicOscillator]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-(2, 3)-(2, 3)]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-None-False-None]
0.02s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-2]
0.02s call     tests/infer/test_valid_models.py::test_iplate_ok[TraceTMC_ELBO-subsample]
0.02s call     tests/infer/test_autoguide.py::test_factor[AutoLowRankMultivariateNormal-TraceEnum_ELBO]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-False-unscaled]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[2-3-1-2]
0.02s call     tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-parallel-None]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[2-3-1-1]
0.02s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-0-None]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-3]
0.02s call     tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-sequential-1-False]
0.02s call     tests/infer/test_enum.py::test_compute_marginals_restrictions[False-sequential-1-False]
0.02s call     tests/infer/test_enum.py::test_elbo_plate[2-None-parallel-None]
0.02s call     tests/distributions/test_mvt.py::test_mean_var[()]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-2-1]
0.02s call     tests/nn/test_module.py::test_to_pyro_module_
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-2-1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-2-1]
0.02s call     tests/distributions/test_spanning_tree.py::test_partition_function[5]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-2-1]
0.02s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-1-None]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-2-1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-2-1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-2-3]
0.02s call     tests/ops/test_gaussian.py::test_logsumexp[3-(4,)]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-4]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.55-0.33]
0.02s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-parallel-None]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-2-3]
0.02s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-3]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-2-3]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-2-3]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-False-unscaled]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-2-3]
0.02s call     tests/distributions/test_rejector.py::test_rejector[1.0-1.0]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-3-3]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-2-3]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_non_enumerated_plate_ok[None]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-2-1]
0.02s call     tests/infer/reparam/test_transform.py::test_log_normal[()]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-()-(3,)]
0.02s call     tests/infer/test_valid_models.py::test_nonnested_plate_plate_ok[TraceEnum_ELBO]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-nonreparam-unscaled]
0.02s call     tests/infer/test_enum.py::test_elbo_plate[1-parallel-None-None]
0.02s call     tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-None-None]
0.02s call     tests/distributions/test_transforms.py::TransformTests::test_conditional_householder
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-2-1]
0.02s call     tests/pyroapi/test_pyroapi.py::test_constraints[minipyro-jit]
0.02s call     tests/distributions/test_polya_gamma.py::test_polya_gamma[batch_shape0]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-3-2]
0.02s call     tests/infer/test_autoguide.py::test_factor[AutoMultivariateNormal-TraceEnum_ELBO]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-nonreparam-scaled]
0.02s call     tests/ops/test_contract.py::test_einsum_linear[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.02s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-4]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-True-sample_shapes1]
0.02s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-None-None]
0.02s call     tests/distributions/test_rejector.py::test_rejector[1.0-0.5]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-True-sample_shapes1]
0.02s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-parallel-2000]
0.02s call     tests/infer/test_valid_models.py::test_dim_allocation_ok[Trace_ELBO-False]
0.02s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-parallel-300]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-2]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.55-0.43]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-False-sample_shapes0]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_3[batch-10]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-2-3]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-False-sample_shapes1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-False-unscaled]
0.02s call     tests/infer/test_enum.py::test_elbo_plate[2-parallel-None-None]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.55-0.44]
0.02s call     tests/distributions/test_hmm.py::test_discrete_hmm_categorical[8]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-3-3]
0.02s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-3]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-True-True]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-2-2]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-False-unscaled]
0.02s call     tests/ops/test_contract.py::test_einsum_linear[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-True-sample_shapes1]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-2-3]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-3-2]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0-0.5]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-nonreparam-unscaled]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-3-1]
0.02s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-TraceTMC_ELBO]
0.02s call     tests/ops/test_gaussian.py::test_logsumexp[2-(3, 2)]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-(2, 3)-(2, 3)]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0-1.0]
0.02s call     tests/ops/test_contract.py::test_einsum_linear[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.02s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-parallel-None]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-3]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-4]
0.02s call     tests/distributions/test_transforms.py::TransformTests::test_conditional_planar
0.02s call     tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[8]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-1-3]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-2-2]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-scaled]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-1-2]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-3-1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-True-unscaled]
0.02s call     tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-None-parallel]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-2-1]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-True-True]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-3-3]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-(2, 1)-(3,)]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-nonreparam-scaled]
0.02s call     tests/ops/test_contract.py::test_ubersum[,i,i,ia,ia->,i,ia-i]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-True-scaled]
0.02s call     tests/pyroapi/test_pyroapi.py::test_plate_ok[minipyro-jit]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_3[masked-10]
0.02s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-TraceTMC_ELBO]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[inf]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoDiagonalNormal]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-None-1]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-(2, 1)-(3,)]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-2-1]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-True-True]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-True-True]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-1-3]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-1-2]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-(2, 1)-(3,)]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[1-7-1-2]
0.02s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-1-0.5]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[1-7-1-1]
0.02s call     tests/infer/mcmc/test_valid_models.py::test_potential_fn_pickling[True]
0.02s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-10]
0.02s call     tests/infer/test_autoguide.py::test_factor[AutoIAFNormal-TraceEnum_ELBO]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-3-2]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-True-unscaled]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.55-0.33]
0.02s call     tests/infer/test_enum.py::test_elbo_iplate[1-sequential-sequential]
0.02s call     tests/infer/test_enum.py::test_elbo_plate[1-sequential-sequential-None]
0.02s call     tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[minipyro-jit]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.55-0.44]
0.02s call     tests/contrib/test_minipyro.py::test_local_param_ok[pyro-jit]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-True-sample_shapes0]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_3[masked-1]
0.02s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoDiagonalNormal]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-3-3]
0.02s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-TraceTMC_ELBO]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.55-0.43]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class11]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-(2, 1)-(3,)]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.27-0.43]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class11]
0.02s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-3]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-2]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoDiagonalNormal]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.27-0.33]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.27-0.33]
0.02s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-3]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-True-sample_shapes0]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.39-0.33]
0.02s call     tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-parallel-parallel]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-unscaled]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-False-sample_shapes1]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-False-sample_shapes1]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-3-2]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-2-1]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class11]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-2-1]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-19]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-True-scaled]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-2-1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-2-1]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-1-1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-True-unscaled]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-2-1]
0.02s call     tests/infer/test_enum.py::test_elbo_plate[2-parallel-parallel-10000]
0.02s call     tests/infer/test_discrete.py::test_distribution_2[map]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-2-1]
0.02s call     tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-parallel-None]
0.02s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-None-None]
0.02s call     tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-3-True]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-True-sample_shapes0]
0.02s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-3]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class12]
0.02s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-10]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class10]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-False-scaled]
0.02s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-10]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class10]
0.02s call     tests/infer/test_enum.py::test_differentiable_loss[parallel-single]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-unscaled]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_1[1]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class12]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class9]
0.02s call     tests/ops/test_newton.py::test_newton_step[2-None-batch_shape4]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class9]
0.02s call     tests/distributions/test_stable.py::test_sample_2[0.999999--1.0]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_3[batch-1]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-4]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class10]
0.02s call     tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape0]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-1-1]
0.02s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-10]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-2-3]
0.02s call     tests/ops/test_newton.py::test_newton_step[2-None-batch_shape5]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-()-()]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-18]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class12]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-()-()]
0.02s call     tests/distributions/test_rejector.py::test_shape_augmented_beta[0.5-1.0]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-3-3]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class9]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-()-()]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-False-sample_shapes0]
0.02s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-()-()]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-3-2]
0.02s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-parallel-inf]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-scaled]
0.02s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceTMC_ELBO-subsample]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_1[10]
0.02s call     tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape0]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-2-2]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-True-sample_shapes0]
0.02s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.02s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceTMC_ELBO-full]
0.02s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.02s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.02s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.02s call     tests/infer/test_valid_models.py::test_iplate_ok[TraceEnum_ELBO-subsample]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-True-sample_shapes0]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-3-1]
0.02s call     tests/distributions/test_rejector.py::test_shape_augmented_beta[0.5-4.0]
0.02s call     tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.02s call     tests/distributions/test_rejector.py::test_shape_augmented_beta[4.0-0.5]
0.02s call     tests/distributions/test_hmm.py::test_discrete_hmm_categorical[7]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-2-3]
0.02s call     tests/infer/test_enum.py::test_elbo_plate[2-None-None-None]
0.02s call     tests/ops/test_gamma_gaussian.py::test_cat[1-(4, 7, 6)--1-(2, 1, 3)]
0.02s call     tests/ops/test_gamma_gaussian.py::test_cat[1-(4, 7, 6)--2-(1, 1, 2, 3)]
0.02s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceTMC_ELBO-subsample]
0.02s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceTMC_ELBO-full]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1--1.0]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-17]
0.02s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-3-2-7]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-4]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.27-0.43]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_in_iplate[5]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-2-2]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-None-parallel-None]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.55-0.44]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-(2, 1)-(3,)]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.39-0.44]
0.02s call     tests/ops/test_contract.py::test_ubersum_jit[,i,ia->,i,ia-i]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-1-3]
0.02s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-TraceTMC_ELBO]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-3-1]
0.02s call     tests/ops/test_gamma_gaussian.py::test_cat[1-(4, 7, 6)-1-(1, 1, 2, 3)]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-1-3]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-True-sample_shapes0]
0.02s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceTMC_ELBO-full]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.39-0.33]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-1-3]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-1-3]
0.02s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-10]
0.02s call     tests/infer/test_discrete.py::test_hmm_smoke[map-10]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-1-3]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.39-0.44]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-1-3]
0.02s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceTMC_ELBO-subsample]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-(2, 1)-(3,)]
0.02s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-3-False]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.39-0.33]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-3]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-1-3]
0.02s call     tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-3-False]
0.02s call     tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[7]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-3]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-1-3]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[2]
0.02s call     tests/distributions/test_transforms.py::TransformTests::test_planar
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-8]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-(2, 1)-(3,)]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-True-sample_shapes0]
0.02s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-TraceEnum_ELBO]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-True-sample_shapes0]
0.02s call     tests/ops/test_contract.py::test_adjoint_marginal[a,abi,bcij,bdik->-ijk]
0.02s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceTMC_ELBO-subsample]
0.02s call     tests/infer/test_enum.py::test_elbo_plate[1-sequential-None-None]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-8]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99-0.5]
0.02s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-10]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-parallel-1]
0.02s call     tests/infer/test_valid_models.py::test_enum_in_model_plate_reuse_ok
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-2-3]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-2]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[0]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[1]
0.02s call     tests/infer/test_valid_models.py::test_dim_allocation_ok[TraceGraph_ELBO-False]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-2-1]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-False-sample_shapes0]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99--0.5]
0.02s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-10]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-()-()]
0.02s call     tests/infer/test_autoguide.py::test_factor[AutoDiagonalNormal-TraceEnum_ELBO]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-2-2]
0.02s call     tests/distributions/test_stable.py::test_sample_2[0.999999--0.5]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-3-1]
0.02s call     tests/distributions/test_stable.py::test_sample_2[1.000001--0.5]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-8]
0.02s call     tests/distributions/test_stable.py::test_sample_2[1.000001--1.0]
0.02s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceTMC_ELBO-full]
0.02s call     tests/distributions/test_stable.py::test_sample_2[0.999999-1.0]
0.02s call     tests/distributions/test_stable.py::test_sample_2[1.000001-0.5]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-None-parallel-None]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-3-3]
0.02s call     tests/poutine/test_mapdata.py::test_cond_indep_stack[nested_iplate-20]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-False-True]
0.02s call     tests/distributions/test_stable.py::test_sample_2[1.000001-1.0]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-2-1]
0.02s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-2]
0.02s call     tests/distributions/test_stable.py::test_sample_2[0.999999-0.5]
0.02s call     tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-10]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-8]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-8]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-8]
0.02s call     tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[pyro-jit]
0.02s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-glgssm-1-3]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-unscaled]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-3-3]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-8]
0.02s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceTMC_ELBO-subsample]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-(2, 3)-(2, 3)]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-False-True]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-False-sample_shapes0]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_median-TraceEnum_ELBO]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-False-sample_shapes0]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_5[1]
0.02s call     tests/contrib/test_minipyro.py::test_plate_ok[pyro-jit]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-2-1]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-False-True]
0.02s call     tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape1-effective_sample_size]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-(2, 3)-(2, 3)]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-3-2]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-False-sample_shapes0]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-False-sample_shapes0]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-parallel-None-None]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-parallel-None-None]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-16]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-8]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-8]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-8]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_2[masked-10]
0.02s call     tests/infer/test_valid_models.py::test_plate_no_size_ok[TraceTMC_ELBO]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-False-sample_shapes0]
0.02s call     tests/distributions/test_hmm.py::test_discrete_hmm_categorical[6]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-1-1]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-2]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_median-TraceEnum_ELBO]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-1-1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-scaled]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-None-inf]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-3-2]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-3-3]
0.02s call     tests/ops/test_contract.py::test_naive_ubersum[,i,i,ia,ia->,i,ia-i]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-True-1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-1-3]
0.02s call     tests/distributions/test_stable.py::test_sample_2[0.999999-0.0]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-1-1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-1-1]
0.02s call     tests/distributions/test_stable.py::test_sample_2[1.000001-0.0]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-1-1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-1-3]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-3-2]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_2[masked-1]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-False-sample_shapes0]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-1-1]
0.02s call     tests/infer/reparam/test_neutra.py::test_reparam_log_joint[dirichlet_categorical-kwargs1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-1-3]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-1-3]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-1-3]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-1-3]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-False-True]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-1]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-2-1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-True-scaled]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-8]
0.02s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceTMC_ELBO-full]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-15]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-1-1]
0.02s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-3-True]
0.02s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-3]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_feasible-TraceEnum_ELBO]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_2[batch-10]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-3-3]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1-1.0]
0.02s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-0-2000]
0.02s call     tests/infer/test_autoguide.py::test_factor[AutoLaplaceApproximation-TraceEnum_ELBO]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-(2, 3)-(2, 3)]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_mean-TraceEnum_ELBO]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_2[batch-1]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-2-3]
0.02s call     tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[6]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-1-1]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_5[10]
0.02s call     tests/ops/test_contract.py::test_ubersum_5[naive_ubersum]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-(2, 3)-(2, 3)]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-False-unscaled]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8-0.0]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-3-2]
0.02s call     tests/infer/test_enum.py::test_elbo_iplate[2-sequential-parallel]
0.02s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-dense-3]
0.02s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-1-2000]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_sample-TraceEnum_ELBO]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1-0.5]
0.02s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-sequential-None]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4-0.0]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8-0.5]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1--0.5]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_feasible-TraceEnum_ELBO]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-sequential-None]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-8]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3--1.0]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-True-scaled]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8--0.5]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-2-2]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_mean-TraceEnum_ELBO]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0-0.5]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-2]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_sample-TraceEnum_ELBO]
0.02s call     tests/infer/test_enum.py::test_differentiable_loss[sequential-batch]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7--0.5]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_median-TraceEnum_ELBO]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_mean-TraceEnum_ELBO]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4--1.0]
0.02s call     tests/ops/test_contract.py::test_einsum_linear[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3--0.5]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7--1.0]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-3-1]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_single_single_ok
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-19]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-2-3]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7-0.5]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-None-1]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_4[1]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0--0.5]
0.02s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-flat-3]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_feasible-TraceEnum_ELBO]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-True-unscaled]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-3]
0.02s call     tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-2-2]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-2]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_6[parallel]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-nonreparam-unscaled]
0.02s call     tests/distributions/test_stable.py::test_normal[0.5-0]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-2-3]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-(2, 3)-(2, 3)]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-19]
0.02s call     tests/ops/test_tensor_utils.py::test_dct_dim[dct--4]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_sample-TraceEnum_ELBO]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-3-1]
0.02s call     tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-2-1]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-2]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-(2, 3)-(2, 3)]
0.02s call     tests/infer/test_valid_models.py::test_plate_ok[TraceTMC_ELBO-subsample]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-2-2]
0.02s call     tests/infer/test_jit.py::test_traceenum_elbo[1]
0.02s call     tests/infer/test_predictive.py::test_shapes[False]
0.02s call     tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceTMC_ELBO-subsample]
0.02s call     tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-2-2]
0.02s call     tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-2-1]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-14]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-nonreparam-unscaled]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-3]
0.02s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-sequential-None]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-2-2]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-scaled]
0.02s call     tests/infer/test_autoguide.py::test_factor[AutoDelta-TraceEnum_ELBO]
0.02s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-subsample-inf]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoDelta]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-18]
0.02s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-subsample-inf]
0.02s call     tests/distributions/test_stable.py::test_sample_2[0.99--1.0]
0.02s call     tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceTMC_ELBO-full]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-None-None-None]
0.02s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-subsample-0]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-sequential-parallel-None]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-sequential-None]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-3-0.37-0.51-0.33]
0.02s call     tests/infer/test_valid_models.py::test_plate_ok[TraceTMC_ELBO-full]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-False-initial_shape1-proposed_shape1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-False-scaled]
0.02s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-2]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-3-1]
0.02s call     tests/distributions/test_rejector.py::test_shape_augmented_beta[1.0-0.5]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-13]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-3]
0.02s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-subsample-0]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-sequential-parallel-None]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-4-0.37-0.51-0.33]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-4-0.37-0.51-0.33]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-scaled]
0.02s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-glgssm-3-1]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-2-3]
0.02s call     tests/distributions/test_rejector.py::test_shape_augmented_beta[4.0-1.0]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-2]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-3-1]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-2-2]
0.02s call     tests/distributions/test_rejector.py::test_shape_augmented_beta[4.0-4.0]
0.02s call     tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[5]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_4[10]
0.02s call     tests/infer/test_enum.py::test_svi_step_guide_uses_grad[sequential]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-1-1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-1-1]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-18]
0.02s call     tests/distributions/test_rejector.py::test_shape_augmented_beta[1.0-1.0]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-1-1]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_median-TraceEnum_ELBO]
0.02s call     tests/distributions/test_rejector.py::test_shape_augmented_beta[1.0-4.0]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-False-initial_shape1-proposed_shape1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-1-1]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-1-1]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-3-0.37-0.51-0.33]
0.02s call     tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-1-1]
0.02s call     tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-parallel-None]
0.02s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-TraceEnum_ELBO]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-scaled]
0.02s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-TraceEnum_ELBO]
0.02s call     tests/infer/test_valid_models.py::test_three_indep_plate_at_different_depths_ok
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-3]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(3, 2)-(5,)]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-parallel-10000]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-False-initial_shape0-proposed_shape0]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-19]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-(2, 1)-(3,)]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-True-sample_shapes0]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-(2, 1)-(3,)]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-(2, 1)-(3,)]
0.02s call     tests/ops/test_contract.py::test_einsum_linear[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-2-1]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-True-sample_shapes0]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-4-0.37-0.77-0.33]
0.02s call     tests/infer/test_jit.py::test_infer_discrete[sample-1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-True-scaled]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-3-0.37-0.51-0.33]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-parallel-parallel-None]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.39-0.44]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-3-0.37-0.77-0.33]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-2-1]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-17]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-True-unscaled]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-(2, 1)-(3,)]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-17]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-12]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-3-0.37-0.51-0.33]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(3, 2)-(5,)]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-False-unscaled]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-3-2]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-2]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-4-0.37-0.77-0.33]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-7]
0.02s call     tests/ops/test_contract.py::test_einsum_linear[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8--1.0]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99-1.0]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-3-0.37-0.77-0.33]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-4-0.37-0.51-0.33]
0.02s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-2]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1--1.0]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-subsample-reparam]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1-0.0]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99-0.0]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-7]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-3-2]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-3-2]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-4-0.37-0.77-0.33]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8-1.0]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99--1.0]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-3-0.37-0.77-0.33]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-2-1]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4-0.5]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7-1.0]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-4-0.37-0.77-0.33]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-4-0.37-0.51-0.33]
0.02s call     tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-3-0.37-0.77-0.33]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-3-2]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoDelta]
0.02s call     tests/ops/test_contract.py::test_einsum_linear[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4-1.0]
0.02s call     tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoDelta]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-False-initial_shape0-proposed_shape0]
0.02s call     tests/infer/test_autoguide.py::test_discrete_parallel[AutoLowRankMultivariateNormal]
0.02s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-4]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-3-1]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4--0.5]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-4]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-3-2]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01-0.0]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-3-2]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-7]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.55-0.33]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-3-1]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-1-3]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-3-1]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(3, 2)-(5,)]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.39-0.33]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-7]
0.02s call     tests/distributions/test_hmm.py::test_discrete_hmm_categorical[5]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-3-2]
0.02s call     tests/infer/test_enum.py::test_differentiable_loss[None-single]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-7]
0.02s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-tuple-sample_shape2]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01--1.0]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.39-0.44]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-parallel-parallel-None]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-3-2]
0.02s call     tests/infer/test_discrete.py::test_distribution_masked[sample-elbo]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-False-sample_shapes0]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3-0.5]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-3-1]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.55-0.44]
0.02s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-3-2]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3-1.0]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-2-1]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.55-0.44]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01-1.0]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_median-TraceEnum_ELBO]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01-0.5]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01--0.5]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-3-1]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-3-1]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_mean-TraceEnum_ELBO]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-True-sample_shapes0]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0-0.0]
0.02s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-1]
0.02s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-3-2]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_feasible-TraceEnum_ELBO]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-1-2]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-3-1]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7-0.0]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-3-1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-nonreparam-scaled]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3-0.0]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-nonreparam-unscaled]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-7]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-3]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.55-0.33]
0.02s call     tests/infer/test_enum.py::test_elbo_plate[1-None-parallel-None]
0.02s call     tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-parallel-None]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-7]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.55-0.44]
0.02s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-(2, 3)-(2, 3)]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-3-2]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.39-0.33]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(3, 2)-(5,)]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.39-0.44]
0.02s call     tests/ops/test_contract.py::test_einsum_linear[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-18]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(3, 2)-(5,)]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-False-initial_shape0-proposed_shape0]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-7]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-False-initial_shape1-proposed_shape1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-subsample-nonreparam]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-1-3]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-full-reparam]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-11]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-subsample-reparam]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_sample-TraceEnum_ELBO]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.55-0.33]
0.02s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-list-sample_shape2]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[Binomial-False-initial_shape1-proposed_shape1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-True-unscaled]
0.02s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-tuple-sample_shape2]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[Binomial-False-initial_shape0-proposed_shape0]
0.02s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-Size-sample_shape2]
0.02s call     tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape3]
0.02s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-Size-sample_shape2]
0.02s call     tests/ops/test_linalg.py::test_sym_rinverse[True-A5]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-False-scaled]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-()-()]
0.02s call     tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape3]
0.02s call     tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-2-True]
0.02s call     tests/infer/test_discrete.py::test_distribution_1[sample-elbo]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-1-2]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-3-1]
0.02s call     tests/infer/test_enum.py::test_elbo_iplate[1-None-None]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-False-initial_shape1-proposed_shape1]
0.02s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-tuple-sample_shape2]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-()-()]
0.02s call     tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[5]
0.02s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-Size-sample_shape2]
0.02s call     tests/ops/test_gaussian.py::test_rsample_distribution[1-(3, 2)]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-False-unscaled]
0.02s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-list-sample_shape2]
0.02s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-True-10]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-7]
0.02s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-list-sample_shape2]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-7]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-7]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-unscaled]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-17]
0.02s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-parallel-10000]
0.02s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-3]
0.02s call     tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTraceMeanField_ELBO]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.55-0.33]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-full-nonreparam]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(3, 2)-(5,)]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[Geometric-False-initial_shape0-proposed_shape0]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[Geometric-False-initial_shape1-proposed_shape1]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.55-0.44]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-False-initial_shape0-proposed_shape0]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[Bernoulli-False-initial_shape1-proposed_shape1]
0.02s call     tests/distributions/test_stable.py::test_sample_2[0.99--0.5]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-subsample-reparam]
0.02s call     tests/distributions/test_stable.py::test_sample_2[0.99-0.5]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-True-sample_shapes0]
0.02s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-0-None]
0.02s call     tests/distributions/test_stable.py::test_sample_2[1.01--0.5]
0.02s call     tests/distributions/test_stable.py::test_sample_2[1.01--1.0]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-full-nonreparam]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-unscaled]
0.02s call     tests/distributions/test_stable.py::test_sample_2[1.01-0.5]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-nonreparam-scaled]
0.02s call     tests/distributions/test_stable.py::test_sample_2[1.01-1.0]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.39-0.33]
0.02s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-7]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-subsample-nonreparam]
0.02s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-4]
0.02s call     tests/infer/test_discrete.py::test_distribution_1[map]
0.02s call     tests/distributions/test_stable.py::test_sample_2[0.99-1.0]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(3, 2)-(5,)]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.39-0.44]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-10]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-3-3]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-full-reparam]
0.02s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0--0.5]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-3-2]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_feasible-TraceEnum_ELBO]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-1-2]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_mean-TraceEnum_ELBO]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-1-2]
0.02s call     tests/distributions/test_extended.py::test_extended_beta_binomial[0.02]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-False-initial_shape2-proposed_shape2]
0.02s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-1-None]
0.02s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-TraceEnum_ELBO]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-1-2]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-1-2]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[Bernoulli-False-initial_shape0-proposed_shape0]
0.02s call     tests/ops/test_contract.py::test_ubersum[,i,ia->,i,ia-i]
0.02s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-TraceEnum_ELBO]
0.02s call     tests/infer/test_valid_models.py::test_iplate_ok[Trace_ELBO-full]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[Poisson-False-initial_shape0-proposed_shape0]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-False-initial_shape2-proposed_shape2]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-19]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-1-2]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-nonreparam-scaled]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-1-1]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-1-2]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-1-1]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-True-sample_shapes0]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-1-1]
0.02s call     tests/infer/test_autoguide.py::test_discrete_parallel[AutoMultivariateNormal]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-1-2]
0.02s call     tests/ops/test_linalg.py::test_sym_rinverse[False-A5]
0.02s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.39-0.33]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-1-1]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-1-2]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[2-1-5-2]
0.02s call     tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-4]
0.02s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-False-sample_shapes0]
0.02s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_sample-TraceEnum_ELBO]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-1-1]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-full-reparam]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-()-()]
0.02s call     tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceTMC_ELBO-True]
0.02s call     tests/infer/test_autoguide.py::test_discrete_parallel[AutoNormal]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(3, 2)-(5,)]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-1-1]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[2-1-5-1]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-16]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-1-1]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-1-1]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-()-()]
0.02s call     tests/ops/test_newton.py::test_newton_step[1-None-batch_shape3]
0.02s call     tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[2]
0.02s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-2-1-7]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-9]
0.02s call     tests/ops/test_contract.py::test_ubersum_jit[ab->,a,b,ab,ba-]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-1-2]
0.02s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-False-scaled]
0.02s call     tests/distributions/test_polya_gamma.py::test_polya_gamma[batch_shape2]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-2-1]
0.02s call     tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceEnum_ELBO-sequential]
0.02s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-3-2-7]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-()-()]
0.02s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-()-()]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-3-2]
0.02s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[11]
0.02s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceEnum_ELBO-full]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[2-1-2-2]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-1-1]
0.02s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-3-1]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-16]
0.02s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-1-2]
0.02s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-Size-sample_shape2]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-2-3]
0.02s call     tests/infer/test_valid_models.py::test_not_has_rsample_ok[TraceTMC_ELBO]
0.02s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-tuple-sample_shape2]
0.02s call     tests/distributions/test_stable.py::test_sample_2[1.01-0.0]
0.02s call     tests/infer/test_smcfilter.py::test_smoke[2-1-2-1]
0.02s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-18]
0.02s call     tests/distributions/test_distributions.py::test_expand_error[Poisson-False-initial_shape1-proposed_shape1]
0.02s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-list-sample_shape2]
0.02s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-2-2]
0.02s call     tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-19]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-3-2]
0.01s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-list-sample_shape2]
0.01s call     tests/infer/test_enum.py::test_elbo_plate[1-None-None-None]
0.01s call     tests/ops/test_gamma_gaussian.py::test_logsumexp[2-(4,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-15]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-Size-sample_shape2]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.99-0.0]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-True-sample_shapes0]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_median-TraceEnum_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-3-2]
0.01s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-dense-3]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-3-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-3-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-16]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-3-1]
0.01s call     tests/infer/test_enum.py::test_multi_dependence_enumeration
0.01s call     tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-tuple-sample_shape2]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_1[batch-1]
0.01s call     tests/infer/test_enum.py::test_hmm_enumerate_model[5]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-list-sample_shape2]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_4[False-True-1]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-3-3]
0.01s call     tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-glgssm-3-1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-3-2]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-3-lgssmgp]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(4,)-(5,)]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_2[10]
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceTMC_ELBO-subsample]
0.01s call     tests/optim/test_optim.py::test_dynamic_lr[scheduler1]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceEnum_ELBO-subsample]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-15]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-2]
0.01s call     tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceTMC_ELBO-False]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-True-sample_shapes0]
0.01s call     tests/ops/test_contract.py::test_einsum_linear[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.01s call     tests/optim/test_optim.py::test_dynamic_lr[scheduler0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-3-2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-3-1]
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceEnum_ELBO-subsample]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-1-2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-17]
0.01s call     tests/infer/test_csis.py::test_csis_parameter_update
0.01s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-None-None-None]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceEnum_ELBO-subsample]
0.01s call     tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-4]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(4,)-(5,)]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_8[plate-iplate-1-1]
0.01s call     tests/distributions/test_transforms.py::TransformTests::test_haar_transform
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-3-1]
0.01s call     tests/infer/test_enum.py::test_svi_step_smoke[parallel-single]
0.01s call     tests/optim/test_optim.py::test_dynamic_lr[scheduler2]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceEnum_ELBO-subsample]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_1[batch-10]
0.01s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(4,)-(5,)]
0.01s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_reparameterized_N_is_17
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceEnum_ELBO-subsample]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-1-1]
0.01s call     tests/contrib/test_minipyro.py::test_constraints[pyro-jit]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-4]
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceTMC_ELBO-full]
0.01s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.55-0.33]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_1[masked-1]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_1[masked-10]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-2-lgssmgp]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-1-2]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[2-1-1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-18]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Binomial-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-14]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-15]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_5
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Bernoulli-False-initial_shape2-proposed_shape2]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[2-1-1-2]
0.01s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-0.1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-1-2]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-6]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceEnum_ELBO-full]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-False-sample_shapes0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(3, 2)-()]
0.01s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-flat-3]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-1-1]
0.01s call     tests/distributions/test_spanning_tree.py::test_log_prob[4]
0.01s call     tests/optim/test_optim.py::test_dynamic_lr[scheduler3]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-2]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-()-()]
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceEnum_ELBO-full]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-1-1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-()-()]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-6]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceEnum_ELBO-full]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-1-1]
0.01s call     tests/infer/test_valid_models.py::test_plate_ok[TraceEnum_ELBO-subsample]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-1-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(4,)-(5,)]
0.01s call     tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-parallel-parallel]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceEnum_ELBO-full]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-1]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-6]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-()-()]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-3-2-1]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceEnum_ELBO-full]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-()-()]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_4[False-True-10]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-14]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-6]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0-0.0]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-()-()]
0.01s call     tests/infer/test_valid_models.py::test_plate_enum_discrete_batch_ok
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-1-1]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[idct--4]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-Size-sample_shape1]
0.01s call     tests/ops/test_gaussian.py::test_logsumexp[2-(4,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(3, 2)-()]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_3[single-10]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(4,)-(5,)]
0.01s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-1-3]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0-0.5]
0.01s call     tests/infer/test_valid_models.py::test_plate_reuse_ok[Trace_ELBO]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoNormal-TraceGraph_ELBO]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-3-5-2]
0.01s call     tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(3, 2)-()]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99-0.0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-17]
0.01s call     tests/infer/test_valid_models.py::test_plate_ok[TraceEnum_ELBO-full]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-1-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-13]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-1-1]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_feasible-TraceEnum_ELBO]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-6]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-1-2]
0.01s call     tests/infer/test_enum.py::test_elbo_normal[parallel-differentiable_loss]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-14]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_8[plate-plate-1-1]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-()-()]
0.01s call     tests/ops/test_newton.py::test_newton_step_converges[2-1.0]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-TraceEnum_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-1-2]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_mean-TraceEnum_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-Size-sample_shape1]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-6]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(3, 2)-(5,)]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoLowRankMultivariateNormal-TraceGraph_ELBO]
0.01s call     tests/ops/test_contract.py::test_ubersum_total[ab,bcdi,deij->eij-ij]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-3-5-1]
0.01s call     tests/infer/test_enum.py::test_elbo_plate[1-parallel-sequential-None]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-list-sample_shape1]
0.01s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.55-0.44]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.1--1.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-Size-sample_shape1]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_3[single-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(4,)-(5,)]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-0.5]
0.01s call     tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-19]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-6]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-True-sample_shapes0]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_sample-TraceEnum_ELBO]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[idct--3]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0-1.0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-13]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-8]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoLowRankMultivariateNormal-Trace_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-1-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(3, 2)-(5,)]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[idct--2]
0.01s call     tests/infer/test_autoguide.py::test_discrete_parallel[AutoDiagonalNormal]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-1-3]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-6]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.05]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-6]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[dct-0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(3, 2)-()]
0.01s call     tests/infer/test_enum.py::test_elbo_plate[1-parallel-parallel-10000]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-13]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[idct-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(4,)-(5,)]
0.01s call     tests/infer/test_enum.py::test_elbo_plate[2-parallel-sequential-None]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[idct-1]
0.01s call     tests/distributions/test_transforms.py::TransformTests::test_radial
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-19]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Geometric-False-initial_shape2-proposed_shape2]
0.01s call     tests/infer/test_jit.py::test_infer_discrete[map-1]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-6]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_2[10]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_3[10]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[1.0]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.5]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[idct-0]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.1]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-parallel-inf]
0.01s call     tests/infer/test_autoguide.py::test_discrete_parallel[AutoIAFNormal]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[dct--2]
0.01s call     tests/ops/test_contract.py::test_einsum_linear[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(3, 2)-()]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.2]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoNormal]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[idct-2]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_2[1]
0.01s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-1.0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-19]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-3-2-2]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-False-sample_shapes0]
0.01s call     tests/infer/test_valid_models.py::test_plate_shape_broadcasting
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-0.5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(3, 2)-(5,)]
0.01s call     tests/optim/test_optim.py::OptimTests::test_per_param_optim
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-3-lgssmgp]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(4,)-(5,)]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[idct--1]
0.01s call     tests/ops/test_gaussian.py::test_cat[1-(4, 7, 6)--1-(2, 1, 3)]
0.01s call     tests/ops/test_gaussian.py::test_cat[1-(4, 7, 6)--2-(1, 1, 2, 3)]
0.01s call     tests/ops/test_integrator.py::test_trajectory[HarmonicOscillator]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-0.5]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-()-()]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-3-lgssmgp]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-0.5]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_binomial
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-0.5]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[dct--1]
0.01s call     tests/distributions/test_transforms.py::TransformTests::test_conditional_radial
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_3[1]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-True-list]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-18]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-2-lgssmgp]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-0.5]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[dct--3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-1-2]
0.01s call     tests/ops/test_gaussian.py::test_cat[1-(4, 7, 6)-1-(1, 1, 2, 3)]
0.01s call     tests/infer/test_enum.py::test_elbo_iplate[1-parallel-sequential]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-18]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[dct-1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-()-()]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-()-()]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-2-lgssmgp]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[dct-2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_categorical[4]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-1.5]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoNormal-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-True-tuple]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-True-Size]
0.01s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-0-2000]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-1.5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-1-3]
0.01s call     tests/infer/reparam/test_neutra.py::test_reparam_log_joint[neals_funnel-kwargs0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-7]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-1.5]
0.01s call     tests/nn/test_module.py::test_torch_serialize_attributes
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-12]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-1.5]
0.01s call     tests/ops/test_tensor_utils.py::test_dct_dim[dct-3]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceEnum_ELBO-subsample]
0.01s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-subsample-inf]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-1.5]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-6]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_1[single-1]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-1.5]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_2[single-10]
0.01s call     tests/infer/test_valid_models.py::test_plate_no_size_ok[TraceEnum_ELBO]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_3[1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(3, 2)-()]
0.01s call     tests/distributions/test_extended.py::test_extended_beta_binomial[0.05]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_1[1]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoMultivariateNormal-TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-17]
0.01s call     tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-18]
0.01s call     tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-subsample-0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(3, 2)-(5,)]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-sequential-inf]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Poisson-False-initial_shape2-proposed_shape2]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_2[single-1]
0.01s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-subsample-inf]
0.01s call     tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-subsample-0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-12]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_median-TraceGraph_ELBO]
0.01s call     tests/contrib/easyguide/test_easyguide.py::test_overlapping_plates_ok
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_1[10]
0.01s call     tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-12]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-16]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-Size-sample_shape0]
0.01s call     tests/infer/test_enum.py::test_avoid_nan[sequential]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-3-2-7]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-tuple-sample_shape0]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.1--0.5]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01-0.0]
0.01s call     tests/contrib/gp/test_models.py::test_forward[SparseGPRegression_y2D]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(4,)-()]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-(2, 1)-(3,)]
0.01s call     tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape5]
0.01s call     tests/infer/test_enum.py::test_elbo_plate[2-sequential-parallel-None]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_1[single-10]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(3, 2)-(5,)]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoMultivariateNormal-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-Size-sample_shape2]
0.01s call     tests/ops/test_contract.py::test_ubersum[bi,ia->,i,ia,ib,iab-i]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-17]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-list-sample_shape2]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_hmm[4]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-11]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-tuple-sample_shape2]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwxj,ixyj,iyzj,izwj->ij]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-Size-sample_shape2]
0.01s call     tests/poutine/test_mapdata.py::test_replay[nested_iplate-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-17]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-tuple-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-tuple-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[LogNormal-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Multinomial-False-initial_shape0-proposed_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-False-initial_shape0-proposed_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-11]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[4]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-Size-sample_shape2]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-(2, 1)-(3,)]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-(2, 1)-(3,)]
0.01s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-2-False]
0.01s call     tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceTMC_ELBO-full]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-tuple-sample_shape2]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[LogNormal-False-initial_shape0-proposed_shape0]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-4]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.8-1.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-False-initial_shape0-proposed_shape0]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_median-Trace_ELBO]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.8--1.0]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.8-0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Multinomial-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-11]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-tuple-sample_shape2]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.1-0.5]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.1-1.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-15]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.8--0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-list-sample_shape2]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-3-1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-6]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-(2, 1)-(3,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-list-sample_shape2]
0.01s call     tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceTMC_ELBO-subsample]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-False-initial_shape1-proposed_shape1]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-3-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-tuple-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-False-list]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-list-sample_shape1]
0.01s call     tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-parallel-parallel]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-False-tuple]
0.01s call     tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->cij-ij]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Binomial-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-False-Size]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-Size-sample_shape2]
0.01s call     tests/ops/test_gaussian.py::test_rsample_distribution[3-()]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[4-5]
0.01s call     tests/infer/test_enum.py::test_hmm_enumerate_model[4]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-3-lgssmgp]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-2-lgssmgp]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Binomial-tuple-sample_shape2]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1-0.0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-list-sample_shape2]
0.01s call     tests/distributions/test_extended.py::test_extended_beta_binomial[0.1]
0.01s call     tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-tuple-sample_shape2]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-()-(3,)]
0.01s call     tests/nn/test_module.py::test_mixin_factory
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-False-sample_shapes0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-()-(3,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Binomial-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-16]
0.01s call     tests/contrib/gp/test_models.py::test_forward[SparseGPRegression_y1D]
0.01s call     tests/infer/test_enum.py::test_backwardsample_posterior_2
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-True-sample_shapes0]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[4-2]
0.01s call     tests/infer/test_enum.py::test_elbo_iplate[1-sequential-parallel]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-()-()]
0.01s call     tests/infer/test_valid_models.py::test_nested_plate_plate_subsample_param_ok[Trace_ELBO]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-()-(3,)]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_feasible-TraceGraph_ELBO]
0.01s call     tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Categorical-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-10]
0.01s call     tests/ops/test_gamma_gaussian.py::test_logsumexp[1-(3, 2)]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_mean-TraceGraph_ELBO]
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceEnum_ELBO-subsample]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-()-(5,)]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4-0.0]
0.01s call     tests/distributions/test_stable.py::test_normal[0.5-1]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-3-lgssm]
0.01s call     tests/ops/test_contract.py::test_adjoint_marginal[ai,abij,acik->i-ijk]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(3, 2)-(5,)]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(4,)-()]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-2-lgssm]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-False-initial_shape0-proposed_shape0]
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceEnum_ELBO-full]
0.01s call     tests/infer/test_autoguide.py::test_discrete_parallel[AutoDelta]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-()-(5,)]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-2]
0.01s call     tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-2-False]
0.01s call     tests/distributions/test_stable.py::test_normal[0.5-21]
0.01s call     tests/distributions/test_stable.py::test_normal[2-21]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8-0.0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-15]
0.01s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.39-0.44]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-10]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-14]
0.01s call     tests/distributions/test_transforms.py::TransformTests::test_householder
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-tuple-sample_shape2]
0.01s call     tests/infer/test_enum.py::test_elbo_plate[1-sequential-parallel-None]
0.01s call     tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-parallel-None]
0.01s call     tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-None-parallel]
0.01s call     tests/distributions/test_stable.py::test_normal[1-0]
0.01s call     tests/distributions/test_stable.py::test_normal[0.5--1]
0.01s call     tests/infer/test_valid_models.py::test_plate_reuse_ok[TraceGraph_ELBO]
0.01s call     tests/distributions/test_stable.py::test_normal[2--1]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3-0.0]
0.01s call     tests/distributions/test_stable.py::test_normal[1-20]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-10]
0.01s call     tests/distributions/test_stable.py::test_normal[0.5-20]
0.01s call     tests/distributions/test_stable.py::test_normal[1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-4]
0.01s call     tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-None-parallel]
0.01s call     tests/distributions/test_stable.py::test_normal[1--1]
0.01s call     tests/distributions/test_stable.py::test_normal[1-21]
0.01s call     tests/distributions/test_stable.py::test_normal[2-20]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-Size-sample_shape2]
0.01s call     tests/distributions/test_stable.py::test_normal[2-1]
0.01s call     tests/distributions/test_stable.py::test_normal[2-0]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Exponential-False-initial_shape0-proposed_shape0]
0.01s call     tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-False-initial_shape0-proposed_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Categorical-False-initial_shape0-proposed_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-False-initial_shape0-proposed_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-list-sample_shape2]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7-0.0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-(3,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-1-1]
0.01s call     tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.39-0.33]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(3, 2)-(5,)]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_sample-TraceGraph_ELBO]
0.01s call     tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-2-True]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-Size-sample_shape2]
0.01s call     tests/ops/test_newton.py::test_newton_step[3-None-batch_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-Size-sample_shape2]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-(3,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-tuple-sample_shape2]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-()-(5,)]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0-0.0]
0.01s call     tests/infer/test_valid_models.py::test_vectorized_num_particles[Trace_ELBO]
0.01s call     tests/infer/test_autoguide.py::test_discrete_parallel[AutoLaplaceApproximation]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Normal-False-initial_shape1-proposed_shape1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-(3,)-()]
0.01s call     tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-parallel-None]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(3, 2)-()]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8-1.0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-1-1]
0.01s call     tests/distributions/test_stable.py::test_sample_2[1.3--1.0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-1-1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-(3,)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(4,)-()]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[None-differentiable_loss-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Normal-False-initial_shape0-proposed_shape0]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoIAFNormal-TraceGraph_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-True-sample_shapes0]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8-0.5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-9]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_mean-Trace_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-9]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4--0.5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(4,)-(5,)]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-()-(3,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-tuple-sample_shape2]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3-0.5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(4,)-(5,)]
0.01s call     tests/distributions/test_stable.py::test_sample_2[1.7--0.5]
0.01s call     tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape0]
0.01s call     tests/distributions/test_stable.py::test_sample_2[1.7--1.0]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixyj,iyzj->ij]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-tuple-sample_shape0]
0.01s call     tests/distributions/test_stable.py::test_sample_2[1.7-0.5]
0.01s call     tests/distributions/test_stable.py::test_sample_2[1.7-1.0]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-3-2-1]
0.01s call     tests/infer/test_enum.py::test_svi_step_smoke[None-single]
0.01s call     tests/distributions/test_stable.py::test_sample_2[1.3--0.5]
0.01s call     tests/distributions/test_stable.py::test_sample_2[1.3-1.0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-1-1]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plates_3[10]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-3-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-16]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_feasible-Trace_ELBO]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoIAFNormal-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-Size-sample_shape2]
0.01s call     tests/infer/test_enum.py::test_avoid_nan[parallel]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-13]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_stable.py::test_sample_2[1.3-0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-Size-sample_shape2]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7--1.0]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-2]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-()-()]
0.01s call     tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-3-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-()-(5,)]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-True-False]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-list-sample_shape2]
0.01s call     tests/ops/test_contract.py::test_naive_ubersum[,i,ia->,i,ia-i]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-True-False]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoMultivariateNormal]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-1-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-3-2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-9]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-3-3]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-True-False]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-3-3]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-True-False]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-3-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-2-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-2-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-tuple-sample_shape0]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_sample-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-Size-sample_shape0]
0.01s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-parallel-None]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-(2, 3)-(2, 3)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-14]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-tuple-sample_shape0]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-19]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-2-2]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-False-sample_shapes0]
0.01s call     tests/ops/test_gaussian.py::test_rsample_distribution[1-(4,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(3, 2)-()]
0.01s call     tests/infer/test_valid_models.py::test_iplate_ok[TraceGraph_ELBO-full]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(4,)-(5,)]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-1-1-7]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-5]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-()-()]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-()-()]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-()-()]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-15]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-list-sample_shape2]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-(2, 3)-(2, 3)]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-3-lgssm]
0.01s call     tests/ops/test_contract.py::test_ubersum_total[a,ai,abij->bij-ij]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-True-sample_shapes0]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Multinomial-False-initial_shape2-proposed_shape2]
0.01s call     tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->bij-ij]
0.01s call     tests/infer/test_enum.py::test_elbo_normal[parallel-loss_and_grads]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-5]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-1-1]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-3-lgssm]
0.01s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-1.0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-2-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-16]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Exponential-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-3-2]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-(2, 3)-(2, 3)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-2-2]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-2-lgssm]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-()-()]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-3-2-7]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-3-3]
0.01s call     tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTraceEnum_ELBO]
0.01s call     tests/ops/test_contract.py::test_ubersum[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_categorical[3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-()-(5,)]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-(2, 3)-(2, 3)]
0.01s call     tests/ops/test_contract.py::test_adjoint_marginal[a,abi,bcij->-ij]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-10]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-6]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-2-lgssm]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[None-loss_and_grads-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-16]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-1-2]
0.01s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-sequential-None]
0.01s call     tests/contrib/gp/test_models.py::test_gplvm[VariationalSparseGP_y2D]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-5]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.4--0.5]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-5]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.4--1.0]
0.01s call     tests/ops/test_integrator.py::test_energy_conservation[HarmonicOscillator]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-1-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-19]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.4-0.5]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.4-1.0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-18]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-False-initial_shape2-proposed_shape2]
0.01s call     tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape4]
0.01s call     tests/poutine/test_poutines.py::QueueHandlerDiscreteTest::test_queue_enumerate
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-12]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-13]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[LogNormal-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-2-2]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-3-2-1]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-True-sample_shapes0]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[wxi,xyi,yzi,zwi->i]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-15]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Normal-False-initial_shape2-proposed_shape2]
0.01s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-sequential-None]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-tuple-sample_shape1]
0.01s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-1-2000]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-False-sample_shapes0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-list-sample_shape1]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(4,)-(5,)]
0.01s call     tests/distributions/test_stable.py::test_sample_2[1.7-0.0]
0.01s call     tests/distributions/test_stable.py::test_sample_2[2.0-0.0]
0.01s call     tests/infer/test_valid_models.py::test_plate_iplate_ok[Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-list-sample_shape1]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoDiagonalNormal-TraceGraph_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-14]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-tuple-sample_shape1]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-5]
0.01s call     tests/contrib/test_minipyro.py::test_constraints[minipyro-jit]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Categorical-False-initial_shape2-proposed_shape2]
0.01s call     tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-15]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-1.0]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-1.0]
0.01s call     tests/ops/test_newton.py::test_newton_step[1-None-batch_shape5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-17]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwxj,ixyj,iyzj,izwj->ji]
0.01s call     tests/ops/test_contract.py::test_ubersum_jit[,i,i->,i-i]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-2-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(3, 2)-()]
0.01s call     tests/ops/test_contract.py::test_ubersum_2[naive_ubersum]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-3-1]
0.01s call     tests/infer/test_valid_models.py::test_iplate_plate_ok[Trace_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-3-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-3-1]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-1.0]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-1.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Bernoulli-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(4,)-()]
0.01s call     tests/ops/test_newton.py::test_newton_step[1-None-batch_shape4]
0.01s call     tests/contrib/gp/test_models.py::test_gplvm[VariationalSparseGP_y1D]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.8-0.0]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-1.0]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-1.0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-3-1]
0.01s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-parallel-None]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-True-sample_shapes0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-11]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-18]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_stable.py::test_sample_2[1.3-0.0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-2-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-13]
0.01s call     tests/distributions/test_extended.py::test_extended_binomial[0.02]
0.01s call     tests/infer/reparam/test_stable.py::test_subsample_smoke[StableReparam-subsample]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(4,)-(5,)]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-3-lgssm]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(3, 2)-()]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-list-sample_shape1]
0.01s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-0.01]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-Size-sample_shape1]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-2-1-7]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-2-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-6]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(3, 2)-()]
0.01s call     tests/infer/mcmc/test_valid_models.py::test_enumeration_in_tree[TraceEinsumEvaluator]
0.01s call     tests/contrib/gp/test_models.py::test_forward[VariationalSparseGP_y2D]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-2-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-tuple-sample_shape1]
0.01s call     tests/infer/test_enum.py::test_differentiable_loss[parallel-batch]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Bernoulli-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-14]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-12]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-19]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-2-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Bernoulli-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(4,)-()]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4-1.0]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-4]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoDiagonalNormal-Trace_ELBO]
0.01s call     tests/infer/test_enum.py::test_elbo_iplate[2-parallel-parallel]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.4-0.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(4,)-(5,)]
0.01s call     tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-2-lgssm]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-8]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Exponential-False-initial_shape2-proposed_shape2]
0.01s call     tests/contrib/gp/test_models.py::test_forward[VariationalGP_y2D]
0.01s call     tests/distributions/test_stable.py::test_sample_2[0.1-0.0]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8-0.0]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01-0.0]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-True-sample_shapes0]
0.01s call     tests/ops/test_contract.py::test_ubersum_3[naive_ubersum]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7-0.0]
0.01s call     tests/infer/reparam/test_stable.py::test_subsample_smoke[StableReparam-full]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-14]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-Size-sample_shape2]
0.01s call     tests/infer/test_valid_models.py::test_reparam_stable
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[3]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.1]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_beta_binomial
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-False-sample_shapes0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-17]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.05]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-tuple-sample_shape1]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01-0.5]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-13]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(4,)-()]
0.01s call     tests/contrib/forecast/test_forecaster.py::test_svi_custom_smoke[False]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(4,)-(5,)]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3-0.0]
0.01s call     tests/infer/test_discrete.py::test_distribution_masked[map]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-sequential-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-list-sample_shape2]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1-1.0]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.5]
0.01s call     tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[1.0]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01-1.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-tuple-sample_shape2]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1--0.5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-8]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-Size-sample_shape2]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99-0.0]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01--0.5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(3, 2)-()]
0.01s call     tests/infer/test_valid_models.py::test_plate_enum_discrete_no_discrete_vars_warning[False]
0.01s call     tests/pyroapi/test_pyroapi.py::test_local_param_ok[pyro-py]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1-0.0]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7--0.5]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99--1.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-Size-sample_shape2]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4--1.0]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4-0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-Size-sample_shape0]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8--1.0]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[wi,wxi,xyi,yzi->i]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8--0.5]
0.01s call     tests/infer/test_enum.py::test_hmm_enumerate_model[3]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-auto_class7]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7-0.5]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4-0.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0--0.5]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3--0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Binomial-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-12]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Binomial-list-sample_shape1]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3--1.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-()-(5,)]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0-0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-list-sample_shape2]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1-0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Binomial-tuple-sample_shape1]
0.01s call     tests/infer/test_enum.py::test_svi_step_smoke[sequential-batch]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01--1.0]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99-1.0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-8]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-10]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-7]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3-1.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99-0.5]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99--0.5]
0.01s call     tests/distributions/test_extended.py::test_extended_binomial[0.05]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-18]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-2]
0.01s call     tests/contrib/forecast/test_forecaster.py::test_svi_custom_smoke[True]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-True-sample_shapes1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-()-(5,)]
0.01s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7-1.0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-11]
0.01s call     tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-parallel-None]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-19]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-tuple-sample_shape2]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-2.0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-7]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-3.0]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-2.0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-13]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-2.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-list-sample_shape2]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-3.0]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-4]
0.01s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-0.1]
0.01s call     tests/distributions/test_extended.py::test_extended_binomial[0.1]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-2.0]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[3-5]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-2]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-(3,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-Size-sample_shape2]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-2.0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(3, 2)-(6, 5)]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-3.0]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-3.0]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-(3,)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(4,)-()]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-3.0]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-2.0]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-3-2-1]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-(3,)-()]
0.01s call     tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-3.0]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoDiagonalNormal]
0.01s call     tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceEnum_ELBO-subsample]
0.01s call     tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceEnum_ELBO-full]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-Size-sample_shape2]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-tuple-sample_shape2]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(3, 2)-(5,)]
0.01s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-nonreparam-scaled]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-(3,)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-()-(5,)]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(3, 2)-(6, 5)]
0.01s call     tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-parallel-parallel]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-()-(3,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(3, 2)-()]
0.01s call     tests/infer/test_enum.py::test_svi_step_guide_uses_grad[parallel]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-11]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-()-(5,)]
0.01s call     tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceEnum_ELBO-True]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-2]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-3-2-7]
0.01s call     tests/infer/test_valid_models.py::test_nested_plate_plate_ok[Trace_ELBO]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[None-loss-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-9]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-5]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixyj,iyzj->ji]
0.01s call     tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[minipyro-jit]
0.01s call     tests/contrib/test_minipyro.py::test_plate_ok[minipyro-jit]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-17]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-tuple-sample_shape1]
0.01s call     tests/ops/test_newton.py::test_newton_step_converges[3-10.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-7]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(4,)-()]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-12]
0.01s call     tests/infer/reparam/test_split.py::test_normal[(3, 2)-(2, 3, 1)-[1, 2]--2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-18]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-Size-sample_shape2]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[3-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-list-sample_shape0]
0.01s call     tests/ops/test_gaussian.py::test_rsample_distribution[2-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-Size-sample_shape0]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-auto_class6]
0.01s call     tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-parallel-None]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-tuple-sample_shape2]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-10]
0.01s call     tests/contrib/gp/test_models.py::test_forward[GPRegression_y2D]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-2]
0.01s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-False-unscaled]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-3-2-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(3, 2)-(5,)]
0.01s call     tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape2-effective_sample_size]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-True-sample_shapes0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-16]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-()-(3,)]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-False-sample_shapes0]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-()-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-Size-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-6]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-1]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoLaplaceApproximation-TraceGraph_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-list-sample_shape2]
0.01s call     tests/infer/test_valid_models.py::test_not_has_rsample_ok[TraceEnum_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-1-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Poisson-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-Size-sample_shape0]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-()-(3,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Poisson-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-tuple-sample_shape0]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-()-(3,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Poisson-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-1-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-tuple-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(4,)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-5]
0.01s call     tests/contrib/gp/test_models.py::test_inference_deepGP
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-3-2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-12]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-False-10]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-list-sample_shape0]
0.01s call     tests/ops/test_contract.py::test_ubersum[abi,b->,b,ai,abi-i]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-False-sample_shapes1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Geometric-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-11]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Geometric-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-list-sample_shape2]
0.01s call     tests/infer/test_valid_models.py::test_nonnested_plate_plate_ok[Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-Size-sample_shape1]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[wx,xy,yz,zw->]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-17]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Geometric-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-6]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-1-2]
0.01s call     tests/ops/test_contract.py::test_einsum_linear[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.01s call     tests/infer/test_enum.py::test_differentiable_loss[None-batch]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoLaplaceApproximation-Trace_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-tuple-sample_shape0]
0.01s call     tests/contrib/gp/test_models.py::test_forward[VariationalSparseGP_y1D]
0.01s call     tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[True-None-1-False]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-1-3]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-2-1-7]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-tuple-sample_shape0]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-4]
0.01s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-nonreparam-unscaled]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-4]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-1-3]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-3]
0.01s call     tests/distributions/test_extended.py::test_extended_beta_binomial[0.0]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-1-2]
0.01s call     tests/ops/test_contract.py::test_ubersum_total[,ai,abij->aij-ij]
0.01s call     tests/infer/test_discrete.py::test_plate_smoke[2-sample-elbo]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-15]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-list-sample_shape2]
0.01s call     tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceEnum_ELBO-False]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Binomial-list-sample_shape0]
0.01s call     tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-1-2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-9]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-3]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[7]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Binomial-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[VonMises-False-initial_shape1-proposed_shape1]
0.01s call     tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-False-1]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5--0.9]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-list-sample_shape0]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-3]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-2]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-3-2-7]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-6]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(3, 2)-(6, 5)]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-3]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[None-loss_and_grads-10]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(4,)-(5,)]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-2]
0.01s call     tests/ops/test_newton.py::test_newton_step[2-None-batch_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Binomial-tuple-sample_shape0]
0.01s call     tests/contrib/oed/test_finite_spaces_eig.py::test_nmc_eig_finite_space_model
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-1-2]
0.01s call     tests/ops/test_contract.py::test_einsum_linear[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.01s call     tests/distributions/test_extended.py::test_extended_binomial[0.0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-1-3]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Multinomial-Size-sample_shape2]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-2]
0.01s call     tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-parallel-parallel]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-Size-sample_shape2]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Multinomial-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-()-(5,)]
0.01s call     tests/contrib/gp/test_models.py::test_forward[VariationalGP_y1D]
0.01s call     tests/pyroapi/test_pyroapi.py::test_plate_ok[pyro-py]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Multinomial-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-tuple-sample_shape2]
0.01s call     tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-True-list]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-True-tuple]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-()-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-True-Size]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-9]
0.01s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-False-scaled]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoDelta-Trace_ELBO]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-6]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_median-TraceGraph_ELBO]
0.01s call     tests/ops/test_stats.py::test_hpdi
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-5]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_restrictions[True-None-1-False]
0.01s call     tests/contrib/gp/test_models.py::test_forward[GPRegression_y1D]
0.01s call     tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-3]
0.01s call     tests/contrib/gp/test_models.py::test_gplvm[SparseGPRegression_y2D]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-()-(5,)]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9-0.9]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(3, 2)-(6, 5)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[VonMises-False-initial_shape0-proposed_shape0]
0.01s call     tests/infer/test_discrete.py::test_distribution_masked[sample]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(3, 2)-(6, 5)]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-11]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-tuple-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-6]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-parallel-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-4]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-16]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-10]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-Size-sample_shape2]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-False-False]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-True-sample_shapes0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-5]
0.01s call     tests/contrib/gp/test_models.py::test_gplvm[VariationalGP_y2D]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-False-False]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-tuple-sample_shape0]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_hmm[3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(4,)-(5,)]
0.01s call     tests/contrib/gp/test_models.py::test_gplvm[SparseGPRegression_y1D]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-False-initial_shape0-proposed_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-tuple-sample_shape1]
0.01s call     tests/infer/test_valid_models.py::test_nested_plate_plate_subsample_param_ok[TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-14]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-(2, 3)-(2, 3)]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Beta-False-initial_shape0-proposed_shape0]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[inf]
0.01s call     tests/infer/test_autoguide.py::test_factor[AutoDelta-TraceGraph_ELBO]
0.01s call     tests/ops/test_contract.py::test_einsum_linear[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-()-()]
0.01s call     tests/pyroapi/test_pyroapi.py::test_constraints[pyro-py]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-False-sample_shapes0]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-list-sample_shape1]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-3-2-1]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-()-()]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoLaplaceApproximation]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-()-(5,)]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_median-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-Size-sample_shape0]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-(2, 3)-(2, 3)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(3, 2)-(5,)]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[w,wx,xy,yz->]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-(2, 3)-(2, 3)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-()-(5,)]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-(2, 3)-(2, 3)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-9]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-list-sample_shape1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-()-(3,)]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(3, 2)-(5,)]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-True]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-tuple-sample_shape1]
0.01s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[8-size1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-15]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(3, 2)-()]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-2]
0.01s call     tests/ops/test_gamma_gaussian.py::test_reshape[3-(5, 6)-(5, 3, 2)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-False-Size]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_single_ok
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5-0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Gamma-False-initial_shape1-proposed_shape1]
0.01s call     tests/infer/test_valid_models.py::test_iplate_ok[Trace_ELBO-subsample]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5-0.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-False-tuple]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-10]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Uniform-False-initial_shape1-proposed_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-6]
0.01s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-()-(3,)]
0.01s call     tests/contrib/gp/test_models.py::test_gplvm[VariationalGP_y1D]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5--0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Cauchy-False-initial_shape0-proposed_shape0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-()-(3,)]
0.01s call     tests/infer/reparam/test_split.py::test_normal[(3, 2)-(4, 2)-[1, 3]--2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-4]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-Size-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0-0.9]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Cauchy-False-initial_shape1-proposed_shape1]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-False]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-False-False]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-False-initial_shape0-proposed_shape0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-()-(3,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-False-list]
0.01s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-1]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5-0.5]
0.01s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-0-None]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-4]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0--0.9]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_median-TraceGraph_ELBO]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9--0.9]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-13]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0--0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-tuple-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-tuple-sample_shape1]
0.01s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-1]
0.01s call     tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[pyro-py]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-4]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-list-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5-0.0]
0.01s call     tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Uniform-False-initial_shape0-proposed_shape0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5-0.0]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-mask2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(4,)-(5,)]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-scale2]
0.01s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-parallel-300]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-scale3]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Dirichlet-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5-0.9]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_in_iplate[2]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-False-False]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoDelta]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-mask4]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-mask3]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-False]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0-0.5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Dirichlet-False-initial_shape0-proposed_shape0]
0.01s call     tests/infer/test_valid_models.py::test_nested_plate_plate_ok[TraceGraph_ELBO]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9-0.0]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_mean-TraceGraph_ELBO]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-5]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-True]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Gamma-False-initial_shape0-proposed_shape0]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-mask3]
0.01s call     tests/infer/test_enum.py::test_svi_step_guide_uses_grad[None]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-19]
0.01s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9--0.5]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-1]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_median-TraceGraph_ELBO]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-Size-sample_shape0]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-1]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_sample-TraceGraph_ELBO]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-1]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-1]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_median-Trace_ELBO]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5--0.5]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-8]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-16]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Beta-False-initial_shape1-proposed_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-tuple-sample_shape0]
0.01s call     tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-tuple-sample_shape0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0-0.0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(3, 2)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-5]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-scale2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-5]
0.01s call     tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[pyro-jit]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-mask4]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-6]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-scale3]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Multinomial-list-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9--0.5]
0.01s call     tests/ops/test_special.py::test_log_beta_stirling[0.02]
0.01s call     tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-Adam-optim_kwargs0-jit]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-mask2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-tuple-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-6]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9--0.9]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-Size-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5-0.9]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_feasible-TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(4,)-(5,)]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9--0.9]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-list-sample_shape1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9-0.9]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-Trace_ELBO]
0.01s call     tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTraceGraph_ELBO]
0.01s call     tests/infer/test_valid_models.py::test_markov_history[1]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-2]
0.01s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-parallel-300]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_restrictions[False-None-2-False]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-3-2-1]
0.01s call     tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-1-None]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9--0.9]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-8]
0.01s call     tests/infer/test_valid_models.py::test_vectorized_num_particles[TraceGraph_ELBO]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5--0.5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-6]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9--0.9]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(3, 2)-()]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5-0.0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-6]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-1]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-2-1-1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9-0.9]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-6]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9--0.9]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_mean-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-list-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9--0.5]
0.01s call     tests/infer/reparam/test_split.py::test_normal[(3, 2)-(2, 5)-[2, 3]--1]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9-0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-Size-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5-0.5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-2]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5--0.9]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-tuple-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9-0.9]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-0.1]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_feasible-Trace_ELBO]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0-0.9]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-(2, 1)-(3,)]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5-0.5]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-2]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5-0.9]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5--0.5]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_sample-Trace_ELBO]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_mean-TraceGraph_ELBO]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_feasible-TraceGraph_ELBO]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0--0.5]
0.01s call     tests/ops/test_gamma_gaussian.py::test_logsumexp[1-(4,)]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5--0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-list-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9-0.0]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-1-5-2]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9-0.0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(4,)-()]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5-0.9]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-9]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-list-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5--0.9]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-0.1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0-0.5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(3, 2)-()]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5-0.0]
0.01s call     tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[2]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0-0.0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-14]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5--0.9]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_restrictions[False-parallel-1-False]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5-0.9]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_median-Trace_ELBO]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0--0.9]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5--0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[VonMises-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9--0.5]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-1-5-1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-(2, 1)-(3,)]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0--0.9]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9-0.0]
0.01s call     tests/ops/test_linalg.py::test_sym_rinverse[True-A4]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9-0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-Size-sample_shape1]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9--0.9]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Binary_1D]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9--0.9]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-6]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0-0.9]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-(2, 1)-(3,)]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5-0.5]
0.01s call     tests/ops/test_contract.py::test_naive_ubersum[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9-0.0]
0.01s call     tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-None-2-False]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5--0.9]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-True-Size]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9--0.5]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_median-TraceGraph_ELBO]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9-0.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-False-Size]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5--0.9]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0--0.9]
0.01s call     tests/infer/test_valid_models.py::test_markov_history[0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0-0.0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-4]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0-0.5]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-2]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0-0.5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-15]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0--0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-Size-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5--0.5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-18]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5-0.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-list-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5-0.5]
0.01s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceTMC_ELBO-subsample]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5--0.9]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9--0.5]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0-0.9]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-14]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5--0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-True-list]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_sample-TraceGraph_ELBO]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9-0.9]
0.01s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.55-0.33]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_mean-Trace_ELBO]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9-0.0]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0-0.5]
0.01s call     tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-ClippedAdam-optim_kwargs1-jit]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9--0.9]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-True-tuple]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9--0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0--0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0--0.5]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-2]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5--0.5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_categorical[2]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0--0.9]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9-0.5]
0.01s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.27-0.43]
0.01s call     tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9--0.9]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Multinomial-Size-sample_shape1]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_feasible-Trace_ELBO]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-6]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9-0.9]
0.01s call     tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9-0.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-Size-sample_shape1]
0.01s call     tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0-0.5]
0.01s call     tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5-0.5]
0.01s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceTMC_ELBO-full]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-12]
0.01s call     tests/ops/test_linalg.py::test_sym_rinverse[False-A4]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Multinomial-tuple-sample_shape1]
0.01s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.55-0.43]
0.01s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-False-list]
0.01s call     tests/infer/test_enum.py::test_elbo_scale
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(3, 2)-(5,)]
0.01s call     tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.27-0.33]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(4,)-()]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-2-1-7]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-16]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-19]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[2]
0.01s call     tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-parallel-1-False]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(4,)-(5,)]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-1-2-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-False-tuple]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_mean-TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-19]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-1-2-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-3]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-7]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(4,)-(5,)]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-Size-sample_shape1]
0.01s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(3, 2)-()]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_feasible-TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-13]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-3]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_median-Trace_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-8]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Beta-False-initial_shape2-proposed_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-4]
0.01s call     tests/infer/test_valid_models.py::test_nonnested_plate_plate_ok[TraceGraph_ELBO]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Bernoulli-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-list-sample_shape1]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-2-1-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Bernoulli-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-1.0]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_sample-Trace_ELBO]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-3]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-17]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(3, 2)-()]
0.01s call     tests/ops/test_gaussian.py::test_logsumexp[1-(3, 2)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-4]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-0.5]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-3]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-4]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-1-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-4]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-11]
0.01s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-0.1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-tuple-sample_shape1]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_MultiClass_2D]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_mean-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Bernoulli-tuple-sample_shape1]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_feasible-Trace_ELBO]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-2]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-2.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-Size-sample_shape2]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-0.0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-3]
0.01s call     tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-tuple-sample_shape1]
0.01s setup    tests/integration_tests/test_tracegraph_elbo.py::ExponentialGammaTests::test_elbo_reparameterized
0.01s call     tests/infer/test_valid_models.py::test_iplate_plate_ok[TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-15]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-list-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-3]
0.01s call     tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-1]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-14]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-(3,)-()]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoNormal]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-8]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_missing_config_warning[False]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-list-sample_shape1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-(3,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-19]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-(3,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-list-sample_shape2]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_sample-TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-18]
0.01s call     tests/infer/test_enum.py::test_elbo_plate[2-parallel-parallel-None]
0.01s call     tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTrace_ELBO]
0.01s call     tests/infer/test_valid_models.py::test_plate_iplate_ok[TraceGraph_ELBO]
0.01s call     tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalSparseGP_y2D]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-4]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-True-list]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-tuple-sample_shape2]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceEnum_ELBO-parallel]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-1]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-()-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-tuple-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Gamma-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-tuple-sample_shape2]
0.01s call     tests/nn/test_module.py::test_svi_smoke
0.01s call     tests/infer/reparam/test_haar.py::test_uniform[(2, 5)--1-True]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-(3,)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(4,)-()]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-1-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Cauchy-False-initial_shape2-proposed_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-Size-sample_shape2]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-3]
0.01s call     tests/infer/test_inference.py::test_sequential_plating_sum
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-list-sample_shape2]
0.01s call     tests/infer/test_smcfilter.py::test_smoke[1-1-1-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-1]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_MultiClass_1D]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-6]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(3, 2)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-4]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Binary_2D]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-True-tuple]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-True-Size]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-tuple-sample_shape1]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-tuple-sample_shape1]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Poisson_2D]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-7]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-Size-sample_shape1]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_median-TraceGraph_ELBO]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(4,)-()]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-list-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-4]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-18]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-tuple-sample_shape2]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_sample-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(4,)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-18]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-7]
0.01s call     tests/infer/reparam/test_haar.py::test_uniform[(2, 5)--1-False]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-True-tuple]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-True-Size]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-17]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-13]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-True-list]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-False-list]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-False-Size]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-12]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-False-tuple]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Geometric-Size-sample_shape0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-(2, 1)-(3,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-8]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Uniform-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-13]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-list-sample_shape2]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-8]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-1-1-7]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-Size-sample_shape1]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_median-Trace_ELBO]
0.01s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[7-size1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-17]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Poisson-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-10]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Poisson-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-4]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-Size-sample_shape1]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-tuple-sample_shape1]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-4]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-()-(3,)]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-8]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-False-Size]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-tuple-sample_shape1]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-8]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_feasible-TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-14]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-list-sample_shape1]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-False-list]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Poisson-tuple-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-tuple-sample_shape1]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-8]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoNormal]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-False-Size]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-8]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Geometric-list-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-False-tuple]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Dirichlet-False-initial_shape2-proposed_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-False-list]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Geometric-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-3]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-2-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-14]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Geometric-tuple-sample_shape1]
0.01s call     tests/infer/reparam/test_split.py::test_normal[(4,)-(2, 3, 1)-[1, 2]--2]
0.01s call     tests/distributions/test_transforms.py::TransformTests::test_batchnorm
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-False-tuple]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-Trace_ELBO]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_mean-TraceGraph_ELBO]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-True-0-1000-False-1]
0.01s call     tests/distributions/test_mask.py::test_mask[2-2-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(3, 2)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-1]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-8]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-2]
0.01s call     tests/ops/test_indexing.py::test_hmm_example[-3--4]
0.01s call     tests/infer/test_enum.py::test_elbo_plate[1-parallel-parallel-None]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-5]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-()-(3,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-3]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(3, 2)-()]
0.01s call     tests/distributions/test_mask.py::test_mask[2-2-1]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-2-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-17]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-Trace_ELBO]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_sample-TraceGraph_ELBO]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_mean-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Beta-list-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Beta-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(4,)-(5,)]
0.01s call     tests/infer/test_enum.py::test_elbo_iplate[1-parallel-parallel]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Beta-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(4,)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-3]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-7]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-1]
0.01s call     tests/infer/test_enum.py::test_mixture_of_diag_normals[MixtureOfDiagNormals-scale0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-3]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_feasible-Trace_ELBO]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-1]
0.01s call     tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(3, 2)-(5,)]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-8]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-False-sample_shapes1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-9]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-4]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-19]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-2]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoLowRankMultivariateNormal]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-list-sample_shape0]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-8]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-8]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[None-loss-10]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-11]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-19]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-19]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-13]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-8]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-2-1-7]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(3, 2)-(5,)]
0.01s call     tests/ops/test_indexing.py::test_hmm_example[-4--5]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[None-differentiable_loss-10]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-14]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(4,)-()]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-parallel-inf]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Categorical-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-3]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-tuple-sample_shape0]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_median-TraceGraph_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-Size-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-5]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-8]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_sample-Trace_ELBO]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-6]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(3, 2)-()]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_feasible at 0x7f6d86b7e730>1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Bernoulli-Size-sample_shape0]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_sample at 0x7f6d86b7e7b8>1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-list-sample_shape2]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-(3,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-list-sample_shape1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-(3,)-()]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-()-()]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-True-sample_shapes1]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-2]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Bernoulli-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-12]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-(3,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-tuple-sample_shape0]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[4-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-6]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-Size-sample_shape0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-(3,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Categorical-list-sample_shape2]
0.01s setup    tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_nonreparameterized_decaying_baseline
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Categorical-Size-sample_shape2]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-1]
0.01s call     tests/contrib/oed/test_linear_models_eig.py::test_nmc_eig_linear_model
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-2-1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-8]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-Size-sample_shape0]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_feasible at 0x7f6d86b7e730>0]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-3]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-()-(3,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Bernoulli-tuple-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-tuple-sample_shape1]
0.01s call     tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalSparseGP_y1D]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-2]
0.01s call     tests/ops/test_gamma_gaussian.py::test_logsumexp[3-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-Size-sample_shape1]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_median-Trace_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-16]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-tuple-sample_shape0]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-()-(3,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-Size-sample_shape0]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_single[OneHotCategorical-prior6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-18]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_sample at 0x7f6d86b7e7b8>0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-list-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-10]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(3, 2)-()]
0.01s call     tests/ops/test_contract.py::test_naive_ubersum[bi,ia->,i,ia,ib,iab-i]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-4]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-1-1-7]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Poisson_1D]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-tuple-sample_shape0]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-4]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-list-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-list-sample_shape0]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_mean-TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-17]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-tuple-sample_shape1]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-False-0-1000-False-1]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Binary_1D]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-0.5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-1]
0.01s call     tests/pyroapi/test_pyroapi.py::test_mean_field_ok[pyro]
0.01s call     tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalSparseGP_y1D]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(3, 2)-(5,)]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_single[OneHotCategorical-prior4]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_feasible-TraceGraph_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-True-Size]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-tuple-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-11]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-True-sample_shapes1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-4]
0.01s call     tests/ops/test_contract.py::test_ubersum_total[a,ai,bij->bij-ij]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-Size-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-4]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-0.0]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-(3,)-()]
0.01s call     tests/ops/test_indexing.py::test_hmm_example[-5--3]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-1]
0.01s call     tests/ops/test_gaussian.py::test_reshape[3-(5, 6)-(5, 3, 2)]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(4,)-()]
0.01s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceEnum_ELBO-full]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-15]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-13]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-3]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_mean-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-False-Size]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(3, 2)-()]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_single[OneHotCategorical-prior5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-14]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-13]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-False-list]
0.01s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-0.01]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-()-()]
0.01s call     tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalSparseGP_y2D]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-18]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-18]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-11]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-Size-sample_shape0]
0.01s call     tests/ops/test_gaussian.py::test_rsample_distribution[1-()]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[3-5]
0.01s call     tests/distributions/test_mask.py::test_mask[2-2-0]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-False-tuple]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-True-tuple]
0.01s call     tests/infer/test_valid_models.py::test_mean_field_mask_ok[True]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-True-list]
0.01s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceEnum_ELBO-subsample]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_feasible-Trace_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-True-list]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-5]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_sample-TraceGraph_ELBO]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-2-1-7]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Poisson-tuple-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-True-Size]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-True-tuple]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[3.0-0.75]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-()-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-12]
0.01s call     tests/infer/reparam/test_split.py::test_normal[(3, 2)-(6,)-[2, 1, 3]--1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-7]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-(3,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-Size-sample_shape0]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-3]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-False-0-1000-False-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-Size-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-3]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-1.0]
0.01s call     tests/infer/test_valid_models.py::test_mean_field_ok
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-16]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-12]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-0.5]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Poisson-list-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-6]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-True-0-1000-False-1]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-2.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Poisson-Size-sample_shape0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-()-()]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(5,)-3]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-6]
0.01s call     tests/infer/reparam/test_haar.py::test_uniform[(4, 2)--2-False]
0.01s call     tests/infer/reparam/test_split.py::test_normal[(4,)-(4, 2)-[1, 3]--2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-tuple-sample_shape0]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-True-0-1000-False-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(3, 2)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-4]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-Size-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-3]
0.01s call     tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-0.01]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Geometric-tuple-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-16]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-False-sample_shapes1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-16]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Geometric-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(4,)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-tuple-sample_shape0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-()-()]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_sample-Trace_ELBO]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-0.0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(4,)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-1]
0.01s call     tests/ops/test_newton.py::test_newton_step[1-None-batch_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-list-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-4]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(3, 2)-(5,)]
0.01s call     tests/infer/test_enum.py::test_mixture_of_diag_normals[MixtureOfDiagNormalsSharedCovariance-scale1]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-True-0-1000-False-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-list-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-1]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-False-0-1000-False-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-19]
0.01s call     tests/infer/test_enum.py::test_elbo_normal[parallel-loss]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-1]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[4-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-tuple-sample_shape0]
0.01s call     tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-parallel-None]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-5]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-False-0-1000-False-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-tuple-sample_shape2]
0.01s call     tests/infer/reparam/test_split.py::test_normal[(4,)-(2, 5)-[2, 3]--1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-9]
0.01s call     tests/infer/reparam/test_haar.py::test_uniform[(6,)--1-False]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-14]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(3, 2)-()]
0.01s call     tests/ops/test_special.py::test_log_beta_stirling[0.05]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-17]
0.01s call     tests/infer/test_enum.py::test_svi_step_smoke[parallel-batch]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Normal-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-17]
0.01s call     tests/infer/test_valid_models.py::test_iplate_ok[TraceGraph_ELBO-subsample]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Normal-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Normal-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-list-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-6]
0.01s call     tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->bi-ij]
0.01s call     tests/ops/test_contract.py::test_ubersum[,i,i->,i-i]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-2-1-1]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-True-0-1000-False-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-15]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(4,)-(5,)]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoMultivariateNormal]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-False-0-1000-False-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-11]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-11]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-7]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(3, 2)-()]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-7]
0.01s call     tests/ops/test_gaussian.py::test_logsumexp[3-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(3, 2)-(5,)]
0.01s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[8-size2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-10]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-15]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-Trace_ELBO]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-2]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-7]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-7]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(4,)-()]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoMultivariateNormal]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-()-(5,)]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-False-0-1000-False-1]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-7]
0.01s call     tests/ops/test_tensor_utils.py::test_next_fast_len
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-Size-sample_shape0]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_mean-TraceGraph_ELBO]
0.01s call     tests/ops/test_gamma_gaussian.py::test_reshape[2-(5, 6)-(5, 3, 2)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-3]
0.01s call     tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-True-0-1000-False-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-4]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-7]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-16]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(4,)-(6, 5)]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(4,)-(5,)]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-Trace_ELBO]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[3-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(4,)-(5,)]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_feasible-TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(4,)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-15]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-True-list]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-TraceGraph_ELBO]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(4,)-(6, 5)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-2]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-True-tuple]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(3, 2)-()]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-2-1-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-True-Size]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-(3,)-()]
0.01s call     tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-parallel-None]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-True-list]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-3-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-3]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-5]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(2, 3)-3]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_feasible-Trace_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-True-tuple]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-13]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-True-Size]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(3, 2)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-4]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_mean-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-list-sample_shape1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(4,)-(6, 5)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-list-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-5]
0.01s call     tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-Size-sample_shape0]
0.01s call     tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data1]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_categorical[1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(4,)-()]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-7]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-()-(5,)]
0.01s call     tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape2]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-(3,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-18]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-13]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-14]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-8]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-False-Size]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-tuple-sample_shape1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(4,)-(6, 5)]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-0.5]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_MultiClass_1D]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixj->ij]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-7]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Exponential-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Exponential-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Exponential-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-()-(5,)]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_MultiClass_2D]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-8]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_sample-TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(3, 2)-(5,)]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[3-3-(3, 2)]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-True-Size]
0.01s call     tests/pyroapi/test_pyroapi.py::test_constraints[minipyro-py]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-True-tuple]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-tuple-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-True-list]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-list-sample_shape0]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-7]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-7]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-7]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-16]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-False-tuple]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-1-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-False-list]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(3, 2)-(6, 5)]
0.01s call     tests/contrib/test_minipyro.py::test_local_param_ok[pyro-py]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_single[Categorical-prior2]
0.01s call     tests/infer/reparam/test_haar.py::test_uniform[(4, 2)--2-True]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-19]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-9]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-7]
0.01s call     tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-13]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-False-Size]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-14]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(3, 2)-()]
0.01s call     tests/infer/test_enum.py::test_svi_step_smoke[None-batch]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-False-Size]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-()-(5,)]
0.01s call     tests/ops/test_gaussian.py::test_logsumexp[1-(4,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-True-list]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-Size-sample_shape1]
0.01s call     tests/test_generic.py::test_rng_state[pyro-beta_binomial]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-True-Size]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-False-tuple]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-10]
0.01s call     tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_sample-Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-True-tuple]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-()-(5,)]
0.01s call     tests/infer/reparam/test_stable.py::test_subsample_smoke[SymmetricStableReparam-subsample]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-4]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_single[Categorical-prior3]
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_param_ok[Trace_ELBO-subsample]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-Trace_ELBO-subsample]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-tuple-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-()-(5,)]
0.01s call     tests/ops/test_tensor_utils.py::test_block_diag[mat_size2-3-batch_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-False-list]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(3, 2)-()]
0.01s call     tests/infer/test_enum.py::test_hmm_enumerate_model[2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-False-list]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-17]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-Trace_ELBO-subsample]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-12]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-False-tuple]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Categorical-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(3, 2)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-5]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[sequential-loss_and_grads-10]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-7]
0.01s call     tests/infer/reparam/test_stable.py::test_subsample_smoke[SymmetricStableReparam-full]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-(2, 1)-(3,)]
0.01s call     tests/contrib/gp/test_models.py::test_gplvm[GPRegression_y1D]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Categorical-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Categorical-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(3, 2)-(5,)]
0.01s call     tests/ops/test_contract.py::test_einsum_linear[abi,b->,b,ai,abi-i]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[sequential-loss_and_grads-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-16]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(3, 2)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-12]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-16]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-12]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-1-1-7]
0.01s call     tests/infer/test_predictive.py::test_shapes[True]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(2, 3)-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-Size-sample_shape1]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<pyro.infer.autoguide.initialization._InitToGenerated object at 0x7f6d757450f0>]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-5]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-2.0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-()-(5,)]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-(2, 3)-(2, 3)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-1]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[3.0-0.1]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-Trace_ELBO-subsample]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-19]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-15]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-Size-sample_shape0]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-Trace_ELBO-subsample]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-1-1-1]
0.01s call     tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalGP_y2D]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-1.0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-2]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-0.0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(4,)-(6, 5)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-19]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-tuple-sample_shape0]
0.01s call     tests/distributions/test_mask.py::test_mask[0-2-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-False-Size]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-()-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-list-sample_shape0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(4,)-(6, 5)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-15]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-Trace_ELBO-subsample]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-False-list]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-1]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(5,)-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(3, 2)-()]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-(2, 1)-(3,)]
0.01s call     tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[pyro-jit]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-False-tuple]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-False-tuple]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-list-sample_shape2]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-(2, 1)-(3,)]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-()-(3,)]
0.01s call     tests/ops/test_contract.py::test_adjoint_marginal[ai,abij->i-ij]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-13]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[3.0-0.25]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-19]
0.01s call     tests/contrib/gp/test_models.py::test_gplvm[GPRegression_y2D]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-False-Size]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-Size-sample_shape2]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Poisson_2D]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-4]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(4,)-(6, 5)]
0.01s call     tests/distributions/test_mask.py::test_mask[0-2-1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-()-(3,)]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-()-(3,)]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(3, 2)-(5,)]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[sequential-differentiable_loss-10]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-False-list]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-()-(5,)]
0.01s call     tests/infer/mcmc/test_valid_models.py::test_enumeration_in_tree[TraceTreeEvaluator]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Binary_2D]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-list-sample_shape1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-()-(3,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-18]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-(2, 3)-(2, 3)]
0.01s call     tests/infer/test_valid_models.py::test_set_has_rsample_ok[Trace_ELBO-True]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-17]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(4,)-(6, 5)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(3, 2)-()]
0.01s call     tests/distributions/test_conjugate_update.py::test_beta_binomial[(3, 2)-(3, 2)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-Size-sample_shape1]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_MultiClass_2D]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_median at 0x7f6d86b7e840>]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[sequential-differentiable_loss-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-9]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-(2, 3)-(2, 3)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(3, 2)-(5,)]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Poisson_1D]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-14]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Beta-list-sample_shape1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-(2, 3)-(2, 3)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-18]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Beta-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-list-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-6]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_single[Categorical-prior1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Beta-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-12]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-3]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[2-3-(3, 2)]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_MultiClass_1D]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[1.0-0.75]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-Size-sample_shape2]
0.01s call     tests/ops/test_linalg.py::test_sym_rinverse[True-A3]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-Size-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-tuple-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-5]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-mask3]
0.01s call     tests/infer/test_enum.py::test_avoid_nan[None]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-18]
0.01s call     tests/infer/test_valid_models.py::test_mean_field_mask_ok[False]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-list-sample_shape0]
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_param_ok[Trace_ELBO-full]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-11]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(4,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(3, 2)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-3]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_MultiClass_2D]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-Size-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-15]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-()-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Gamma-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-18]
0.01s call     tests/infer/test_inference.py::TestFixedModelGuide::test_guide_fixed
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-6]
0.01s call     tests/ops/test_newton.py::test_newton_step_converges[2-10.0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-3]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-Trace_ELBO-full]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-14]
0.01s call     tests/distributions/test_stable.py::test_sample[1.0-0.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-tuple-sample_shape2]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-Trace_ELBO-full]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-tuple-sample_shape0]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[3-2-(3, 2)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-10]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Gamma-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-list-sample_shape2]
0.01s call     tests/infer/test_inference.py::TestFixedModelGuide::test_model_fixed
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-Trace_ELBO-full]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-11]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-list-sample_shape0]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-Trace_ELBO-full]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-Size-sample_shape2]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-True]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-list-sample_shape0]
0.01s call     tests/infer/test_valid_models.py::test_plate_ok[Trace_ELBO-subsample]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-Size-sample_shape1]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_single[Bernoulli-0.2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-list-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-6]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-()-()]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(3, 2)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-6]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-()-()]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-Trace_ELBO-full]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-17]
0.01s call     tests/infer/test_inference.py::TestFixedModelGuide::test_guide_and_model_both_fixed
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(4,)-(5,)]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-False]
0.01s call     tests/ops/test_linalg.py::test_sym_rinverse[False-A3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-4]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Gamma-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Stable-False-initial_shape0-proposed_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(4,)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-1]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-mask2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-19]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(4,)-()]
0.01s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-nonreparam-unscaled]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-11]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-6]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-False]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-6]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixj->ji]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(4,)-()]
0.01s call     tests/infer/test_inference.py::TestFixedModelGuide::test_guide_and_model_free
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-4]
0.01s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-nonreparam-scaled]
0.01s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-False-unscaled]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-17]
0.01s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[7-size2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(4,)-(5,)]
0.01s call     tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[minipyro-jit]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-True]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-auto_class7]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-18]
0.01s call     tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-False-scaled]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-13]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-()-2]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-Trace_ELBO]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-mask3]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-TraceGraph_ELBO]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Poisson_2D]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[Trace_ELBO-scale2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-()-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-tuple-sample_shape1]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[1.0-0.9]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-1]
0.01s call     tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape2-_cummin]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_to_value
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[TraceMeanField_ELBO-scale2]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoDiagonalNormal]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-12]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-(3,)-()]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[wi,wxi,xi->i]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-list-sample_shape1]
0.01s call     tests/infer/test_enum.py::test_iter_discrete_traces_vector[True-dense]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(4,)-(6, 5)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(3, 2)-(5,)]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-(3,)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Multinomial-Size-sample_shape0]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-TraceGraph_ELBO]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-12]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(3, 2)-()]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-auto_class7]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-1]
0.01s call     tests/infer/test_enum.py::test_iter_discrete_traces_vector[True-flat]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(3, 2)-()]
0.01s call     tests/infer/test_enum.py::test_iter_discrete_traces_vector[False-dense]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-(3,)-()]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-mask2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-17]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-Size-sample_shape0]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_mean at 0x7f6d86b7e8c8>1]
0.01s call     tests/infer/test_valid_models.py::test_plate_ok[Trace_ELBO-full]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Multinomial-list-sample_shape0]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-(3,)-()]
0.01s call     tests/infer/mcmc/test_valid_models.py::test_potential_fn_pickling[False]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-()-()]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-2.0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-tuple-sample_shape0]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[functools.partial(<function init_to_uniform at 0x7f6d86b7e950>, radius=0.1)]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-6]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[TraceMeanField_ELBO-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(4,)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-()-(5,)]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoDiagonalNormal]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Multinomial-tuple-sample_shape0]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-(2, 1)-(3,)]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[TraceMeanField_ELBO-0.1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-tuple-sample_shape0]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_value at 0x7f6d86b7e9d8>]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[Trace_ELBO-0.1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-()-()]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-8]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-8]
0.01s call     tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-1.0]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-True-sample_shapes1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(4,)-(5,)]
0.01s call     tests/infer/mcmc/test_valid_models.py::test_log_prob_eval_iterates_in_correct_order
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-8]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Stable-False-initial_shape1-proposed_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-10]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-list-sample_shape0]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[3.0-0.9]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-10]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-6]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(4,)-(6, 5)]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(4,)-(6, 5)]
0.01s call     tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-2-expand_shape3]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-6]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-6]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_uniform at 0x7f6d86b7e950>]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-list-sample_shape2]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-6]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(4,)-(5,)]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-1-1-1]
0.01s call     tests/infer/test_valid_models.py::test_reparam_ok[Trace_ELBO]
0.01s call     tests/nn/test_module.py::test_sample
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(4,)-()]
0.01s call     tests/infer/test_valid_models.py::test_reparam_ok[TraceMeanField_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Normal-tuple-sample_shape1]
0.01s call     tests/infer/test_enum.py::test_iter_discrete_traces_vector[False-flat]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-5]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-True-sample_shapes0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-17]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-()-()]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[2-2-(3, 2)]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-17]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-19]
0.01s call     tests/distributions/test_mask.py::test_mask[0-2-0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(4,)-(5,)]
0.01s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(4,)-(6, 5)]
0.01s call     tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(3, 2)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-6]
0.01s call     tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data3]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-tuple-sample_shape2]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[Trace_ELBO-1]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[functools.partial(<function init_to_median at 0x7f6d86b7e840>, num_samples=4)]
0.01s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape3-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-6]
0.01s call     tests/infer/reparam/test_split.py::test_normal[(4,)-(6,)-[2, 1, 3]--1]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[3.0-0.5]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-()-()]
0.01s call     tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(4,)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-3]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-True-list]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-11]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-True-tuple]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-11]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-True-Size]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-6]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_mean at 0x7f6d86b7e8c8>0]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[0.1-0.1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-9]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-3]
0.01s call     tests/infer/test_valid_models.py::test_plate_no_size_ok[Trace_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-18]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-2]
0.01s call     tests/ops/test_contract.py::test_einsum_linear[bi,ia->,i,ia,ib,iab-i]
0.01s call     tests/infer/reparam/test_haar.py::test_uniform[(2, 3, 1)--2-True]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Normal-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_conjugate.py::test_beta_binomial_dependent_sample
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-tuple-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Beta-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(4,)-(5,)]
0.01s call     tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(3, 2)-(3, 2)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Normal-list-sample_shape1]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-1-1-7]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-list-sample_shape0]
0.01s call     tests/ops/test_contract.py::test_ubersum_jit[,i->,i-i]
0.01s call     tests/contrib/gp/test_parameterized.py::test_parameterized
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-Size-sample_shape0]
0.01s call     tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-Adam-optim_kwargs0-jit]
0.01s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-dense-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-19]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(4,)-()]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[0.1-0.5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-17]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-12]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Beta-list-sample_shape0]
0.01s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-flat-2]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[3-1-(3, 2)]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-3]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-()-(5,)]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[1-3-(3, 2)]
0.01s call     tests/ops/test_arrowhead.py::test_utilities[2]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-15]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-()-()]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-6]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[0.1-0.75]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-tuple-sample_shape0]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[1.0-0.25]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Beta-tuple-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-Size-sample_shape0]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-6]
0.01s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-False-sample_shapes0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-list-sample_shape0]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-()-()]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[1.0-0.1]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-2-1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-()-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-4]
0.01s call     tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[functools.partial(<function init_to_value at 0x7f6d86b7e9d8>, values={'x': tensor(3.)})]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(2, 3)-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-17]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-()-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-19]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[1.0-0.5]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-5]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Poisson_1D]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-()-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-9]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-18]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-4]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(4,)-(5,)]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Binary_1D]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Poisson_1D]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-7]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-4]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Poisson_2D]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-8]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[0.1-0.9]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-()-(5,)]
0.01s call     tests/infer/test_enum.py::test_elbo_zip[0.1-0.25]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Binary_2D]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(3, 2)-(5,)]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-()-()]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(2, 3)-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-Size-sample_shape0]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-1-1-1]
0.01s call     tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalGP_y1D]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-18]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_hmm[2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-5]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-auto_class6]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-5]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-()-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-9]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-6]
0.01s call     tests/infer/test_enum.py::test_elbo_normal[None-loss_and_grads]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(4,)-()]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-8]
0.01s call     tests/infer/test_enum.py::test_elbo_normal[None-differentiable_loss]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(2, 3)-3]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-5]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(2, 3)-3]
0.01s call     tests/infer/reparam/test_haar.py::test_uniform[(6,)--1-True]
0.01s call     tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-0-expand_shape0]
0.01s call     tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalGP_y2D]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-19]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-17]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(5,)-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-list-sample_shape0]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-8]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-11]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-19]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-8]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-8]
0.01s call     tests/infer/reparam/test_stable.py::test_subsample_smoke[LatentStableReparam-full]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-6]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-8]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-list-sample_shape0]
0.01s call     tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalGP_y1D]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-17]
0.01s call     tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-0-expand_shape1]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(5,)-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-8]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(4,)-(5,)]
0.01s call     tests/ops/test_contract.py::test_adjoint_marginal[a,abi->-i]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-auto_class6]
0.01s call     tests/infer/reparam/test_haar.py::test_uniform[(2, 3, 1)--2-False]
0.01s call     tests/pyroapi/test_pyroapi.py::test_plate_ok[minipyro-py]
0.01s call     tests/distributions/test_distributions.py::test_expand_error[Stable-False-initial_shape2-proposed_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-3]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_MultiClass_1D]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-Size-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-10]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-tuple-sample_shape2]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[3-3-(4,)]
0.01s call     tests/infer/test_valid_models.py::test_not_has_rsample_ok[Trace_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-Size-sample_shape2]
0.01s call     tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-list-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(5,)-3]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-4]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-list-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-18]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-10]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-Size-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-16]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(5,)-3]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-5]
0.01s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-(2, 3)-(2, 3)]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-()-(5,)]
0.01s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-(2, 1)-(3,)]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(3, 2)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-4]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-tuple-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-6]
0.01s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-(2, 1)-(3,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(4,)-()]
0.01s call     tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[Trace_ELBO-subsample]
0.01s call     tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[Trace_ELBO-full]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-1]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(3, 2)-(6, 5)]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(3, 2)-()]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-parallel-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-18]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-1]
0.01s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape3-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-Size-sample_shape1]
0.01s call     tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-10]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-10]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(2, 3)-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-8]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(3, 2)-(6, 5)]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(3, 2)-(6, 5)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-5]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-1-1-7]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(2, 3)-3]
0.01s call     tests/ops/test_ssm_gp.py::test_matern_kernel[2.5-3]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoDelta]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(5,)-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-list-sample_shape2]
0.01s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[3]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-6]
0.01s call     tests/distributions/test_mask.py::test_mask[0-1-1]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[2-1-(3, 2)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-17]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-19]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(5,)-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-1-expand_shape2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[1]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(4,)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-tuple-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-16]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-list-sample_shape2]
0.01s call     tests/ops/test_newton.py::test_newton_step[3-None-batch_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-tuple-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(5,)-3]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(3, 2)-(6, 5)]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoLaplaceApproximation]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Dirichlet-list-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Dirichlet-tuple-sample_shape2]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[2-3-(4,)]
0.01s call     tests/ops/test_gamma_gaussian.py::test_logsumexp[2-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-()-(5,)]
0.01s call     tests/infer/mcmc/test_valid_models.py::test_all_discrete_sites_log_prob[TraceEinsumEvaluator]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-3]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-6]
0.01s call     tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-4]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Dirichlet-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(2, 3)-2]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[sequential-loss-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-16]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-1-1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-14]
0.01s call     tests/infer/test_valid_models.py::test_set_has_rsample_ok[Trace_ELBO-False]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-17]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(2, 3)-2]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(5,)-2]
0.01s call     tests/infer/test_enum.py::test_elbo_bern[sequential-loss-10]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-18]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(2, 3)-2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(2, 3)-2]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-TraceGraph_ELBO]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(4,)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-6]
0.01s call     tests/ops/einsum/test_adjoint.py::test_marginal[w,wx,x->]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Cauchy-tuple-sample_shape2]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Binary_1D]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Cauchy-list-sample_shape2]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoLaplaceApproximation]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-9]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-7]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-7]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Cauchy-Size-sample_shape2]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(3, 2)-(6, 5)]
0.01s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceEinsumEvaluator-data2-expected_log_prob2]
0.01s call     tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-3]
0.01s call     tests/ops/test_contract.py::test_naive_ubersum[,i,i->,i-i]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-3]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-()-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-5]
0.01s call     tests/distributions/test_transforms.py::TransformTests::test_lower_cholesky_affine
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[Trace_ELBO-subsample]
0.01s call     tests/infer/reparam/test_stable.py::test_subsample_smoke[LatentStableReparam-subsample]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[3-2-(4,)]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-()-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-9]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-8]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(2, 3)-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(3, 2)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-3]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(3, 2)-(6, 5)]
0.01s call     tests/ops/test_arrowhead.py::test_utilities[5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-19]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(4,)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-2]
0.01s call     tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Binary_2D]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(5,)-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Normal-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-list-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-6]
0.01s call     tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoDelta]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(5,)-2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-15]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-19]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-tuple-sample_shape0]
0.01s call     tests/distributions/test_conjugate_update.py::test_gamma_poisson[(3, 2)-(3, 2)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Normal-tuple-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-17]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-list-sample_shape1]
0.01s call     tests/nn/test_module.py::test_torch_serialize_decorators
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-15]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-()-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Exponential-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-()-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Normal-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(4,)-()]
0.01s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-20-iplate]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-9]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Exponential-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-4]
0.01s call     tests/ops/test_linalg.py::test_sym_rinverse[False-A2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Exponential-tuple-sample_shape1]
0.01s call     tests/ops/test_gamma_gaussian.py::test_reshape[1-(5, 6)-(5, 3, 2)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-19]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-17]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-18]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(2, 3)-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(3, 2)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-()-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-18]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-()-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-tuple-sample_shape2]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[1-2-(3, 2)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-Size-sample_shape2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-list-sample_shape2]
0.01s call     tests/infer/test_enum.py::test_compute_marginals_restrictions[False-None-2-True]
0.01s call     tests/ops/test_linalg.py::test_sym_rinverse[True-A2]
0.01s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceEinsumEvaluator-data0-expected_log_prob0]
0.01s call     tests/distributions/test_tensor_type.py::test_conflicting_types
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-1]
0.01s call     tests/contrib/gp/test_kernels.py::test_transforming
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-18]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-4]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-()-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-15]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-3]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-()-3]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-()-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(5,)-2]
0.01s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 2)-(4, 5)-9]
0.01s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-None-iplate]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(5,)-2]
0.01s call     tests/ops/test_newton.py::test_newton_step[3-None-batch_shape1]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-()-3]
0.01s call     tests/ops/test_gaussian.py::test_reshape[2-(5, 6)-(5, 3, 2)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-tuple-sample_shape1]
0.01s call     tests/ops/test_gaussian.py::test_logsumexp[2-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-Size-sample_shape1]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-()-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(4,)-()]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-()-()]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-()-3]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-tuple-sample_shape1]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape0-df_shape1]
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[Trace_ELBO-full]
0.01s call     tests/distributions/test_mask.py::test_mask[1-2-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(3, 2)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-4]
0.01s call     tests/distributions/test_mvn.py::test_shape[5-cov_shape2-loc_shape1]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-1-1-1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-tuple-sample_shape1]
0.01s call     tests/ops/test_contract.py::test_ubersum[ab->,a,b,ab,ba-]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-list-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-list-sample_shape0]
0.01s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceEinsumEvaluator-data1-expected_log_prob1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(4,)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-4]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-()-()]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceGraph_ELBO-subsample]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-Size-sample_shape1]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-()-()]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(2, 3)-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape2-df_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-18]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[2-2-(4,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-14]
0.01s call     tests/poutine/test_poutines.py::QueueHandlerDiscreteTest::test_queue_single
0.01s call     tests/distributions/test_mask.py::test_mask[1-2-1]
0.01s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-20-iplate]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-list-sample_shape1]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceGraph_ELBO-subsample]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-2]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape2-df_shape0]
0.01s call     tests/infer/test_valid_models.py::test_no_log_prob_ok[EnergyDistance_noprior]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-()-(3,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-4]
0.01s call     tests/contrib/test_minipyro.py::test_plate_ok[pyro-py]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape2-df_shape2]
0.01s call     tests/test_generic.py::test_rng_state[minipyro-logistic_regression]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(3, 2)-()]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-tuple-sample_shape2]
0.01s call     tests/test_generic.py::test_rng_state[pyro-logistic_regression]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-15]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceGraph_ELBO-subsample]
0.01s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[BetaBinomial]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-Size-sample_shape2]
0.01s call     tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[minipyro-py]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-8]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-()-(3,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-16]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-list-sample_shape2]
0.01s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-None-iplate]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(5,)-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-7]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-()-(3,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(3, 2)-(5,)]
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceGraph_ELBO-subsample]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(4,)-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-4]
0.01s call     tests/poutine/test_mapdata.py::test_replay[iplate-20]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[3-1-(4,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Uniform-list-sample_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-10]
0.01s call     tests/contrib/gp/test_conditional.py::test_conditional[3]
0.01s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[ExtendedBetaBinomial]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-17]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape2-df_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Uniform-Size-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-9]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-9]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-13]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-13]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-()-(3,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-()-(5,)]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Uniform-tuple-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-()-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-19]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-5]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceGraph_ELBO-subsample]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceGraph_ELBO-subsample]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-2]
0.01s call     tests/ops/test_contract.py::test_ubersum_5[ubersum]
0.01s call     tests/infer/test_autoguide.py::test_guide_list[AutoIAFNormal]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape0-df_shape0]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-()-2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-()-2]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-7]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-13]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-7]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-3]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-()-2]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-5]
0.01s call     tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceGraph_ELBO-full]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-(2, 3)-(2, 3)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-6]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-()-2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-7]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-3]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-7]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-14]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-3]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-()-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-4]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-()-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-Size-sample_shape0]
0.01s call     tests/distributions/test_mask.py::test_mask[0-1-0]
0.01s call     tests/distributions/test_mvn.py::test_shape[5-cov_shape2-loc_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-6]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-5]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape1-df_shape1]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-7]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(2, 3)-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-()-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-4]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-()-2]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-2]
0.01s call     tests/infer/test_smcfilter.py::test_systematic_sample[31]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-12]
0.01s call     tests/infer/reparam/test_conjugate.py::test_beta_binomial_static_sample
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-()-(5,)]
0.01s call     tests/distributions/test_mvn.py::test_shape[5-cov_shape2-loc_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-()-()]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(4,)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-tuple-sample_shape0]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-7]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_mvn.py::test_shape[5-cov_shape1-loc_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-17]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-5]
0.01s call     tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-None-2-True]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-7]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape2-df_shape1]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(5,)-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-4]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-Size-sample_shape1]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-(2, 3)-(2, 3)]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-6]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape0-df_shape2]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-4]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[1-3-(4,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-list-sample_shape1]
0.01s call     tests/contrib/test_minipyro.py::test_constraints[pyro-py]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape1-df_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-13]
0.01s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape3-sample_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-8]
0.01s call     tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-1]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape1-df_shape2]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceGraph_ELBO-full]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-4]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(4,)-()]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(3, 2)-()]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape0-df_shape2]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-11]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-Size-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-7]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-5]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(3, 2)-(7,)]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-(2, 3)-(2, 3)]
0.01s call     tests/infer/test_enum.py::test_elbo_normal[None-loss]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-list-sample_shape0]
0.01s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-(2, 3)-(2, 3)]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-15]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-16]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(3, 2)-()]
0.01s call     tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceGraph_ELBO-True]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(4,)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-1]
0.01s call     tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[pyro-py]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-tuple-sample_shape0]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape2-df_shape2]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceGraph_ELBO-full]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-()-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-2]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(3, 2)-(6, 5)]
0.01s call     tests/infer/mcmc/test_valid_models.py::test_enumeration_in_dag[TraceTreeEvaluator]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-()-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-2]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape1-df_shape2]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceGraph_ELBO-full]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[2-1-(4,)]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceGraph_ELBO-full]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-2]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-()-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(3, 2)-(5,)]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(4,)-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-13]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-Size-sample_shape0]
0.01s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-TraceTMC_ELBO]
0.01s call     tests/distributions/test_mask.py::test_mask[2-1-1]
0.01s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceGraph_ELBO-full]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-18]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-tuple-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-()-(5,)]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-5]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-6]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-14]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-6]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-2]
0.01s call     tests/infer/reparam/test_split.py::test_normal[()-(2, 3, 1)-[1, 2]--2]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-4]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-9]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-10]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-list-sample_shape1]
0.01s call     tests/infer/test_valid_models.py::test_plate_ok[TraceGraph_ELBO-subsample]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(3, 2)-()]
0.01s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 1)-(4, 5)-9]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-6]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Dirichlet-Size-sample_shape1]
0.01s call     tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[Trace_ELBO-None]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(3, 2)-()]
0.01s call     tests/contrib/gp/test_conditional.py::test_conditional[4]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-Size-sample_shape1]
0.01s call     tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-TraceGraph_ELBO]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Exponential-Size-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-5]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-17]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-19]
0.01s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(2,)-(4, 5)-9]
0.01s call     tests/contrib/gp/test_conditional.py::test_conditional_whiten[3]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-tuple-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-Size-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-2]
0.01s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-5]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-5]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Categorical-Size-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-17]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-list-sample_shape1]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(3, 2)-(7,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-tuple-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-list-sample_shape0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-12]
0.01s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(2,)-(4, 5)-9]
0.01s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 2)-(4, 5)-9]
0.01s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 2)-(4, 5)-9]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Gamma-list-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-1]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(3, 2)-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-tuple-sample_shape1]
0.01s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[RationalQuadratic]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Categorical-list-sample_shape0]
0.01s call     tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-2-NcvContinuous]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-tuple-sample_shape0]
0.01s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-3-2-7]
0.01s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 1)-(4, 5)-9]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-3]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-Size-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Gamma-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-list-sample_shape0]
0.01s call     tests/distributions/test_mvn.py::test_shape[5-cov_shape0-loc_shape2]
0.01s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Binomial]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Gamma-tuple-sample_shape1]
0.01s call     tests/infer/test_discrete.py::test_plate_smoke[2-sample]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-Size-sample_shape1]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Exponential-list-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Exponential-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-3]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-()-()]
0.01s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Dirichlet-list-sample_shape1]
0.01s call     tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-1-1]
0.01s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 2)-(4, 5)-9]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Dirichlet-tuple-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-11]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[Categorical-tuple-sample_shape0]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-Size-sample_shape0]
0.01s call     tests/infer/test_smcfilter.py::test_systematic_sample[30]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-(3,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-13]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-19]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-6]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-19]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-19]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-()-(5,)]
0.01s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-()-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-tuple-sample_shape0]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-6]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-(3,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-9]
0.01s call     tests/infer/test_valid_models.py::test_plate_ok[TraceGraph_ELBO-full]
0.01s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-()-(4, 5)-9]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[1-2-(4,)]
0.01s call     tests/ops/test_gaussian.py::test_affine_normal[1-1-(3, 2)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-12]
0.01s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-(3,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-17]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-17]
0.01s call     tests/distributions/test_mask.py::test_mask[1-2-0]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-7]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-18]
0.01s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-dense-2]
0.01s call     tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-3-NcvContinuous]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-12]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-scale3]
0.01s call     tests/infer/test_discrete.py::test_hmm_smoke[sample-2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-15]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-18]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(4,)-(5,)]
0.01s call     tests/infer/reparam/test_split.py::test_normal[()-(4, 2)-[1, 3]--2]
0.01s call     tests/contrib/gp/test_conditional.py::test_conditional_whiten[2]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-4]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-4]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-mask2]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-5]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(3, 2)-(6, 5)]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-14]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(4,)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-17]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-True-Size]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape2-df_shape1]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape1-df_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-7]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-3]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-18]
0.01s call     tests/infer/reparam/test_split.py::test_normal[()-(2, 5)-[2, 3]--1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-10]
0.01s call     tests/ops/test_contract.py::test_einsum_linear[,i,i,ia,ia->,i,ia-i]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(3, 2)-()]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-8]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-15]
0.01s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-16]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-()-(5,)]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape2-df_shape2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-19]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-5]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-14]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-11]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-True-tuple]
0.01s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(3, 2)-(7,)]
0.01s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[ExtendedBinomial]
0.01s call     tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-2-NcpContinuous]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-18]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-13]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0--1.0-1.0]
0.01s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-flat-2]
0.01s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-True-list]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-6]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-12]
0.01s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-()-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-5]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape2-df_shape0]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(4,)-()]
0.01s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-4]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-mask4]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-3]
0.01s call     tests/ops/test_contract.py::test_einsum_linear[ab->,a,b,ab,ba-]
0.01s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-2]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(3, 2)-(5,)]
0.01s call     tests/infer/test_smcfilter.py::test_systematic_sample[29]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-True]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-19]
0.01s call     tests/poutine/test_poutines.py::ReplayHandlerTests::test_replay_full_repeat
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-4]
0.01s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-9]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-9]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0--1.0-1.0]
0.01s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[1-size1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-15]
0.01s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(3, 2)-(6, 5)]
0.01s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-False]
0.01s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-scale2]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-16]
0.01s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape0-df_shape2]
0.01s call     tests/infer/test_valid_models.py::test_reparam_ok[EnergyDistance_prior]
0.01s call     tests/ops/test_stats.py::test_resample[True]
0.01s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-()-(5,)]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-list-sample_shape1]
0.01s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-11]
0.01s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-Size-sample_shape1]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-2]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-1]
0.01s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-6]
0.01s call     tests/test_generic.py::test_rng_seed[pyro-beta_binomial]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-2]
0.01s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-18]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-tuple-sample_shape1]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-1]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-mask3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-10]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-13]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-17]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-2]
0.00s call     tests/infer/test_valid_models.py::test_no_log_prob_ok[EnergyDistance_prior]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(4,)-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-8]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-12]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0-1.0--1.0]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-3]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0-1.0-1.0]
0.00s call     tests/infer/test_valid_models.py::test_tail_adaptive_ok
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-13]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0--1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0--1.0--1.0]
0.00s call     tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-3-NcvContinuous]
0.00s call     tests/contrib/gp/test_conditional.py::test_conditional_whiten[0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-11]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-()-()]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0-1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0--1.0--1.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-True-Size]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-True-list]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0--1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0-1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0--1.0--1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-16]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-17]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0--1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0-1.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-14]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0-1.0--1.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-True-tuple]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0-1.0--1.0]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-0.1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0--1.0-1.0]
0.00s call     tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape2-False]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-15]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-16]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-()-(5,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0--1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0-1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0-1.0-1.0]
0.00s call     tests/distributions/test_mask.py::test_mask[2-1-0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-19]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0--1.0--1.0]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 1)-(4, 5)-9]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0-1.0--1.0]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape3-sample_shape1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0--1.0--1.0]
0.00s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-()-()]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0-1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0-1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0--1.0-1.0]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-False-sample_shapes0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0-1.0--1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-12]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-4]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0--1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0-1.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(4,)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0-1.0--1.0]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(4,)-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-13]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0--1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0-1.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-17]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(3, 2)-(6, 5)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-Size-sample_shape0]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-3-2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0--1.0--1.0]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-15]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-7]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-()-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-19]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-14]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-TraceEnum_ELBO]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-18]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-10]
0.00s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-10]
0.00s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[8-size0]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-5]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-3-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_logsumexp[1-()]
0.00s call     tests/infer/test_valid_models.py::test_plate_no_size_ok[TraceGraph_ELBO]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-16]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-18]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-11]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(3, 2)-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-list-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-11]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-tuple-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-9]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[abcd]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-True-list]
0.00s call     tests/ops/test_contract.py::test_ubersum_jit[a->,a-]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-5]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-Size-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-11]
0.00s call     tests/poutine/test_mapdata.py::test_cond_indep_stack[nested_iplate-5]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-True-tuple]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-10]
0.00s call     tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[2]
0.00s call     tests/infer/reparam/test_split.py::test_normal[()-(6,)-[2, 1, 3]--1]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[11-(2, 3)-(4, 5)-9]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[28]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-12]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-True-Size]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(3, 2)-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-list-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-17]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(2, 3)-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-9]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-tuple-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-17]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(2, 3)-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-11]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-14]
0.00s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-11]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-1]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[11-(6,)-(4, 5)-9]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(2, 3)-3]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-13]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-19]
0.00s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-2]
0.00s call     tests/infer/test_valid_models.py::test_not_has_rsample_ok[TraceGraph_ELBO]
0.00s call     tests/poutine/test_poutines.py::test_replay_enumerate_poutine[2--3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-10]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[acbd]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-18]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Cauchy-tuple-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(4,)-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-15]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-19]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-12]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-8]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-()-(5,)]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[marginal-a,abi,bcij,bdik->-ijk]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Cauchy-list-sample_shape1]
0.00s call     tests/poutine/test_poutines.py::test_replay_enumerate_poutine[2--2]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[abdc]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(2, 3)-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(5,)-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Cauchy-Size-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-8]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-4]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(5,)-3]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-3-1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 2)-(4, 5)-9]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(4,)-()]
0.00s call     tests/infer/test_enum.py::test_elbo_bern[parallel-differentiable_loss-10]
0.00s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 1)-(4, 5)-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(5,)-3]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 2)-(4, 5)-9]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(2, 3)-2]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 2)-(4, 5)-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-18]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-True-sample_shapes0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-14]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-1-2]
0.00s call     tests/distributions/test_conjugate_update.py::test_beta_binomial[(4,)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-()-()]
0.00s call     tests/distributions/test_mask.py::test_mask[1-1-1]
0.00s call     tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape2-autocorrelation]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-(2, 1)-(3,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-(3,)-()]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(3, 2)-(7,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-False-Size]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 2)-(4, 5)-9]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-True-sample_shapes0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(2, 3)-2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-6]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(5,)-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(5,)-2]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-3-2]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-3-2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(3, 2)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(5,)-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-9]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-False-list]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(2,)-(4, 5)-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-14]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Gamma-tuple-sample_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(2,)-(4, 5)-9]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceGraph_ELBO-full]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-18]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-11]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-False-tuple]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-()-(4, 5)-9]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-3-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-10]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-False-Size]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-8]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-19]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-(3,)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-False-tuple]
0.00s call     tests/infer/test_enum.py::test_elbo_bern[parallel-differentiable_loss-1]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-3-2]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceGraph_ELBO-subsample]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-tuple-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-False-list]
0.00s call     tests/poutine/test_poutines.py::test_replay_enumerate_poutine[2--1]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-8]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-4]
0.00s call     tests/infer/test_enum.py::test_elbo_bern[parallel-loss_and_grads-10]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[27]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-12]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-16]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(3, 2)-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-Size-sample_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-6]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-8]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-list-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Gamma-Size-sample_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-6]
0.00s call     tests/infer/test_discrete.py::test_hmm_smoke[map-2]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-Trace_ELBO-full]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-Size-sample_shape0]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-4]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-13]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-3-2]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-3-2]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-3-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-list-sample_shape1]
0.00s call     tests/test_generic.py::test_model_sample[pyro-beta_binomial]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-mask2]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-3-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-2]
0.00s call     tests/infer/test_enum.py::test_iter_discrete_traces_scalar[flat]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Gamma-list-sample_shape0]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-10]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-12]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(3, 2)-(6, 5)]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-4]
0.00s call     tests/contrib/tracking/test_ekf.py::test_EKFState_with_NcvContinuous
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-6]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-3-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-tuple-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-5]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[VonMises]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-5]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwxj,ixyj,iyzj,izwj->ji]
0.00s call     tests/infer/test_enum.py::test_iter_discrete_traces_scalar[dense]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-9]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-3-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-(2, 3)-(2, 3)]
0.00s call     tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-2-NcpContinuous]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-13]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-()-2]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-3-1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-()-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-15]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-3-1]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-3-1]
0.00s call     tests/params/test_param.py::test_dict_interface
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-8]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-1]
0.00s call     tests/ops/test_gaussian.py::test_affine_normal[1-1-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-8]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Matern52]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-9]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-()-(5,)]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s call     tests/infer/test_enum.py::test_elbo_bern[parallel-loss_and_grads-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-4]
0.00s call     tests/contrib/gp/test_conditional.py::test_conditional[0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-15]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-8]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[3-0.1]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-2]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-3-1]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s call     tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape2-autocovariance]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwxj,ixyj,iyzj,izwj->ji]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-Trace_ELBO-subsample]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-1]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-3-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-10]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-17]
0.00s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-10]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[adbc]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-13]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-11]
0.00s call     tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-3-NcpContinuous]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-1-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(4,)-()]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(3, 2)-()]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-True]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-9]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-mask4]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-()-(5,)]
0.00s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-10]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[cadb]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwxj,ixyj,iyzj,izwj->ij]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-3]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(4,)-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-2]
0.00s call     tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceGraph_ELBO-False]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-14]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-15]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-10]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[3-1.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-()-(3,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(4,)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-()-3]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-1-2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(4,)-(7,)]
0.00s call     tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape1-_cummin]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-16]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(4,)-()]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-11]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-()-3]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[3-10.0]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-3]
0.00s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-10]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-False-sample_shapes0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-19]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-16]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Matern32]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-1-2]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-scale2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-11]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-()-(3,)]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[acdb]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-()-3]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-1-2]
0.00s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-10]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[bcda]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-17]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-14]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-10]
0.00s call     tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape1-False]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwxj,ixyj,iyzj,izwj->ij]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-18]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-5]
0.00s call     tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-10]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-5]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[26]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-1-1]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[bdca]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-1-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-Size-sample_shape1]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-3]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(3, 2)-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-list-sample_shape0]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-4]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-1-1]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-tuple-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-12]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-()-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-4]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-Size-sample_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-(3,)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-4]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-scale3]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(4,)-(6, 5)]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-16]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[cabd]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[cbad]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-6]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-1-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-(3,)-()]
0.00s call     tests/ops/test_contract.py::test_naive_ubersum[abi,b->,b,ai,abi-i]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[bdac]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(4,)-()]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-mask3]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-6]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[cdab]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-12]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[cbda]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-()-2]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-1-1]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[dcba]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-()-()]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[badc]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-10]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-False]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-tuple-sample_shape1]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[bcad]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(3, 2)-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-()-()]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[dacb]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[dabc]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-list-sample_shape1]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[dcab]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(3, 2)-(6, 5)]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[adcb]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(4,)-(5,)]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-2]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(4,)-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(4,)-()]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-1-1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-()-()]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[dbca]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[cdba]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-1-1]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[bacd]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-()-(3,)]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-()-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Uniform-Size-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Uniform-list-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-12]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-()-()]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-1-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(3, 2)-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Uniform-tuple-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-6]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[dbac]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(4,)-(7,)]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-3]
0.00s call     tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-5]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-()-(5,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-6]
0.00s call     tests/ops/test_arrowhead.py::test_utilities[0]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_model_error_stray_batch_dims[NUTS-kwargs1]
0.00s call     tests/infer/test_jit.py::test_scatter_workaround
0.00s call     tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalGP_y1D]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Exponential]
0.00s call     tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceGraph_ELBO-subsample]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-()-()]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-()-(5,)]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-5]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-5]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-4]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-17]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Cosine]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-2]
0.00s call     tests/ops/test_contract.py::test_naive_ubersum[ab->,a,b,ab,ba-]
0.00s call     tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalSparseGP_y1D]
0.00s call     tests/ops/test_ssm_gp.py::test_matern_kernel[2.5-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-9]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-()-(5,)]
0.00s call     tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalSparseGP_y2D]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[25]
0.00s call     tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-Adam-optim_kwargs0-py]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-15]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-13]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-(2, 1)-(3,)]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-4]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixyj,iyzj->ij]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-10]
0.00s call     tests/contrib/gp/test_conditional.py::test_conditional[2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-()-(3,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-18]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-11]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-()-(3,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-19]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-1]
0.00s call     tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->a-ij]
0.00s call     tests/contrib/gp/test_kernels.py::test_combination
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-()-(3,)]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-Size-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-tuple-sample_shape0]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixyj,iyzj->ji]
0.00s call     tests/contrib/gp/test_conditional.py::test_conditional[1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-10]
0.00s call     tests/contrib/tracking/test_assignment.py::test_dense_smoke
0.00s call     tests/contrib/test_util.py::test_rdiag
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-19]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(4,)-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-list-sample_shape0]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-8]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Cauchy-list-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Cauchy-Size-sample_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-8]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-14]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-3]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-8]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-3-2]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-0.1]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-3-2]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[RBF]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Cauchy-tuple-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[OneHotCategorical]
0.00s call     tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceGraph_ELBO-full]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(4,)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-18]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-()-()]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-8]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-3-2]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-3-1]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-3-2]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-8]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-()-()]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape3-sample_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-8]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(4,)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(4,)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-8]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(4,)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(4,)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-9]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-3-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-16]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-2]
0.00s call     tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalGP_y2D]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-3-2]
0.00s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[7-size0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-12]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(3, 2)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(4,)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-9]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-1-1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(4,)-(6, 5)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-8]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-10]
0.00s call     tests/distributions/test_mvn.py::test_shape[3-cov_shape2-loc_shape0]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(4,)-(6, 5)]
0.00s call     tests/infer/test_discrete.py::test_plate_smoke[2-map]
0.00s call     tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape2-split_gelman_rubin]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-8]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-(3,)-()]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-3]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape3-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-1]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-1-2]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 2)-(4, 5)-9]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-15]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-13]
0.00s call     tests/infer/test_autoguide.py::test_guide_list[AutoLowRankMultivariateNormal]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-5]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-7]
0.00s call     tests/ops/test_newton.py::test_newton_step[2-None-batch_shape1]
0.00s call     tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape0-True]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-()-()]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[23]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-()-(5,)]
0.00s call     tests/ops/test_newton.py::test_newton_step[2-None-batch_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-8]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-1-2]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-1-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(3, 2)-(7,)]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[24]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-11]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-1-2]
0.00s call     tests/contrib/tracking/test_ekf.py::test_EKFState_with_NcpContinuous
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-8]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-Size-sample_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-8]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-()-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-3]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-14]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-13]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-1-1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(4,)-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-13]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixyj,iyzj->ji]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-tuple-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-13]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-list-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-10]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 2)-(4, 5)-9]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[5-(2, 3)-(4, 5)-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-16]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-11]
0.00s call     tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-17]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-1-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-3]
0.00s call     tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-12]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-15]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-11]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data3-expected_log_prob3]
0.00s call     tests/distributions/test_mask.py::test_mask[1-1-0]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(2,)-(4, 5)-9]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-wxi,xyi,yzi,zwi->i]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-15]
0.00s call     tests/ops/test_contract.py::test_einsum_linear[,i,ia->,i,ia-i]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(4,)-()]
0.00s call     tests/ops/test_contract.py::test_adjoint_marginal[ab,bc->-]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-9]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(4,)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-4]
0.00s call     tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-ClippedAdam-optim_kwargs1-py]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_reshape[3-(6,)-(3, 2)]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_all_discrete_sites_log_prob[TraceTreeEvaluator]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixyj,iyzj->ij]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-4]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(4,)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-14]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-19]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(4,)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(4,)-(6, 5)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-True-tuple]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-11]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[5-(6,)-(4, 5)-9]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 2)-(4, 5)-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-15]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-10]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-()-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-9]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 1)-(4, 5)-9]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(4,)-()]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[22]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape2-df_shape1]
0.00s call     tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceGraph_ELBO-None]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-True-Size]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-()-(4, 5)-9]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(4,)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-14]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-True-list]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(4,)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-14]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-10]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(2,)-(4, 5)-9]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape0-df_shape0]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-18]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-11]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-7]
0.00s call     tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape0-False]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape2-df_shape2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(4,)-(7,)]
0.00s call     tests/ops/test_special.py::test_log_beta_stirling[0.10]
0.00s call     tests/infer/test_valid_models.py::test_plate_enum_discrete_no_discrete_vars_warning[True]
0.00s setup    tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_nonreparameterized_nn_baseline
0.00s call     tests/distributions/test_mvn.py::test_shape[3-cov_shape1-loc_shape2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(4,)-(6, 5)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-wxi,xyi,yzi,zwi->i]
0.00s call     tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[pyro-py]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape2-df_shape0]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-13]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-7]
0.00s call     tests/ops/test_contract.py::test_adjoint_marginal[ai,ai->i-i]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-15]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Periodic]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-(3,)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-12]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-wx,xy,yz,zw->]
0.00s call     tests/distributions/test_mvn.py::test_shape[3-cov_shape2-loc_shape2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(4,)-(6, 5)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 2)-(4, 5)-9]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape2-df_shape2]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_model_error_stray_batch_dims[HMC-kwargs0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-7]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-10]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-16]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-11]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(3, 2)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(3, 2)-(6, 5)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-13]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwj,ixj->ij]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 1)-(4, 5)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-8]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Uniform-Size-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-Size-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-()-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-12]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-Size-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-12]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape1-df_shape2]
0.00s call     tests/distributions/test_mvn.py::test_shape[3-cov_shape2-loc_shape1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-15]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-()-()]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape2-df_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[3-(6,)-(4, 5)-9]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(3, 2)-(7,)]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-list-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-14]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape1-df_shape1]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape2-df_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-14]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape0-df_shape2]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-(3,)-()]
0.00s call     tests/contrib/tracking/test_assignment.py::test_sparse_smoke
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-tuple-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-10]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(3, 2)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-16]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Uniform-list-sample_shape0]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape0-df_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-16]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-4]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-16]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-12]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Uniform-tuple-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-9]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-13]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-8]
0.00s call     tests/infer/test_valid_models.py::test_enum_recycling_interleave
0.00s call     tests/contrib/test_minipyro.py::test_constraints[minipyro-py]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-()-(3,)]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape1-df_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(3, 2)-(7,)]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape0-df_shape2]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-13]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,wxi,xyi,yzi->i]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape1-df_shape2]
0.00s call     tests/distributions/test_conjugate_update.py::test_beta_binomial[(3, 2)-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-11]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-11]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-False-Size]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-False-tuple]
0.00s call     tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(4,)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-12]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-5]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-False-list]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-15]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-7]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-5]
0.00s call     tests/ops/test_ssm_gp.py::test_matern_kernel[2.5-1]
0.00s call     tests/test_generic.py::test_rng_seed[pyro-logistic_regression]
0.00s call     tests/distributions/test_mvn.py::test_shape[1-cov_shape0-loc_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-5]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-5]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(2,)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-12]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[3-(2, 3)-(4, 5)-9]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-Size-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-14]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 1, 1)-(7,)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwj,ixj->ji]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(4,)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-10]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(4,)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-5]
0.00s call     tests/infer/test_autoguide.py::test_guide_list[AutoMultivariateNormal]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-11]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-9]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-14]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-10]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-15]
0.00s call     tests/ops/test_gaussian.py::test_reshape[1-(5, 6)-(5, 3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(4,)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-()-()]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-()-()]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[20]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-list-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-12]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-13]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-16]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(2,)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-16]
0.00s call     tests/infer/test_enum.py::test_elbo_bern[parallel-loss-1]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-5]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-tuple-sample_shape0]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(3, 1)-(7,)]
0.00s call     tests/test_generic.py::test_rng_seed[minipyro-logistic_regression]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-13]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 1, 1)-(7,)]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[4-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-11]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-4]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-15]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-10]
0.00s call     tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-2-NcvContinuous]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-()-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-4]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-13]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-11]
0.00s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[3-size1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-10]
0.00s call     tests/infer/test_enum.py::test_elbo_bern[parallel-loss-10]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-12]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-7]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-15]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-1.0-0.0]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(3, 1)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-()-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-14]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-()-()]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-()-()]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(4,)-()]
0.00s call     tests/ops/test_contract.py::test_ubersum_1[naive_ubersum]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-8]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 3, 2)-(7,)]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-3]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(3, 2)-(6, 5)]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-4]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-()-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(3, 1)-(7,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-tuple-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-3]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-10]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-list-sample_shape0]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal_100-0.1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-16]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-()-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 3, 2)-(7,)]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[21]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 3, 2)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(2,)-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-7]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-7]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-()-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-5]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-6]
0.00s call     tests/test_generic.py::test_trace_handler[pyro-beta_binomial]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-()-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(2,)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(3, 1)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-4]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-7]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-()-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-()-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(3, 2)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(3, 1)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(3, 1)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-16]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-15]
0.00s call     tests/distributions/test_von_mises.py::test_log_prob_normalized[0.01]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-7]
0.00s call     tests/infer/test_jit.py::test_svi_irregular_batch_size[TraceEnum_ELBO]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceGraph_ELBO-subsample]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal_100-1e-08]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape2-df_shape2]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-7]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(2,)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-14]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-4]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-2]
0.00s call     tests/distributions/test_mvn.py::test_shape[3-cov_shape0-loc_shape2]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-12]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[2-(2, 3)-(4, 5)-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-7]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[sample-a,abi,bcij,bdik->-ijk]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-2]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-()-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-5]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-()-()]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-()-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-()-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 3, 2)-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-7]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal_100-0.0001]
0.00s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape1-df_shape0]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-()-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-()-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-5]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-3]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(2,)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(3, 1)-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-7]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(2,)-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-7]
0.00s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_nonreparameterized_N_is_7
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-1.0--1.0]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(2,)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(3, 1)-(7,)]
0.00s call     tests/contrib/tracking/test_dynamic_models.py::test_NcpContinuous
0.00s call     tests/ops/test_gaussian.py::test_affine_normal[3-3-()]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 3, 2)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-0.0--1.0]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 3, 2)-(7,)]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape1-df_shape2]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[2-(6,)-(4, 5)-9]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-1.0-0.0]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape0-df_shape2]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-14]
0.00s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape1-df_shape1]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-mask2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0--1.0-0.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-12]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(3, 1)-(7,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0--1.0--1.0]
0.00s call     tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 3, 2)-(7,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-0.0-0.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-9]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-False]
0.00s call     tests/ops/test_indexing.py::test_value[()-()-(4, 3, 2)-(7,)]
0.00s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape0-df_shape0]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-()-(7,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-0.0-1.0]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(2,)-(7,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0--1.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-0.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0--1.0--1.0]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-(2, 3)-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-8]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0--1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-1.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,wxi,xyi,yzi->i]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-0.0-1.0]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 3, 2)-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(3, 2)-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(3, 2)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-()-(7,)]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-mask3]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-0.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0--1.0-1.0]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceGraph_ELBO-full]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_noprior-scale2]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 3, 2)-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-7]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0--1.0-1.0]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-1-2]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Bernoulli]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape2-df_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-6]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(2,)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-13]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(2,)-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-6]
0.00s call     tests/contrib/tracking/test_dynamic_models.py::test_NcpDiscrete
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0--1.0--1.0]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(4,)-()]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-0.0--1.0]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-7]
0.00s call     tests/ops/test_indexing.py::test_value[()-(2,)-(4, 3, 2)-(7,)]
0.00s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape0-df_shape1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-1.0--1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-5]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-0.0-0.0]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-True]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-16]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-()-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-(2, 1)-(3,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-7]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-0.0-0.0]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-(2, 1)-(3,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-11]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 3, 2)-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(3, 2)-(7,)]
0.00s call     tests/contrib/tracking/test_dynamic_models.py::test_NcvDiscrete
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0--1.0-0.0]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-(2, 3)-(2, 3)]
0.00s call     tests/distributions/test_von_mises.py::test_log_prob_normalized[0.03]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-1.0-0.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-3]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(3, 1)-(7,)]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape2-df_shape0]
0.00s call     tests/distributions/test_mask.py::test_mask[0-0-0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-10]
0.00s setup    tests/integration_tests/test_tracegraph_elbo.py::RaoBlackwellizationTests::test_plate_in_elbo_with_superfluous_rvs
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-5]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[wi,xi->i]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-2]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(3, 1)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-7]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-()-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-True-tuple]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_noprior-0.1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-3]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[ji,iwj->ji]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-8]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-True-Size]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-3-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-5]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-True-list]
0.00s call     tests/ops/test_gaussian.py::test_affine_normal[3-2-()]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-3-2]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-(3,)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-15]
0.00s call     tests/distributions/test_conjugate_update.py::test_gamma_poisson[(4,)-(3, 2)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwj->ij]
0.00s call     tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape2-gelman_rubin]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-12]
0.00s call     tests/ops/test_gaussian.py::test_affine_normal[2-3-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-True-Size]
0.00s call     tests/contrib/tracking/test_dynamic_models.py::test_NcvContinuous
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-16]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-8]
0.00s call     tests/distributions/test_transforms.py::TransformTests::test_elu
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-()-(7,)]
0.00s call     tests/distributions/test_shapes.py::test_one_hot_categorical_log_prob_shape
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwj->ji]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[ij,iwj->ij]
0.00s call     tests/infer/test_valid_models.py::test_enum_in_model_diamond_error[False]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-7]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-6]
0.00s call     tests/contrib/test_util.py::test_tensor_to_dict
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-1]
0.00s call     tests/infer/test_autoguide.py::test_guide_list[AutoNormal]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-5]
0.00s call     tests/ops/test_ssm_gp.py::test_matern_kernel[1.5-3]
0.00s call     tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-3-NcpContinuous]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-6]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(4,)-(6, 5)]
0.00s call     tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[minipyro-jit]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-14]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-5]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-()-(3,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-6]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-3-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-()-(3,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-8]
0.00s call     tests/distributions/test_mvn.py::test_shape[5-cov_shape1-loc_shape1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixyj,iyzj->ji]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-6]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(4,)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_affine_normal[2-2-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-4]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[19]
0.00s call     tests/distributions/test_mvn.py::test_shape[5-cov_shape1-loc_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-7]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_noprior-1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-3]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(4,)-(7,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-list-sample_shape0]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(4,)-(6, 5)]
0.00s call     tests/infer/test_autoguide.py::test_guide_list[AutoDiagonalNormal]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-tuple-sample_shape0]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-()-(3,)]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-2-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_reshape[2-(6,)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-13]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-()-(5,)]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize2]
0.00s call     tests/infer/test_valid_models.py::test_reparam_ok[EnergyDistance_noprior]
0.00s call     tests/ops/test_linalg.py::test_sym_rinverse[False-A1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-2-1-7]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(3, 2)-(7,)]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape2-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-8]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-True-tuple]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[3-None]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-()-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-Size-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-3]
0.00s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape0-df_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-True-list]
0.00s call     tests/ops/test_gaussian.py::test_affine_normal[3-1-()]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-6]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-()-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-2-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_contract.py::test_contract_to_tensor[example7]
0.00s call     tests/ops/test_linalg.py::test_sym_rinverse[True-A1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-2]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-(3,)-()]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[18]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape2-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-True-tuple]
0.00s call     tests/ops/test_contract.py::test_ubersum_collide_ok_2[naive_ubersum]
0.00s call     tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape2-True]
0.00s call     tests/infer/test_valid_models.py::test_enum_in_model_diamond_error[True]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-8]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-10]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-9]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-True-Size]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-16]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-6]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-True-list]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Polynomial]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-6]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-5]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-3-2-1]
0.00s call     tests/ops/test_gaussian.py::test_affine_normal[1-3-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-15]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-False-tuple]
0.00s call     tests/contrib/test_minipyro.py::test_plate_ok[minipyro-py]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-False-list]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-12]
0.00s call     tests/distributions/test_transforms.py::TransformTests::test_leaky_relu
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-False-tuple]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-False-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-13]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-5]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-False-Size]
0.00s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape1-df_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-7]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape3-sample_shape2]
0.00s call     tests/infer/test_inference.py::test_exponential_gamma[rsvi-RenyiELBO]
0.00s call     tests/ops/test_gaussian.py::test_affine_normal[1-2-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-11]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-7]
0.00s call     tests/ops/test_gaussian.py::test_logsumexp[1-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-False-initial_shape0-proposed_shape0]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-10]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-False-list]
0.00s call     tests/poutine/test_poutines.py::IndirectLambdaHandlerTests::test_scale_factors
0.00s call     tests/ops/test_gaussian.py::test_affine_normal[1-1-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-14]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize1]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-True]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-()-()]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize3]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize0]
0.00s call     tests/ops/test_gaussian.py::test_affine_normal[2-1-()]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(3, 2)-(5,)-9]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(3, 2)-(7,)]
0.00s call     tests/contrib/test_util.py::test_rtril
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-11]
0.00s call     tests/ops/test_contract.py::test_ubersum_collide_ok_1[naive_ubersum]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-15]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-6]
0.00s call     tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(3, 2)-(4,)]
0.00s call     tests/distributions/test_mvn.py::test_shape[5-cov_shape0-loc_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-False-list]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-8]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-1-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-4]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-3-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape1-df_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-tuple-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape2-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-False-initial_shape0-proposed_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-Size-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-16]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-Size-sample_shape0]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[i,wi->i]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-5]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-8]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-tuple-sample_shape2]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-1-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-False-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-12]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-Size-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-list-sample_shape0]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(4,)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-6]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(4,)-(6, 5)]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Dirichlet-Size-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-tuple-sample_shape0]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-4]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[2-1.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-False-Size]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-list-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-tuple-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-Size-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Dirichlet-list-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-8]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-list-sample_shape2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(3, 2)-(7,)]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[4-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-14]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Dirichlet-tuple-sample_shape0]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-3]
0.00s call     tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape1-split_gelman_rubin]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-3-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-5]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-2-(3, 2)-(3, 2)]
0.00s call     tests/test_examples.py::test_coverage
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-(2, 3)-(2, 3)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(3, 2)-(7,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-list-sample_shape2]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(4,)-(6, 5)]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-16]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(3, 2)-(7,)]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-mask3]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-(2, 1)-(3,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-False-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-13]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-(2, 1)-(3,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-(2, 1)-(3,)]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-4]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-1-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-False-tuple]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-False-Size]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-1]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_3[Trace_ELBO]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-()-(5,)]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-(2, 3)-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-11]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[17]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[2-10.0]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[wi,wi->i]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-False]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape2-df_shape1]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-2]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[ShapeAugmentedBeta]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(4,)-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-10]
0.00s call     tests/ops/test_ssm_gp.py::test_matern_kernel[1.5-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-1]
0.00s call     tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-mask2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-7]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[16]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-6]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(3, 2)-(5,)-9]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[2-None]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[Trace_ELBO]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_prior-scale2]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[2-0.1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-6]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-6]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-1]
0.00s call     tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[minipyro-py]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-15]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-5]
0.00s call     tests/infer/test_jit.py::test_scale_and_mask
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[TraceGraph_ELBO]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[marginal-ai,abij,acik->i-ijk]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_3[TraceGraph_ELBO]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(4,)-(7,)]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_prior-0.1]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-()-()]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-1]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s call     tests/distributions/test_mask.py::test_mask[2-0-0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-12]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-8]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-14]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-2]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_3[TraceEnum_ELBO]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-6]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[15]
0.00s call     tests/poutine/test_poutines.py::test_replay_enumerate_poutine[1--3]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape6]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(3, 2)-(7,)]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-None-iplate]
0.00s call     tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape1-True]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-20-iplate]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-3]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-6]
0.00s call     tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape1-autocorrelation]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-1-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-Size-sample_shape1]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-16]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-list-sample_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-5]
0.00s call     tests/contrib/oed/test_glmm.py::test_log_prob_matches[model-sigmoid_example-design6]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-6]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[TraceEnum_ELBO]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-7]
0.00s call     tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_prior-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-Size-sample_shape1]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape0-df_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(4,)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-8]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(4,)-(7,)]
0.00s call     tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-1]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(4,)-(6, 5)]
0.00s call     tests/poutine/test_poutines.py::test_replay_enumerate_poutine[1--1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-()-(6, 5)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-list-sample_shape1]
0.00s call     tests/distributions/test_empirical.py::test_sample_examples[sample0-weights0-expected_mean0-expected_var0]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(3, 2)-(5,)-9]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape1-df_shape1]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[TraceTMC_ELBO]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape2-sample_shape2]
0.00s call     tests/ops/test_newton.py::test_newton_step[1-None-batch_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(4,)-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-6]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(3, 2)-(5,)-9]
0.00s call     tests/poutine/test_poutines.py::test_replay_enumerate_poutine[1--2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-2]
0.00s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_nonreparameterized_N_is_5
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-tuple-sample_shape1]
0.00s call     tests/ops/test_ssm_gp.py::test_matern_kernel[1.5-1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(3, 2)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-5]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-list-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-6]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape2-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-tuple-sample_shape1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-5]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-Size-sample_shape1]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape0-df_shape0]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape1-df_shape0]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-False-initial_shape2-proposed_shape2]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-wx,xy,yz,zw->]
0.00s call     tests/ops/test_gamma_gaussian.py::test_reshape[1-(6,)-(3, 2)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-w,wx,xy,yz->]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-tuple-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-False-initial_shape0-proposed_shape0]
0.00s call     tests/ops/test_contract.py::test_ubersum_collide_implemented[naive_ubersum-True]
0.00s call     tests/contrib/test_util.py::test_lexpand
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-Trace_ELBO]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape2-df_shape1]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape0-df_shape0]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape0-df_shape1]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape1-df_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-3]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[marginal-a,abi,bcij->-ij]
0.00s call     tests/ops/test_contract.py::test_adjoint_marginal[a,a->-]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[1-0.1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-6]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoMultivariateNormal-0.0001]
0.00s call     tests/distributions/test_conjugate_update.py::test_gamma_poisson[(3, 2)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-()-(6, 5)]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[14]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-7]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoMultivariateNormal-1e-08]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-()-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-8]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape2-sample_shape2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(4,)-(7,)]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape1-df_shape0]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape2-df_shape2]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape2-df_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-8]
0.00s call     tests/poutine/test_mapdata.py::test_cond_indep_stack[iplate-20]
0.00s call     tests/infer/test_valid_models.py::test_enum_discrete_missing_config_warning[True]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(4,)-(7,)]
0.00s call     tests/infer/test_jit.py::test_multi_output
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,wx,xy,yz->]
0.00s call     tests/contrib/test_util.py::test_rexpand
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-()-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(4,)-(7,)]
0.00s call     tests/infer/test_autoguide.py::test_scores[AutoIAFNormal]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-False-initial_shape2-proposed_shape2]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal-0.1]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape1-df_shape2]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoMultivariateNormal-0.1]
0.00s call     tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-8]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-()-(4, 5)-9]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-3-1]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape1-df_shape2]
0.00s call     tests/distributions/test_unit.py::test_expand[batch_shape2-sample_shape2]
0.00s call     tests/contrib/gp/test_conditional.py::test_conditional[5]
0.00s call     tests/distributions/test_mvn.py::test_shape[1-cov_shape1-loc_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-False-initial_shape2-proposed_shape2]
0.00s call     tests/distributions/test_empirical.py::test_sample_examples[sample1-weights1-expected_mean1-expected_var1]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal-1e-08]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape0-df_shape2]
0.00s call     tests/distributions/test_mvn.py::test_shape[1-cov_shape2-loc_shape0]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[1-1.0]
0.00s call     tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape1-autocovariance]
0.00s call     tests/infer/test_jit.py::test_svi_irregular_batch_size[TraceGraph_ELBO]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape2-df_shape0]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-2-2-(3, 2)]
0.00s call     tests/distributions/test_mvn.py::test_shape[3-cov_shape1-loc_shape1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0--1.0-0.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(4,)-(7,)]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal-0.0001]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0--1.0--1.0]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape0-df_shape2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-8]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-5]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-7]
0.00s call     tests/distributions/test_mask.py::test_mask[1-0-0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-7]
0.00s call     tests/distributions/test_mvn.py::test_shape[1-cov_shape2-loc_shape2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-4]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape2-df_shape2]
0.00s call     tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape1]
0.00s call     tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape0]
0.00s call     tests/distributions/test_mvn.py::test_shape[1-cov_shape2-loc_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-()-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-5]
0.00s call     tests/distributions/test_mvn.py::test_shape[3-cov_shape1-loc_shape0]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s call     tests/distributions/test_conjugate_update.py::test_beta_binomial[(4,)-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-6]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(3, 2)-(5,)-9]
0.00s call     tests/infer/test_jit.py::test_svi_irregular_batch_size[Trace_ELBO]
0.00s call     tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape0]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[13]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Brownian]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 1)-(4, 5)-9]
0.00s call     tests/ops/test_contract.py::test_ubersum_4[ubersum]
0.00s call     tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-False-HMC-_empty_model]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Linear]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-6]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0--1.0-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-0.0-0.0]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 1)-(4, 5)-9]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape3-sample_shape2]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape2-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-2]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape1-df_shape1]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixj->ij]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0--1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-()-(6, 5)]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape0-df_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-8]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-5]
0.00s call     tests/infer/test_jit.py::test_categorical_enumerate[shape2-True]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.05]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(4,)-(7,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-()-()]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0--1.0-0.0]
0.00s call     tests/distributions/test_lkj.py::test_log_prob_eta1[10]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-()-()]
0.00s call     tests/infer/test_jit.py::test_simple
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-()-(6, 5)]
0.00s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[3-size2]
0.00s call     tests/nn/test_module.py::test_bayesian_gru
0.00s call     tests/infer/test_jit.py::test_svi_irregular_batch_size[TraceMeanField_ELBO]
0.00s call     tests/contrib/test_util.py::test_rmv[A0-b0]
0.00s call     tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape0-df_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-6]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-5]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-5]
0.00s call     tests/contrib/timeseries/test_gp.py::test_dependent_matern_gp[3]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-3-2]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-3-2]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape1-df_shape0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-1.0-0.0]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-3-2]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data3-expected_log_prob3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixj->ji]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-3-7]
0.00s call     tests/infer/test_autoguide.py::test_scores[AutoLowRankMultivariateNormal]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-5]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0--1.0-1.0]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape0-df_shape2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-8]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixj->ji]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwxj,ixyj,iyzj,izwj->ji]
0.00s call     tests/ops/test_special.py::test_log_binomial_stirling[0.02]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-1.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-5]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-1.0-1.0]
0.00s call     tests/ops/test_newton.py::test_newton_step[1-None-batch_shape1]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-3-2]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-True-sample_shapes0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-()-(6, 5)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-0.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-8]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-6]
0.00s call     tests/contrib/gp/test_conditional.py::test_conditional[6]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0--1.0--1.0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-5]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape3-sample_shape1]
0.00s call     tests/ops/test_gaussian.py::test_reshape[3-(6,)-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[ShapeAugmentedBeta]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-7]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-3-2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-5]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-3-(3, 2)-(4,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-1.0-1.0]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[3-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-3-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-()-(6, 5)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(4,)-(7,)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixj->ij]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape2-df_shape1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-0.0--1.0]
0.00s call     tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape1-gelman_rubin]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-3]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-0.0--1.0]
0.00s call     tests/infer/test_autoguide.py::test_guide_list[AutoLaplaceApproximation]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.3]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-0.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-0.0-1.0]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape2-df_shape2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0--1.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-1.0-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.05]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-0.0--1.0]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape2-df_shape0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-0.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-1.0--1.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(4,)-(7,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0--1.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0--1.0-1.0]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-3]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-3-(3, 2)-(4,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-0.0-0.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.2]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-4]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-3-2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-1.0-1.0]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape1-df_shape2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-8]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0--1.0-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.1]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-3-1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0--1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-0.0--1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-6]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(4,)-(7,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0--1.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-1.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-0.0-0.0]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-1.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-1.0-0.0]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwxj,ixyj,iyzj,izwj->ij]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0--1.0-1.0]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(4,)-(7,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.2]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-3-1]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-2-(3, 2)-(4,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-0.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-1.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-0.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0--1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0--1.0-0.0]
0.00s call     tests/infer/test_autoguide.py::test_scores[AutoMultivariateNormal]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape3-sample_shape2]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,a,ab,b,b,b,b->a-1]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-2-(3, 2)-(4,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0--1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0--1.0-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.1]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-1.0-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-7]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-0.0-0.0]
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_three_layers
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-0.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-0.0--1.0]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[12]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0--1.0--1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-5]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-4]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-0.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-0.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-0.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-0.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0--1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-0.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-6]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-1.0-0.0]
0.00s call     tests/nn/test_module.py::test_clear
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-1.0--1.0]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-1-1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0--1.0--1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-1.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-0.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0--1.0-0.0]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-3-1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-0.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-6]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-1.0-1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-0.0--1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.05]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0--1.0-0.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-0.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-3-2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0--1.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-3-1]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-3-1]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(3, 2)-(5,)-9]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0--1.0--1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.3]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-()-(7,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-1.0-0.0]
0.00s call     tests/distributions/test_mvn.py::test_shape[3-cov_shape0-loc_shape1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0--1.0-0.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-7]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-0.0--1.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-True-initial_shape1-proposed_shape1]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoDelta]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0--1.0--1.0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-0.0--1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.3]
0.00s call     tests/distributions/test_mvn.py::test_shape[5-cov_shape0-loc_shape0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-1.0--1.0]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-()-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-3-1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-1.0--1.0]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-()-()]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-()-(7,)]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-1.0-0.0]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-()-(6, 5)]
0.00s call     tests/distributions/test_reshape.py::test_reshape[3-0]
0.00s call     tests/infer/test_jit.py::test_bernoulli_enumerate[shape2-True]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-3-1]
0.00s call     tests/ops/test_contract.py::test_ubersum_collide_ok_3[naive_ubersum]
0.00s call     tests/distributions/test_lkj.py::test_log_prob_eta1[2]
0.00s call     tests/contrib/gp/test_models.py::test_model[VariationalSparseGP_y1D]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape2-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-()-()]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[sample-ai,abij,acik->i-ijk]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-True-initial_shape0-proposed_shape0]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape3]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-1-2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(4,)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(4,)-(7,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-True-initial_shape2-proposed_shape2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-8]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,a,ab,b,b,b,b->a-2]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape2-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-3-1]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[1-10.0]
0.00s call     tests/contrib/gp/test_models.py::test_model[VariationalSparseGP_y2D]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-2-(3, 2)-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-3-1]
0.00s call     tests/ops/test_newton.py::test_newton_step_converges[1-1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-8]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-3-(3, 2)-(4,)]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoDiagonalNormal]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape2-sample_shape1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(4,)-(7,)]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-1]
0.00s call     tests/infer/test_jit.py::test_categorical_enumerate[shape1-True]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-()-(6, 5)]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-1-(3, 2)-(4,)]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-3-2]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape1-sample_shape2]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-()-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-1]
0.00s call     tests/ops/test_contract.py::test_ubersum_3[ubersum]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-5]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-1-1]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-1-1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 1)-(4, 5)-9]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoMultivariateNormal]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape0-df_shape0]
0.00s call     tests/infer/test_autoguide.py::test_init_loc_fn[AutoLowRankMultivariateNormal]
0.00s call     tests/distributions/test_conjugate_update.py::test_beta_binomial[()-(3, 2)]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-()-(6, 5)]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoDelta]
0.00s call     tests/infer/test_autoguide.py::test_unpack_latent
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape1-df_shape1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(4,)-(7,)]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[ShapeAugmentedDirichlet]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-1-(3, 2)-(4,)]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape2-sample_shape1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(4,)-(7,)]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoLaplaceApproximation]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-3-2]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-wxi,xyi,yzi,zwi->i]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-()-(7,)]
0.00s call     tests/test_generic.py::test_rng_seed[minipyro-beta_binomial]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-1-2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-()-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-2]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-()-(4, 5)-9]
0.00s call     tests/test_generic.py::test_rng_state[pyro-eight_schools]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-()-(6, 5)]
0.00s call     tests/infer/test_jit.py::test_jit_arange_workaround
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape0-df_shape1]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-1-1]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape2-sample_shape2]
0.00s call     tests/poutine/test_mapdata.py::test_nested_iplate
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 1)-(4, 5)-9]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[w,x->]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-()-(6, 5)]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[11]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-5]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-3-1]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoDiagonalNormal-1e-08]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-4]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-()-(6, 5)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-5]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-3-(4,)-(3, 2)]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-3-1]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-3-(4,)-(3, 2)]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape1-df_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(4,)-(5,)-9]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-5]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-1-2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-()-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-5]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-1-1-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-7]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-2-(4,)-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-1-3-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-5]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-()-()]
0.00s call     tests/test_generic.py::test_rng_state[minipyro-beta_binomial]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-8]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-4]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-3]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-5]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-()-()]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-5]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-1-2]
0.00s call     tests/contrib/gp/test_models.py::test_model[SparseGPRegression_y1D]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-2-(4,)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-1-2]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[10]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-5]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-6]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-()-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-20-unclamped]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-3-1]
0.00s call     tests/ops/test_contract.py::test_contract_to_tensor[example6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-3]
0.00s call     tests/contrib/gp/test_models.py::test_model[SparseGPRegression_y2D]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-0-3-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-0-3-(3, 2)]
0.00s call     tests/poutine/test_mapdata.py::test_replay[iplate-5]
0.00s call     tests/test_generic.py::test_trace_handler[pyro-logistic_regression]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-1-3-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-1-1]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-1-2]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-20-clamped]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-5]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-Trace_ELBO]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-2-3-(3, 2)]
0.00s call     tests/distributions/test_mvn.py::test_shape[1-cov_shape0-loc_shape1]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape1-sample_shape1]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-1-1]
0.00s call     tests/infer/test_autoguide.py::test_callable[AutoLowRankMultivariateNormal]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[ExtendedBetaBinomial]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-1]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-1-2]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-8]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[3-3-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-1-1]
0.00s call     tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-Adam-optim_kwargs0-py]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape1-df_shape1]
0.00s call     tests/distributions/test_spanning_tree.py::test_log_prob[3]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-5]
0.00s call     tests/contrib/test_hessian.py::test_hessian_mvn
0.00s call     tests/infer/test_autoguide.py::test_callable_return_dict[AutoLowRankMultivariateNormal]
0.00s call     tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape0-df_shape0]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,wxi,xi->i]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(3, 2)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-1-(3, 2)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[2-2-(3, 2)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,wxi,xi->i]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[2-3-(3, 2)]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape1-df_shape0]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-2-(4,)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-4]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-1-1]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(4,)-(5,)-9]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape2-sample_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[3-2-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-2-3-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-7]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-6]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(4,)-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-2-3-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(3, 2)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-3-(4,)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(4,)-(7,)]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[ShapeAugmentedBeta]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-2]
0.00s call     tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-1-1]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[Constant]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape0-df_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-0-3-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-1-3-(3, 2)]
0.00s call     tests/infer/test_autoguide.py::test_init_loc_fn[AutoMultivariateNormal]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-()-(7,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-list-sample_shape2]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-1-(4,)-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-20-clamped]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[3-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-Size-sample_shape2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-8]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-()-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-3]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-20-unclamped]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-1-3-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-tuple-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-tuple-sample_shape2]
0.00s call     tests/infer/test_autoguide.py::test_guide_list[AutoDelta]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-Size-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-list-sample_shape2]
0.00s call     tests/infer/test_jit.py::test_masked_fill
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[3-1-(3, 2)]
0.00s call     tests/optim/test_optim.py::test_clip_norm[1.0-clip_norm-Adam]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape2-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-4]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoDiagonalNormal-0.1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-3]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-1-(4,)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-1-1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-10]
0.00s setup    tests/integration_tests/test_tracegraph_elbo.py::BernoulliBetaTests::test_elbo_reparameterized
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-3]
0.00s call     tests/infer/test_enum.py::test_compute_marginals_hmm[1]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-1-1]
0.00s call     tests/infer/test_autoguide.py::test_init_scale[AutoDiagonalNormal-0.0001]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-20-clamped]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(4,)-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[2-1-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[ShapeAugmentedDirichlet]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[1-3-(3, 2)]
0.00s setup    tests/infer/test_csis.py::test_csis_sampling
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[1-2-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-2-3-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-20-unclamped]
0.00s call     tests/infer/test_discrete.py::test_hmm_smoke[sample-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-wx,xy,yz,zw->]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-8]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(3, 2)-()]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-()-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-1-1]
0.00s call     tests/ops/test_ssm_gp.py::test_matern_kernel[0.5-3]
0.00s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[2-size1]
0.00s call     tests/contrib/timeseries/test_gp.py::test_dependent_matern_gp[1]
0.00s call     tests/infer/test_jit.py::test_bernoulli_enumerate[shape2-False]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-()-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-20-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-20-unclamped]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-()-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-2]
0.00s call     tests/infer/test_jit.py::test_bernoulli_enumerate[shape1-True]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[9]
0.00s call     tests/test_generic.py::test_trace_handler[minipyro-logistic_regression]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-3-2]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-0-3-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-2-3-(3, 2)]
0.00s call     tests/infer/test_jit.py::test_categorical_enumerate[shape0-True]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[1-1-(3, 2)]
0.00s call     tests/distributions/test_mvn.py::test_shape[1-cov_shape1-loc_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-4]
0.00s call     tests/optim/test_optim.py::test_clip_norm[3.0-clip_norm-Adam]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-4]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-0-3-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-4]
0.00s call     tests/infer/test_autoguide.py::test_callable_return_dict[AutoMultivariateNormal]
0.00s call     tests/infer/test_valid_models.py::test_tail_adaptive_error
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-dense-3]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixyj,iyzj->ij]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-3]
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-flat-1]
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-dense-2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[-1.0--1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-6]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-3-2]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[ShapeAugmentedGamma]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-1-3-(3, 2)]
0.00s call     tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(4,)-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-3-2]
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-dense-1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-()-()]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-1-3-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-()-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(4,)-(5,)-9]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[1.0-0.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-3]
0.00s call     tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[pyro-py]
0.00s call     tests/infer/test_autoguide.py::test_callable[AutoMultivariateNormal]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[1.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-3-1]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-0-3-(3, 2)]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-()-(6, 5)]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-2-3-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-True-initial_shape0-proposed_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-2-1-1]
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-flat-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-1]
0.00s call     tests/optim/test_optim.py::test_clip_norm[5.0-clip_norm-Adam]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-3-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-True-initial_shape2-proposed_shape2]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-1]
0.00s call     tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape0-df_shape0]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[0.0--1.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-True-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-1-2]
0.00s call     tests/distributions/test_mvn.py::test_shape[3-cov_shape0-loc_shape0]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(3, 1)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-4]
0.00s call     tests/ops/test_contract.py::test_ubersum[,i->,i-i]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(4,)-(5,)-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-4]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-3-1]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape4-event_shape4]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-10-clamped]
0.00s call     tests/infer/test_discrete.py::test_hmm_smoke[map-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[3-3-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-20-clamped]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape4-event_shape4]
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-flat-3]
0.00s call     tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-list-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-tuple-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(2, 3)-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-Size-sample_shape2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[0.0-0.0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(2, 3)-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-Size-sample_shape2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[-1.0-1.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-list-sample_shape2]
0.00s call     tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape2-_pi]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-tuple-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-3-1]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-()-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 1, 1)-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-20-unclamped]
0.00s call     tests/poutine/test_poutines.py::test_pickling[<lambda>3]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-1-(4,)-(3, 2)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[w,w->]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-1]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[0.0-1.0]
0.00s call     tests/ops/test_newton.py::test_newton_step_trust[1-None]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[8]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[-1.0-0.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-Size-sample_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-6]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-3]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[BetaBinomial]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[1.0--1.0]
0.00s call     tests/test_generic.py::test_model_sample[pyro-logistic_regression]
0.00s call     tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape2-_quantile]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-list-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-tuple-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-3]
0.00s call     tests/ops/test_gaussian.py::test_reshape[2-(6,)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(5,)-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-Size-sample_shape2]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 1)-(4, 5)-9]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(3, 2)-(7,)]
0.00s call     tests/ops/test_ssm_gp.py::test_matern_kernel[0.5-2]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(5,)-3]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape4-event_shape4]
0.00s call     tests/test_generic.py::test_model_sample[minipyro-logistic_regression]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-list-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[ShapeAugmentedGamma]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[2-3-(4,)]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape1-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-tuple-sample_shape2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[2-2-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[3-2-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(2, 3)-3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(2, 3)-2]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 1)-(4, 5)-9]
0.00s call     tests/infer/test_autoguide.py::test_subsample_guide_2[AutoNormal-independent]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(3, 1)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-4]
0.00s call     tests/poutine/test_poutines.py::test_enumerate_poutine[2--3]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-()-(6, 5)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(3, 1)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-()-()]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-()-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-()-(6, 5)]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-Size-sample_shape1]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 1, 1)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(5,)-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-tuple-sample_shape0]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[sample-a,abi,bcij->-ij]
0.00s call     tests/distributions/test_mvn.py::test_shape[1-cov_shape0-loc_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-Size-sample_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-()-()]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(3, 1)-(7,)]
0.00s call     tests/infer/test_enum.py::test_unsafe_bernoulli[sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-Size-sample_shape1]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape3-event_shape3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(5,)-2]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-()-(6, 5)]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(3, 2)-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-20-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-()-(4, 5)-9]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape1-sample_shape2]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-20-clamped]
0.00s call     tests/ops/test_contract.py::test_ubersum_jit[i->,i-i]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-tuple-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-Size-sample_shape0]
0.00s call     tests/infer/test_autoguide.py::test_callable[AutoNormal]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-True-initial_shape2-proposed_shape2]
0.00s call     tests/test_generic.py::test_trace_handler[minipyro-beta_binomial]
0.00s call     tests/ops/test_ssm_gp.py::test_matern_kernel[0.5-1]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-20-unclamped]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-True-initial_shape0-proposed_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-20-unclamped]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape3-sample_shape1]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape3-event_shape3]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-True-initial_shape1-proposed_shape1]
0.00s call     tests/poutine/test_poutines.py::test_enumerate_poutine[2--2]
0.00s call     tests/infer/test_autoguide.py::test_init_loc_fn[AutoDiagonalNormal]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[3-1-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[2-1-(4,)]
0.00s call     tests/optim/test_optim.py::test_clip_norm[5.0-clip_value-Adam]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-w,wx,xy,yz->]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-()-(6, 5)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(3, 2)-()]
0.00s call     tests/infer/test_autoguide.py::test_callable_return_dict[AutoNormal]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-4]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(4,)-(5,)-9]
0.00s call     tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-()-(6, 5)]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-3-(4,)-(4,)]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape3-event_shape3]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-()-(6, 5)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-20-clamped]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-4]
0.00s call     tests/infer/test_autoguide.py::test_subsample_guide_2[AutoNormal-dependent]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[1-1-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-list-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-tuple-sample_shape1]
0.00s call     tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceEnum_ELBO-None]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-20-unclamped]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-1-2]
0.00s call     tests/poutine/test_poutines.py::test_enumerate_poutine[2--1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-()-(6, 5)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-10-unclamped]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(3, 2)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-2-(4,)-(4,)]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-()-(7,)]
0.00s call     tests/ops/test_stats.py::test_waic
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[1-3-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-list-sample_shape1]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-3-(4,)-(4,)]
0.00s call     tests/nn/test_module.py::test_constraints[shape24-constraint_24]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[1-2-(4,)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-wi,wxi,xyi,yzi->i]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-()-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-1-2]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-1-2]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s call     tests/distributions/test_mask.py::test_mask_type[mask3]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape1-sample_shape2]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(3, 1)-(7,)]
0.00s call     tests/infer/test_autoguide.py::test_callable[AutoDiagonalNormal]
0.00s call     tests/distributions/test_spanning_tree.py::test_partition_function[4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-5]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-2-(4,)-(4,)]
0.00s call     tests/nn/test_module.py::test_constraints[shape9-constraint_9]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-1]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape3-sample_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(3, 2)-()]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[3-2-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-list-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-4]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[2-2-(3, 2)]
0.00s call     tests/test_generic.py::test_rng_state[minipyro-neals_funnel]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-3-2-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-list-sample_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-1-3-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[3-3-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-1]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-2]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-1-1]
0.00s call     tests/contrib/test_hessian.py::test_hessian_multi_variables
0.00s call     tests/infer/test_jit.py::test_bernoulli_enumerate[shape1-False]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(4,)-()]
0.00s call     tests/infer/test_autoguide.py::test_subsample_guide_2[AutoDelta-independent]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-tuple-sample_shape0]
0.00s call     tests/optim/test_optim.py::test_clip_norm[1.0-clip_value-Adam]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,b,bc->ac-1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(4,)-()]
0.00s call     tests/infer/test_autoguide.py::test_callable_return_dict[AutoDiagonalNormal]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape1-sample_shape2]
0.00s call     tests/test_generic.py::test_model_sample[minipyro-beta_binomial]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(3, 2)-()]
0.00s call     tests/ops/test_newton.py::test_newton_step_converges[3-None]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-4]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-()-(6, 5)]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-1-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[2-3-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-list-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-1-1]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[5]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-10-unclamped]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[6]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-()-(6, 5)]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-2-(4,)-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-10-clamped]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-3-(4,)-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-3]
0.00s call     tests/infer/test_jit.py::test_categorical_enumerate[shape2-False]
0.00s call     tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape2-_hpdi]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-()-2]
0.00s call     tests/distributions/test_mvn.py::test_shape[1-cov_shape1-loc_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-3]
0.00s call     tests/contrib/gp/test_conditional.py::test_conditional_whiten[5]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape2-sample_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-10]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-tuple-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-tuple-sample_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(3, 2)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-1-(4,)-(4,)]
0.00s call     tests/infer/test_autoguide.py::test_scores[AutoDiagonalNormal]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-()-()]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_MultiClass_2D]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-()-2]
0.00s call     tests/ops/test_jit.py::test_varying_len_args
0.00s call     tests/ops/test_jit.py::test_varying_unhashable_kwargs
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(2,)-(4, 5)-9]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-tuple-sample_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(3, 2)-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-Size-sample_shape0]
0.00s call     tests/ops/test_contract.py::test_ubersum_collide_ok_2[ubersum]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[2-(2, 3)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(2,)-(4, 5)-9]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-Size-sample_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-10-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-10-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-()-(4, 5)-9]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-list-sample_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-()-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-Size-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-tuple-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-False-Size]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-Size-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_contract.py::test_einsum_linear[,i,i->,i-i]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-3]
0.00s call     tests/distributions/test_conjugate_update.py::test_gamma_poisson[(4,)-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-list-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-True-tuple]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(3, 2)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-()-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-Size-sample_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-()-(6,)-9]
0.00s call     tests/infer/test_autoguide.py::test_subsample_guide_2[AutoDelta-dependent]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-1-(4,)-(4,)]
0.00s call     tests/distributions/test_unit.py::test_expand[batch_shape2-sample_shape1]
0.00s call     tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-False-NUTS-_empty_model]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-True-list]
0.00s call     tests/distributions/test_conjugate_update.py::test_beta_binomial[()-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-()-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-tuple-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-False-list]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-False-tuple]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-10-unclamped]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-True-Size]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-4]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(4,)-(5,)-9]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_0_gate[2.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[()-(3, 2)]
0.00s call     tests/distributions/test_mask.py::test_mask_type[True]
0.00s call     tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-True-NUTS-_empty_model]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-3]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-10-clamped]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_MultiClass_1D]
0.00s call     tests/optim/test_optim.py::test_clip_norm[3.0-clip_value-Adam]
0.00s call     tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[minipyro-py]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape2-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-list-sample_shape1]
0.00s call     tests/contrib/oed/test_glmm.py::test_log_prob_matches[model14-normal_inv_gamma_2_2_10_10-design4]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-2-3-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-10]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_MultiClass_2D]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-()-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-list-sample_shape1]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_MultiClass_2D]
0.00s call     tests/infer/test_jit.py::test_bernoulli_enumerate[shape0-True]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-10-unclamped]
0.00s call     tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-True-HMC-_empty_model]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(2,)-(4, 5)-9]
0.00s call     tests/optim/test_optim.py::test_clip_norm[1.0-clip_norm-SGD]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-()-(2,)-(7,)]
0.00s call     tests/optim/test_optim.py::test_clip_norm[5.0-clip_norm-SGD]
0.00s call     tests/ops/test_linalg.py::test_sym_rinverse[True-A0]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[OneHotCategorical]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-5-unclamped]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-5-clamped]
0.00s call     tests/ops/test_linalg.py::test_sym_rinverse[False-A0]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape1-sample_shape2]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[3-1-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[2-1-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-1-3-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-4]
0.00s call     tests/test_generic.py::test_rng_seed[pyro-eight_schools]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-0-3-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[1-2-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-20-unclamped]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-2-3-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-10]
0.00s call     tests/optim/test_optim.py::test_clip_norm[3.0-clip_norm-SGD]
0.00s call     tests/ops/test_contract.py::test_contract_to_tensor[example5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-3]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_MultiClass_1D]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-4]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-4]
0.00s call     tests/infer/test_valid_models.py::test_mean_field_warn
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-2-2-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-10]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-3]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape1-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-0-3-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-1-2-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-1-2-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-2-3-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[1-3-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-3]
0.00s call     tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-1-3-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(4,)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-0-2-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-2-2-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-3]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-2-7]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-10-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-0-3-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(3, 2)-()]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal_mvn[3-(2, 3)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 1)-(6,)-9]
0.00s call     tests/infer/test_util.py::test_multi_frame_tensor
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_MultiClass_2D]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[3-2-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 2)-(6,)-9]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape1-sample_shape1]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-1-(4,)-(4,)]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_MultiClass_1D]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 2)-(6,)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(3, 2)-()]
0.00s call     tests/infer/test_jit.py::test_bernoulli_enumerate[shape0-False]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 2)-(6,)-9]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(2,)-(6,)-9]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-4]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[2-2-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-4]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape1-sample_shape1]
0.00s call     tests/ops/test_gaussian.py::test_reshape[1-(6,)-(3, 2)]
0.00s call     tests/contrib/oed/test_glmm.py::test_log_prob_matches[model12-lm_2p_10_10_1_w12-design2]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(2,)-(6,)-9]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_MultiClass_1D]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(3, 2)-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-5-unclamped]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-0-2-(3, 2)]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape1-sample_shape2]
0.00s call     tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-2]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-5-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-2-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[3-3-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-()-(4, 5)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-1-2-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-0-2-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size2-3]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-2-clamped]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-2-3-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 1)-(6,)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-2-2-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-3]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-2-unclamped]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-2]
0.00s call     tests/ops/test_jit.py::test_varying_kwargs
0.00s call     tests/ops/test_gaussian.py::test_marginalize[2-3-(4,)]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Poisson_1D]
0.00s call     tests/test_generic.py::test_rng_state[pyro-neals_funnel]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 2)-(6,)-9]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-6]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-10-clamped]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-3-(3, 2)-()]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Binary_2D]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Binary_2D]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Poisson_2D]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-10-unclamped]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-3]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Binary_1D]
0.00s call     tests/contrib/gp/test_models.py::test_model[GPRegression_y1D]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[4]
0.00s call     tests/nn/test_module.py::test_constraints[shape13-constraint_13]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-4]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Poisson_2D]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-10-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[11-(2, 3)-(7,)-9]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-5-clamped]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-0-3-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-1-2-(3, 2)]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[2-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-2-3-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-3]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-10-unclamped]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Categorical]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-2-3-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-0-2-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-5-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-3-2-7]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-0-3-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-2-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-20-clamped]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-1]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-2-(3, 2)-()]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-TraceGraph_ELBO]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-5-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-10-clamped]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-False-sample_shapes1]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[3]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-2-(3, 2)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-3-(3, 2)-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-5-unclamped]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-w,wx,x->]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(4,)-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-2-unclamped]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Binary_1D]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-False-tuple]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-10-clamped]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-2]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-5-clamped]
0.00s call     tests/ops/test_contract.py::test_ubersum_2[ubersum]
0.00s call     tests/infer/test_jit.py::test_categorical_enumerate[shape1-False]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-10-unclamped]
0.00s call     tests/distributions/test_conjugate_update.py::test_beta_binomial[(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-3]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-10-unclamped]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-1-3-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-2-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-5-unclamped]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(4,)-()]
0.00s call     tests/contrib/gp/test_models.py::test_model[VariationalGP_y1D]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-5]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(2,)-(4, 5)-9]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Poisson_1D]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[1-1-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-1-3-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-2]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-0-2-(3, 2)]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape3-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-True-tuple]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-10-unclamped]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(4,)-()]
0.00s call     tests/contrib/gp/test_models.py::test_model[VariationalGP_y2D]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-1-3-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-True-Size]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(4,)-()]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[ExtendedBinomial]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Poisson_2D]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-True-list]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 1)-(6,)-9]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-0-3-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Binomial]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal_mvn[2-(2, 3)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwj->ij]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Binary_2D]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-()-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-2-2-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-3-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[11-(6,)-(7,)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-1-2-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-3-(3, 2)-(3, 2)]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Poisson_2D]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-3-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-()-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-1-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-2-clamped]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-0-2-(3, 2)]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Poisson_1D]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-2-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[ShapeAugmentedDirichlet]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-()-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[1-2-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-1-2-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-2-2-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-2]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[iwj->ji]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[2-1-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-5-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-2-clamped]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-6]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-2-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-20-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-5-unclamped]
0.00s call     tests/nn/test_module.py::test_constraints[shape15-constraint_15]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-()-(7,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-False-Size]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[ShapeAugmentedGamma]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[abc]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[3-1-(4,)]
0.00s call     tests/infer/test_autoguide.py::test_init_loc_fn[AutoDelta]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 1)-(6,)-9]
0.00s call     tests/nn/test_module.py::test_constraints[shape16-constraint_16]
0.00s call     tests/nn/test_module.py::test_constraints[shape14-constraint_14]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-1-(3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-()-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-20-unclamped]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-2-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-4]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-3-(3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-()-(7,)]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Binary_2D]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape1-sample_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-5-clamped]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Binary_1D]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-10-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-5-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(2,)-(6,)-9]
0.00s call     tests/infer/test_jit.py::test_scatter
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-False-list]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-()-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(4,)-()]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape2-sample_shape2]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[,w->]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(4,)-()]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Multinomial]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-2-clamped]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-()-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(4,)]
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Binary_1D]
0.00s call     tests/distributions/test_conjugate_update.py::test_beta_binomial[(4,)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-10]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-2-2-(4,)]
0.00s call     tests/poutine/test_poutines.py::test_pickling[<lambda>2]
0.00s call     tests/infer/test_autoguide.py::test_callable[AutoLaplaceApproximation]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-()-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-2-clamped]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-6]
0.00s call     tests/infer/test_autoguide.py::test_callable_return_dict[AutoLaplaceApproximation]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-1-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[marginal-ai,abij->i-ij]
0.00s call     tests/contrib/easyguide/test_easyguide.py::test_overlapping_plates_error
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-2-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-2-unclamped]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(2,)-(7,)]
0.00s call     tests/nn/test_module.py::test_constraints[shape8-constraint_8]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 2)-(6,)-9]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-TraceGraph_ELBO]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-()-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-1-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-()-(7,)]
0.00s call     tests/params/test_module.py::test_random_module[outest]
0.00s call     tests/poutine/test_poutines.py::ReplayHandlerTests::test_replay_full
0.00s call     tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Poisson_1D]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-2-(3, 2)-()]
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-flat-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-10]
0.00s call     tests/contrib/gp/test_models.py::test_model[GPRegression_y2D]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-()-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-5]
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-dense-3]
0.00s call     tests/ops/test_contract.py::test_contract_to_tensor[example4]
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-flat-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-4]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 2)-(6,)-9]
0.00s call     tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-()-()]
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-dense-2]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 3, 2)-()]
0.00s call     tests/distributions/test_conjugate_update.py::test_gamma_poisson[()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(2,)-(6,)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-2-clamped]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[1-3-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(3, 2)-()-9]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-()-(6,)-9]
0.00s call     tests/contrib/randomvariable/test_random_variable.py::test_add
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 2)-(6,)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[2-size2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-10]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-1-(3, 2)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-4]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-20-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-3-1]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceEnum_ELBO-full]
0.00s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[3-size0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-10]
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-dense-1]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-()-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-2-clamped]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceTMC_ELBO-full]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 2)-(6,)-9]
0.00s call     tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-flat-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-2-unclamped]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-()-(7,)]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape2]
0.00s call     tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(3, 2)]
0.00s call     tests/optim/test_optim.py::test_clip_norm[1.0-clip_value-SGD]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape0-sample_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-2-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-20-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-5-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-20-clamped]
0.00s call     tests/contrib/tracking/test_assignment.py::test_sparse_grid_smoke
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(4,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 1, 1)-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-20-unclamped]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-10]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[0.1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-()-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-10-clamped]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(4,)-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-6]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-()-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-10]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(4,)-()]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,ixj->ji]
0.00s call     tests/ops/test_contract.py::test_ubersum_1[ubersum]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-5-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-5-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-5-clamped]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(2,)-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-5-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-10-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-5-unclamped]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(3, 2)]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceTMC_ELBO-subsample]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-10-clamped]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(2,)-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-5-unclamped]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(4,)-()]
0.00s call     tests/nn/test_module.py::test_constraints[shape23-constraint_23]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-2]
0.00s call     tests/distributions/test_spanning_tree.py::test_enumerate_support[4]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,b,bc->ac-2]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal_mvn[1-(2, 3)]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceEnum_ELBO-subsample]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[map-a,abi,bcij,bdik->-ijk]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape0-sample_shape2]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-10-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(3, 2)-()-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(4,)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-10]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(2,)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-()-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(4,)-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-5]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-()-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-10]
0.00s call     tests/infer/test_inference.py::SafetyTests::test_extra_samples
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-5-clamped]
0.00s call     tests/test_generic.py::test_rng_seed[pyro-neals_funnel]
0.00s call     tests/ops/test_indexing.py::test_value[()-(2,)-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(3, 1)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(4,)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(4,)-(7,)]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape2-sample_shape0]
0.00s call     tests/test_generic.py::test_trace_handler[pyro-eight_schools]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 1, 1)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-4]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-()-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-4]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(3, 1)-()]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,wx,x->]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-2-clamped]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(2,)-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-2-unclamped]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape3-sample_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-2-unclamped]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(3, 1)-()]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,a,ab,b,b,b,b->a-2]
0.00s call     tests/test_generic.py::test_rng_seed[minipyro-neals_funnel]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal_mvn[2-(4,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 1, 1)-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-5-clamped]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 3, 2)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-3-(4,)-()]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[1]
0.00s call     tests/optim/test_optim.py::test_clip_norm[3.0-clip_value-SGD]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-5-unclamped]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(2,)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(3, 1)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 1, 1)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 1, 1)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(3, 1)-()]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[OneHotCategorical]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-()-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 1, 1)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-()-()]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape2]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-()-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-()-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 1, 1)-()]
0.00s call     tests/optim/test_optim.py::test_clip_norm[5.0-clip_value-SGD]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 3, 2)-()]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,ixj->ji]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(2,)-()]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 3, 2)-()]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[100]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[bac]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(3, 1)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(2,)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(4,)]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(2,)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(2,)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-2-(4,)-()]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(3, 2)-()-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(4,)-(3, 2)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-()-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-()-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-3-(4,)-()]
0.00s call     tests/infer/test_smcfilter.py::test_systematic_sample[2]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 3, 2)-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-5-unclamped]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 1, 1)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-2-(4,)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(3, 1)-()]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-TraceEnum_ELBO]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-2-unclamped]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(4,)-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-20-clamped]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[acb]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-()-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-1-2-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(4,)-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-5-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-2-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-2-unclamped]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 1, 1)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(3, 1)-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-10-clamped]
0.00s call     tests/infer/test_autoguide.py::test_scores[AutoNormal]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-0-2-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size0-3]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(2,)-(4, 5)-9]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-()-(5,)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(4,)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-3]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[bca]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-6]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape0-sample_shape0]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal_mvn[3-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-()-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(2,)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(3, 1)-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-20-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-()-(4, 5)-9]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(2,)-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-20-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-10-unclamped]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 3, 2)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-3-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-2-2-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-2-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-2-clamped]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(2,)-()]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[cba]
0.00s call     tests/ops/test_newton.py::test_newton_step_converges[1-None]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,ixj->ij]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 3, 2)-()]
0.00s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-dense-1]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-()-()]
0.00s call     tests/ops/test_indexing.py::test_value[()-(2,)-(4, 3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 1, 1)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 1, 1)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-3]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,ixj->ij]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-3-7]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[cab]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size1-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-1]
0.00s call     tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape0-autocorrelation]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(2,)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-1-(3, 2)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-10]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(4,)-(3, 2)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 3, 2)-()]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[1-1-(4,)]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[marginal-a,abi->-i]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(2,)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-()-(7,)]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape2-sample_shape0]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->bc-2]
0.00s call     tests/ops/test_indexing.py::test_value[()-()-(4, 3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape1-sample_shape1]
0.00s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-flat-1]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape3-sample_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(2,)-(4, 5)-9]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(4,)-(3, 2)]
0.00s call     tests/ops/test_newton.py::test_newton_step_converges[2-None]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(3, 1)-()]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(2,)-()]
0.00s call     tests/infer/test_autoguide.py::test_callable[AutoDelta]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-3-()-(3, 2)]
0.00s call     tests/test_generic.py::test_rng_state[minipyro-eight_schools]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-2-unclamped]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-()-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-1-2-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-2-clamped]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-2-(4,)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(4,)-(3, 2)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-()-()]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 3, 2)-()]
0.00s call     tests/ops/test_packed.py::test_broadcast_all[shapes4]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-3]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[wi->i]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-1-2-(4,)]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(3, 1)-()]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-2-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-0-2-(4,)]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 1, 1)-()]
0.00s call     tests/contrib/autoname/test_scoping.py::test_simple_recur
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-()-(5,)-9]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-3-(4,)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-3-7]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[10]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-3-()-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-4]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype0-size2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-2-1-7]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-2-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-2-2-(4,)]
0.00s call     tests/infer/test_autoguide.py::test_callable_return_dict[AutoDelta]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-2-unclamped]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-2-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-3-(3, 2)-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-1]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 3, 2)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-5]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-10-clamped]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape2-sample_shape0]
0.00s call     tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape0-autocovariance]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(4,)]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-()-(4, 3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(4,)-(3, 2)]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-()-()]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 3, 2)-()]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,a,ab,b,b,b,b->a-1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-2-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-10-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-20-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-20-clamped]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(4,)]
0.00s call     tests/test_generic.py::test_rng_seed[minipyro-eight_schools]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-20-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-3-2-7]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-2-clamped]
0.00s call     tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape1-_quantile]
0.00s call     tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-TraceEnum_ELBO]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[5-(2, 3)-(7,)-9]
0.00s call     tests/test_generic.py::test_model_sample[pyro-eight_schools]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-2-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-5-unclamped]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(4,)]
0.00s call     tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape1-_pi]
0.00s call     tests/params/test_param.py::ParamStoreDictTests::test_save_and_load
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-5-clamped]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-1-(4,)-()]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-()-(5,)-9]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(4,)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-1-(4,)-()]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwj->ji]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-0-2-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-20-unclamped]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-3-7]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-1-1-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-3-2-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-3-7]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-10-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-20-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-10-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-5-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-10-clamped]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[5-(6,)-(7,)-9]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape2-sample_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 2)-(6,)-9]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[1-(2, 3)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-10-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-20-clamped]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal_mvn[1-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-4]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(4,)-()-9]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-2-clamped]
0.00s call     tests/poutine/test_poutines.py::test_enumerate_poutine[1--3]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-2-clamped]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-1-2-(4,)]
0.00s call     tests/poutine/test_poutines.py::UnconditionHandlerTests::test_uncondition
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(4,)]
0.00s call     tests/infer/test_jit.py::test_categorical_enumerate[shape0-False]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-5-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-2-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-5-clamped]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-EnergyDistance_prior]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 1)-(6,)-9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-5]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape3]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 3, 2)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-6]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-1-2-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-1-1-(3, 2)]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype1-size2]
0.00s call     tests/infer/test_initialization.py::test_init_to_generated
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-20-clamped]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(4,)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->c-2]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-20-unclamped]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-1-()-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-10]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(4,)-()-9]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 1)-(6,)-9]
0.00s call     tests/ops/test_contract.py::test_contract_tensor_tree[example4]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-5-unclamped]
0.00s call     tests/nn/test_module.py::test_constraints[shape11-constraint_11]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(2,)-(6,)-9]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-()-(6,)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-2-1-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-10]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-2-()-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(4,)-()-9]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwj->ji]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-20-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-20-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-20-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-20-unclamped]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwj->ij]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-20-clamped]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->bc-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,xi->i]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-0-1-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-6]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 2)-(6,)-9]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ac-2]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[3-(2, 3)-(7,)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-2-(3, 2)-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-10]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-1-()-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(3, 2)-()-9]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-3]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-1-2-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 2)-(6,)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-2-(3, 2)-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 2)-(6,)-9]
0.00s call     tests/ops/test_contract.py::test_ubersum_total[ai->ai-i]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-1-(3, 2)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-2-1-(3, 2)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-ji,iwj->ji]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(4,)-(4,)]
0.00s call     tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[TraceTMC_ELBO]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(2,)-(6,)-9]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag[mat_size0-2-batch_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-2-1-(3, 2)]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-EnergyDistance_prior]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-3-(3, 2)-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-0-2-(4,)]
0.00s call     tests/test_generic.py::test_trace_handler[minipyro-eight_schools]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-0-1-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-0-2-(4,)]
0.00s call     tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[TraceEnum_ELBO]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,xi->i]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-0-1-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-2-(3, 2)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-1-1-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-1-1-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-()-(6, 5)]
0.00s call     tests/nn/test_module.py::test_constraints[shape7-constraint_7]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ad-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-3-(3, 2)-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-2-2-(4,)]
0.00s call     tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[()-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-3-(4,)-(3, 2)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->ac-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-ij,iwj->ij]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-2-2-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-0-2-(4,)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-ji,iwj->ji]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-2]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[3-(6,)-(7,)-9]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-10]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(4,)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-()-(6, 5)]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-2-2-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-1-(3, 2)-(4,)]
0.00s call     tests/distributions/test_mask.py::test_mask_type[mask2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-1-1-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwj->ij]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(4,)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(4,)-(4,)]
0.00s call     tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape1-_hpdi]
0.00s call     tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-TraceTMC_ELBO]
0.00s call     tests/poutine/test_poutines.py::test_replay_enumerate_poutine[0--3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-3]
0.00s call     tests/ops/test_stats.py::test_fit_generalized_pareto[0.8-0.2]
0.00s call     tests/poutine/test_poutines.py::test_replay_enumerate_poutine[0--2]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(4,)-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-3]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-TraceTMC_ELBO]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-EnergyDistance_noprior]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-6]
0.00s call     tests/ops/test_contract.py::test_contract_to_tensor[example2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-1]
0.00s call     tests/infer/test_valid_models.py::test_tail_adaptive_warning
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-1-(3, 2)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-3-(4,)-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-()-(5,)-9]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-()-(6, 5)]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-EnergyDistance_noprior]
0.00s call     tests/poutine/test_poutines.py::test_replay_enumerate_poutine[0--1]
0.00s call     tests/ops/test_contract.py::test_ubersum_collide_ok_1[ubersum]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(4,)-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-4]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape0]
0.00s call     tests/nn/test_module.py::test_constraints[shape12-constraint_12]
0.00s call     tests/ops/test_contract.py::test_einsum_linear[a->,a-]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-4]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape1-sample_shape0]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape1-sample_shape0]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(3, 2)-()-9]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape1-sample_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(4,)-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-5]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[1-(5,)]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape3]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-1-(4,)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(4,)-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(4,)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-2-1-(3, 2)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->ac-2]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape2-sample_shape0]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-2-()-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-2]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-2-1-(3, 2)]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-10]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-2-1-(3, 2)]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-0.1]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-3-()-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-2-()-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-3-()-(4,)]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[4-()]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[ij->ji]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[2-(2, 3)-(7,)-9]
0.00s call     tests/ops/test_special.py::test_log_binomial_stirling[0.05]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(4,)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(4,)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-1-(4,)-(3, 2)]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-5-iplate]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-2-(4,)-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(4,)-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-0-1-(4,)]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-5-5-iplate]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[sample-ai,abij->i-ij]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape1-sample_shape2]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(3, 2)-()]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-1-1-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(4,)]
0.00s call     tests/poutine/test_poutines.py::test_enumerate_poutine[1--2]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-1.0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-6]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(3, 2)-()-9]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-0-1-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-10]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->b-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-6]
0.00s call     tests/distributions/test_conjugate_update.py::test_beta_binomial[()-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-10]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ac-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-1-(4,)-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-2-7]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-1-()-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-6]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[2-(6,)-(7,)-9]
0.00s call     tests/ops/test_contract.py::test_einsum_linear[,i->,i-i]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ad-1]
0.00s call     tests/poutine/test_poutines.py::test_enumerate_poutine[1--1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->c-1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->c-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-6]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-()-()]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->ac-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-6]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-0-1-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-1-1-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(4,)]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoIAFNormal]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(2,)-(7,)]
0.00s call     tests/ops/test_contract.py::test_contract_tensor_tree[example7]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-0-1-(3, 2)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->d-2]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->b-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,wi->i]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-2]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-3-(4,)-(3, 2)]
0.00s call     tests/ops/test_contract.py::test_contract_tensor_tree[example6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-4]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-1-()-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-5]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-3-2-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixj->ji]
0.00s call     tests/ops/test_contract.py::test_adjoint_marginal[a->-]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape1-sample_shape0]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-3]
0.00s call     tests/ops/test_contract.py::test_contract_tensor_tree[example5]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->c-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-2-(4,)-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-3]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-()-(2,)-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-1-(4,)-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-2-7]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->a-2]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(4,)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-3-()-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-1-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-2-(4,)-(3, 2)]
0.00s call     tests/distributions/test_one_hot_categorical.py::test_batch_log_dims[dim=1-probs0]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-()-(5,)-9]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoNormal]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(4,)-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-10]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoIAFNormal]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-2]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(2,)-(3, 1)-(7,)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixj->ij]
0.00s call     tests/contrib/test_util.py::test_rvv[a0-b0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-10]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->b-2]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(2,)-(7,)]
0.00s call     tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoNormal]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(3, 2)-()]
0.00s call     tests/ops/test_contract.py::test_ubersum[a->,a-]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-ij,iwj->ij]
0.00s call     tests/contrib/oed/test_glmm.py::test_log_prob_matches[model10-lm_2p_10_10_1-design0]
0.00s call     tests/contrib/oed/test_glmm.py::test_log_prob_matches[model15-lr_10_10-design5]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-()-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(2,)-(7,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-1-1-7]
0.00s call     tests/ops/test_stats.py::test_fit_generalized_pareto[0.8-0.5]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-()-(3, 1)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-1-()-(4,)]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape2]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->a-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-1-1-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-6]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-2-()-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-3-2-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-2-1-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-3]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->d-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-True-initial_shape0-proposed_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-0-1-(4,)]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(3, 1)-(7,)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->b-2]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(4,)-(3, 2)]
0.00s call     tests/ops/test_contract.py::test_ubersum_total[,i->i-i]
0.00s call     tests/ops/test_indexing.py::test_value[()-(2,)-(3, 1)-(7,)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,wi->i]
0.00s call     tests/ops/test_indexing.py::test_value[()-(3, 1)-(2,)-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-True-initial_shape1-proposed_shape1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(4,)-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-True-initial_shape2-proposed_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-5]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[2-0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-0-1-(4,)]
0.00s call     tests/test_generic.py::test_trace_handler[pyro-neals_funnel]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(4,)-(4,)]
0.00s call     tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(3, 2)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-6]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(4,)-()-9]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-()-(7,)]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[1-()]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-i,wi->i]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->b-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(3, 2)-()]
0.00s call     tests/test_generic.py::test_model_sample[minipyro-eight_schools]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(3, 2)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-2-1-7]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->c-1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->c-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-10]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-False-sample_shapes0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-3]
0.00s call     tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceGraph_ELBO-sequential]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-10]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape2]
0.00s call     tests/ops/test_stats.py::test_fit_generalized_pareto[1.3-0.5]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-1.1]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-()-(5,)-9]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-2]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(4,)-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-6]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->b-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-10]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-0.5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(3, 2)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(3, 1)-(7,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-3-2-7]
0.00s call     tests/contrib/oed/test_glmm.py::test_log_prob_matches[model13-nz_lm_2p_10_10_1-design3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-4]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-()-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(4,)-()-9]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-2.0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-10]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-0-1-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-2-1-(4,)]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[3-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-1-1-(4,)]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-0.1]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-0.9]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj->ji]
0.00s call     tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceGraph_ELBO-parallel]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-10.0]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-2.0]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->ac-1]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-0.9]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape3-sample_shape0]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[2-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-2-1-(4,)]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-0.5]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape3-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape1-sample_shape2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-1-1-(4,)]
0.00s call     tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape3]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-1.1]
0.00s call     tests/distributions/test_lkj.py::test_log_prob_d2[0.1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(3, 2)-()]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-1.0]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->a-2]
0.00s call     tests/contrib/autoname/test_scoping.py::test_multi_nested
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-5]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-10.0]
0.00s call     tests/contrib/oed/test_glmm.py::test_log_prob_matches[lm_2p_10_10_1-model21-design1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-()-(3, 2)]
0.00s call     tests/ops/test_special.py::test_log_binomial_stirling[1.0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-1-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-10]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-10.0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-1-1-7]
0.00s call     tests/infer/test_valid_models.py::test_enum_in_model_multi_scale_error
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(3, 2)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-4]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-10]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(3, 1)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-5]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->a-1]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[2-3]
0.00s call     tests/ops/test_stats.py::test_fit_generalized_pareto[1.3-0.2]
0.00s call     tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[minipyro-py]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(3, 2)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-3-2-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-3]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-2.0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-4]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-0.1]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-0.9]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_0_gate[1.0]
0.00s call     tests/ops/test_contract.py::test_ubersum_total[ai->i-i]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-0.5]
0.00s call     tests/distributions/test_mask.py::test_mask_type[False]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape2-sample_shape0]
0.00s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-subsample-nonreparam]
0.00s call     tests/optim/test_optim.py::test_clippedadam_clip[5.0]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->a-2]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-1.0]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[sample-a,abi->-i]
0.00s call     tests/test_generic.py::test_trace_handler[minipyro-neals_funnel]
0.00s call     tests/optim/test_optim.py::test_clippedadam_clip[1.0]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(4,)-()-9]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-1.1]
0.00s call     tests/ops/test_contract.py::test_ubersum_collide_ok_3[ubersum]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[2-4]
0.00s call     tests/distributions/test_lkj.py::test_log_prob_d2[5.0]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-wi,wxi,xi->i]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 1, 1)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-2-1-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(4,)-(4,)]
0.00s call     tests/distributions/test_lowrank_mvn.py::test_scale_tril
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape2-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-2]
0.00s call     tests/distributions/test_lkj.py::test_log_prob_d2[1.0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-True-initial_shape0-proposed_shape0]
0.00s call     tests/poutine/test_poutines.py::QueueHandlerDiscreteTest::test_queue_max_tries
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-()-()]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape2-sample_shape2]
0.00s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-full-nonreparam]
0.00s call     tests/distributions/test_one_hot_categorical.py::test_batch_log_dims[dim=3-probs0]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(4,)]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[2-1]
0.00s call     tests/distributions/test_empirical.py::test_mean_var_non_nan
0.00s call     tests/distributions/test_reshape.py::test_reshape[1-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-4]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-1-()-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(4,)-(4,)]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(4,)-()]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape1-sample_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-()-(4, 5)-9]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 1, 1)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 1, 1)-()]
0.00s call     tests/nn/test_module.py::test_constraints[shape22-constraint_22]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape2-sample_shape2]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(4,)-()]
0.00s call     tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[Trace_ELBO-sequential]
0.00s call     tests/distributions/test_lkj.py::test_log_prob_d2[0.5]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-10-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-True-initial_shape1-proposed_shape1]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(3, 1)-()]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape3-sample_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-1-7]
0.00s call     tests/nn/test_module.py::test_decorator[2-DecoratorModel]
0.00s call     tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(4,)-()]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->a-1]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[map-ai,abij,acik->i-ijk]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-10]
0.00s call     tests/ops/test_contract.py::test_ubersum[i->,i-i]
0.00s call     tests/poutine/test_poutines.py::LiftHandlerTests::test_prior_dict
0.00s call     tests/ops/test_gaussian.py::test_pad[0-2-1-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-3-(4,)-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-True-initial_shape2-proposed_shape2]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 1, 1)-()]
0.00s call     tests/distributions/test_lkj.py::test_log_prob_eta1[4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-6]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(4,)-()]
0.00s call     tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[Trace_ELBO-parallel]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-2]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-i,wi->i]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag[mat_size2-3-batch_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-10-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-10-clamped]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-0-1-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-4]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-w,wx,x->]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-1-(4,)-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-10-unclamped]
0.00s call     tests/optim/test_optim.py::test_clippedadam_clip[3.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-3-(4,)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-2-(4,)-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-3]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(3, 1)-()]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-()-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-4]
0.00s call     tests/nn/test_module.py::test_decorator[1-DecoratorModel]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-2]
0.00s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-subsample-nonreparam]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-3]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-()-()]
0.00s call     tests/distributions/test_lkj.py::test_log_prob_d2[2.0]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(3, 2)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size2-2]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj->ij]
0.00s call     tests/ops/test_special.py::test_log_binomial_stirling[0.11]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-2-1-(4,)]
0.00s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-dense-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-2-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[MultivariateNormal]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-1-7]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(3, 1)-()]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape1]
0.00s call     tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 1, 1)-()]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-()-()]
0.00s call     tests/optim/test_optim.py::test_clippedadam_pass[1.0]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-1-1-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-()-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-3]
0.00s call     tests/ops/test_special.py::test_log_binomial_stirling[0.10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(4,)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-3-2-1]
0.00s call     tests/distributions/test_one_hot_categorical.py::test_batch_log_dims[dim=2-probs0]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-10-unclamped]
0.00s call     tests/test_generic.py::test_model_sample[pyro-neals_funnel]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-2]
0.00s call     tests/ops/test_stats.py::test_quantile
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape1-sample_shape2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-()-(3, 2)]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 1)-(6,)-9]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[GammaPoisson]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-6]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-10-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-()-(6,)-9]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-10-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-2-7]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-10-unclamped]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-1-1-1]
0.00s call     tests/test_generic.py::test_model_sample[minipyro-neals_funnel]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(3, 1)-()]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(4,)-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-Size-sample_shape0]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-3-7]
0.00s call     tests/distributions/test_conjugate_update.py::test_gamma_poisson[()-(4,)]
0.00s call     tests/distributions/test_reshape.py::test_reshape[3-1]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-()-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-1-1-(4,)]
0.00s call     tests/nn/test_module.py::test_constraints[shape6-constraint_6]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-2-1-7]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 1, 1)-(7,)]
0.00s call     tests/optim/test_optim.py::test_clippedadam_pass[3.0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-3]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 1)-(6,)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-2-(4,)-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-2]
0.00s call     tests/poutine/test_poutines.py::LiftHandlerTests::test_splice
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape2]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(4,)-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-3-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-Size-sample_shape0]
0.00s call     tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[2-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(4,)-()]
0.00s call     tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-full-nonreparam]
0.00s call     tests/optim/test_optim.py::test_clippedadam_pass[5.0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-2]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-2-1-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-1-1-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-1-(4,)-(4,)]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-4]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape3-sample_shape2]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-3-7]
0.00s call     tests/distributions/test_reshape.py::test_reshape[3-2]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-0-1-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-2-(4,)-(4,)]
0.00s call     tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-flat-1]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(4,)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-Size-sample_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-3-(4,)-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-3-7]
0.00s call     tests/ops/test_contract.py::test_contract_to_tensor[example3]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-Size-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-tuple-sample_shape0]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,b,bc->ac-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-10]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-Size-sample_shape0]
0.00s call     tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape0-_pi]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-()-()]
0.00s call     tests/distributions/test_reshape.py::test_reshape[3-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(4,)-()]
0.00s call     tests/distributions/test_reshape.py::test_reshape[3-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(4,)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(4,)-()]
0.00s call     tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape4]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-6]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-3-2-7]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-5]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-5]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-list-sample_shape0]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape3-sample_shape1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(3, 2)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-2-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-6]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-1-(4,)-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-()-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[3-2-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[3-3-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-()-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-1-7]
0.00s call     tests/ops/test_special.py::test_log_binomial_stirling[0.2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-tuple-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-3-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-1-7]
0.00s call     tests/ops/test_stats.py::test_weighted_waic
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-()-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-2-1-1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->bc-2]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-()-(4, 5)-9]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(4,)-(4,)]
0.00s call     tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape4]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-2]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-wi->i]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-2-1-7]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-1]
0.00s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_reparameterized_four_layers
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-5]
0.00s call     tests/distributions/test_reshape.py::test_reshape[3-6]
0.00s call     tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-()-()]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[Bernoulli]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[2-3-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-()-(3, 2)]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-5]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(4,)-(4,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[5.0-5-2.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-()-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-tuple-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-list-sample_shape0]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-6]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-()-(3, 2)]
0.00s call     tests/distributions/test_haar.py::test_haar_ortho[9]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-6]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(3, 2)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-3-2-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-10]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-list-sample_shape0]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[WhiteNoise0]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape0-sample_shape2]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-6]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-()-(6,)-9]
0.00s call     tests/ops/test_contract.py::test_naive_ubersum[,i->,i-i]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-3]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape3-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-tuple-sample_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_precision_to_scale_tril[batch_shape1-event_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-1-1-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-2]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype0-size1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 1)-(6,)-9]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-2]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-()-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(4,)-()]
0.00s call     tests/distributions/test_reshape.py::test_reshape[3-4]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(3, 2)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-4]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(4,)-(4,)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,b,bc->ac-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-list-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-Size-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-3-2-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-5]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-4]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-4]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-1-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-()-(3, 2)]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape0-sample_shape1]
0.00s call     tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape0-_quantile]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-tuple-sample_shape0]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-4]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-tuple-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-2-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-5]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape2-sample_shape1]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-5]
0.00s call     tests/poutine/test_poutines.py::test_pickling[<lambda>0]
0.00s call     tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape4]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 1)-(6,)-9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-1-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-2]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape3-sample_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_add[1-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[2-2-()]
0.00s call     tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-2-3-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-5]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(3, 2)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-1-7]
0.00s call     tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[10.0-1.0]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-()-(7,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-1-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-list-sample_shape0]
0.00s call     tests/ops/test_indexing.py::test_value[()-()-(4, 1, 1)-(7,)]
0.00s call     tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape0-_hpdi]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-3-(5,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[2-1-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[1.0-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[5.0-5.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[3-1-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-3-(2, 3)]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[NaiveBeta]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-2-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[1-2-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-2]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Bernoulli]
0.00s call     tests/distributions/test_omt_mvn.py::test_log_prob[OMTMultivariateNormal]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape2-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-1-1-7]
0.00s call     tests/distributions/test_stable.py::test_shape[batch_shape1-sample_shape2]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 1)-(6,)-9]
0.00s call     tests/contrib/tracking/test_hashing.py::test_merge_points_small
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[1-3-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-4]
0.00s call     tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize[1-1-()]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-()-(7,)]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape1-sample_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-6]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-3-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-list-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-2-7]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-()-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-3-(3, 2)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size0-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-2]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[ij->ij]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-()-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[1.0-10.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-()-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-()-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-6]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-5-clamped]
0.00s call     tests/distributions/test_conjugate_update.py::test_gamma_poisson[(3, 2)-()]
0.00s call     tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape2]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(4,)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-()-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-6]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size1-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Binomial-True-initial_shape0-proposed_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-()-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-1-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-2-1-1]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape3-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[FoldedNormal]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-3-2-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-6]
0.00s call     tests/ops/test_indexing.py::test_value[()-(3, 1)-()-(7,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Binomial-True-initial_shape1-proposed_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-2-(3, 2)-()]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_0_gate[0.1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-3-1]
0.00s call     tests/ops/test_newton.py::test_newton_step_converges[1-10.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-6]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-3-2-1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-5-clamped]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape1-sample_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-()-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-3-2-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-1-3-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-5]
0.00s call     tests/distributions/test_stable.py::test_shape[batch_shape2-sample_shape2]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(4,)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-()-(4,)]
0.00s call     tests/distributions/test_spanning_tree.py::test_log_prob[2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-0-3-()]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype1-size1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-5-clamped]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape0-sample_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-2-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-()-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-0-3-()]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape4]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Binomial-True-initial_shape2-proposed_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-1-3-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-3-1]
0.00s call     tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape0-split_gelman_rubin]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape1-sample_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-3]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape3-sample_shape1]
0.00s call     tests/nn/test_module.py::test_constraints[shape5-constraint_5]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-2-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-0-3-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-5-unclamped]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-2-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-3-(3, 2)-()]
0.00s call     tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs3]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-5-unclamped]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-2]
0.00s call     tests/ops/test_packed.py::test_broadcast_all[shapes3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc->ac-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-1-3-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-5]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-()-(7,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_add[3-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-5]
0.00s call     tests/nn/test_module.py::test_constraints[shape10-constraint_10]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-5-unclamped]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-3-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-2-3-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-2-(3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-2-(3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-1-(3, 2)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-()-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-5-clamped]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->bc-1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-5-unclamped]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-2-3-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-()-(4,)]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape0-sample_shape2]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->c-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-3-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(3, 2)-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_add[2-(3, 2)]
0.00s call     tests/distributions/test_stable.py::test_shape[batch_shape0-sample_shape0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-1-(3, 2)-()]
0.00s call     tests/poutine/test_mapdata.py::test_replay[plate-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-4]
0.00s call     tests/poutine/test_poutines.py::IndirectLambdaHandlerTests::test_graph_structure
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ac-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-()-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-3-(3, 2)-()]
0.00s call     tests/distributions/test_haar.py::test_haar_ortho[8]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape2-sample_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-4]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-()-(4, 5)-9]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-True-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-1-(3, 2)-()]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal_mvn[3-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-2-1]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-20-5-iplate]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-6]
0.00s call     tests/contrib/gp/test_parameterized.py::test_nested_parameterized
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-()-(7,)]
0.00s call     tests/distributions/test_omt_mvn.py::test_log_prob[AVFMultivariateNormal]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-4]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-2-1]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-()-(3, 1)-(7,)]
0.00s call     tests/nn/test_module.py::test_constraints[shape21-constraint_21]
0.00s call     tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs2]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(3, 2)-()]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 1)-(6,)-9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-2-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-5]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(3, 2)-()]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape6]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(4,)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-2]
0.00s call     tests/nn/test_module.py::test_cache
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape0-sample_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-5]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(2,)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-3]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[ab]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(4,)-()]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape3]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape6]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-6]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-()-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-True-initial_shape2-proposed_shape2]
0.00s call     tests/distributions/test_lkj.py::test_log_prob_eta1[3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-True-initial_shape0-proposed_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-2-1]
0.00s call     tests/ops/test_indexing.py::test_value[()-()-(3, 1)-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-3-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-()-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-10]
0.00s call     tests/poutine/test_poutines.py::LiftHandlerTests::test_unlifted_param
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(3, 1)-(7,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-2-(2, 3)]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[ba]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-2-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-2]
0.00s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[2-size0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-3-2-7]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[LowRankMultivariateNormal]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-3-7]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape6]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-()-()]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-0-3-()]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape0-sample_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-()-(6,)-9]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ac-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-1-1]
0.00s call     tests/nn/test_module.py::test_constraints[shape4-constraint_4]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-2-(5,)]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-do-normal_normal_model]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(4,)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-()-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-True-initial_shape0-proposed_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-1-1-7]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape6]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-5-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-3-2-7]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-5]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape0-sample_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-()-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-2-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-()-()]
0.00s call     tests/poutine/test_poutines.py::test_pickling[<lambda>1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-3]
0.00s call     tests/ops/test_tensor_utils.py::test_idct[shape2]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-3-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-2-(4,)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[33]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-3]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-5]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-3]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-6]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.05]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-2-()-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-2-1]
0.00s call     tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs0]
0.00s call     tests/ops/test_indexing.py::test_value[()-(3, 1)-(3, 1)-(7,)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->c-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_add[3-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-3-7]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal_mvn[2-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-2-1-7]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape0-sample_shape1]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape3-sample_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-2-clamped]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-6]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-1]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-()-()]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(4,)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_add[2-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-True-initial_shape1-proposed_shape1]
0.00s call     tests/ops/test_contract.py::test_contract_to_tensor[example0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-2-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-True-initial_shape2-proposed_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-1-1]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-1-3-()]
0.00s call     tests/poutine/test_poutines.py::QueueHandlerMixedTest::test_queue_enumerate
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_0_gate[1.1]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 1, 1)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-1-2-()]
0.00s call     tests/ops/test_contract.py::test_einsum_linear[i->,i-i]
0.00s call     tests/distributions/test_lowrank_mvn.py::test_variance
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-3-(4,)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-2-2-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-6]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(3, 2)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-2]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->b-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-1-1-7]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-4]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-2-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-5]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-1-3-()]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-0-3-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-()-()]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-3]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(2,)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-4]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-1-1-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-5]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(2,)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-3-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj->ji]
0.00s call     tests/contrib/autoguide/test_mean_field_entropy.py::test_guide_entropy[mean_field_guide-args1-expected_entropy1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-()-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-1-2-()]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-2-3-()]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(2,)-(6,)-9]
0.00s call     tests/ops/test_indexing.py::test_value[()-(2,)-(4, 1, 1)-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-1.0-0.5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-1-2-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-4]
0.00s call     tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs1]
0.00s call     tests/distributions/test_conjugate_update.py::test_gamma_poisson[(4,)-()]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->b-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-2-2-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-2-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-2-unclamped]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-1-7]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-1-3-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-3-2-7]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ad-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-1-(4,)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-3-2-1]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 1, 1)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 1, 1)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-3-7]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(2,)-()]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-2-3-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-4]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-2-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[11-()-(4, 5)-9]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-2-3-()]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_0_gate[0.5]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-2-clamped]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-2]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(2,)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-0-2-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-3-(4,)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-4]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-do-bernoulli_normal_model]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal_mvn[1-()]
0.00s call     tests/ops/test_special.py::test_log_beta_stirling[0.11]
0.00s call     tests/ops/test_special.py::test_log_beta_stirling[0.2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-2-2-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-2-7]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_0_gate[0.9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-0-2-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-3-(2, 3)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-6]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->c-2]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_0_gate[10.0]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->ac-2]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag[mat_size1-1-batch_shape1]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-5]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-2-unclamped]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-3]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-3-(2, 3)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-2-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-2-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_add[1-(4,)]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape0-sample_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-4]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-3-7]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-()-(6,)-9]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-2-()-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[80]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-6]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-3-2-7]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-()-()]
0.00s call     tests/poutine/test_poutines.py::test_decorator_interface_primitives
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-2-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-()-(6,)-9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-3-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[10.0-1.0-1.0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-1-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-2-(4,)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-1-1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->d-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-3-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-2]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->a-2]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-0-2-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-5]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-2-(2, 3)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj->ij]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-1-7]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-()-()-(7,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-2-7]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->a-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-1-(4,)-()]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc->ac-1]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[map-a,abi,bcij->-ij]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-2-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.05]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-20-clamped]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-()-(4, 1, 1)-()]
0.00s call     tests/ops/test_contract.py::test_naive_ubersum[a->,a-]
0.00s call     tests/contrib/autoguide/test_mean_field_entropy.py::test_guide_entropy[mean_field_guide-args0-expected_entropy0]
0.00s call     tests/distributions/test_unit.py::test_expand[batch_shape1-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape2-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-2]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->c-2]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-0-3-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-3-(4,)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-1-1-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.05]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-2-7]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->b-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-1-1]
0.00s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_two_layers_model_permuted
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-5-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-3-2-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-10.0-1.0]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(2,)-(6,)-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-1-()-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[VonMises]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape2-event_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-3]
0.00s call     tests/poutine/test_properties.py::test_idempotent[trace-bernoulli_normal_model]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[57]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Multinomial-True-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Multinomial-True-initial_shape0-proposed_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-2-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-2-(4,)-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-3-(5,)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-2-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Multinomial-True-initial_shape2-proposed_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-6]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-2-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-1-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-3-(2, 3)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-20-clamped]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-3-()-(4,)]
0.00s call     tests/distributions/test_stable.py::test_shape[batch_shape1-sample_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-2-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-3-()-(3, 2)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-2-unclamped]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->b-2]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-2-unclamped]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape0-sample_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-4]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.05]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-3-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-3-2-1]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape0-sample_shape0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[10.0-1.0-2.0]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-20-unclamped]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ad-1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-10-clamped]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-3-1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-5-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-10-unclamped]
0.00s call     tests/nn/test_module.py::test_decorator[1-AttributeModel]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-1.0-2.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-1-(4,)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-4]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-5-clamped]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-3-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-3-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-10.0-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-1.0-1.0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-3-(2, 3)]
0.00s call     tests/ops/test_special.py::test_log_beta_stirling[1.0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-3-(5,)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-5-unclamped]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[5.0-5-1.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-2-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[10.0-1.0-0.5]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-2-(2, 3)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-10.0-2.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[5.0-5-0.5]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-2-()-(3, 2)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->b-1]
0.00s call     tests/distributions/test_stable.py::test_shape[batch_shape2-sample_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-3-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Categorical]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-5-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-2-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(2,)-(6,)-9]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-1]
0.00s call     tests/distributions/test_tensor_type.py::test_double_type
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-5-clamped]
0.00s call     tests/distributions/test_stable.py::test_shape[batch_shape0-sample_shape2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-2-(2, 3)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-10-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-20-unclamped]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-2-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-2]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-20-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-20-unclamped]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-1-1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-6]
0.00s call     tests/distributions/test_stable.py::test_shape[batch_shape0-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[NaiveBeta]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-20-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-2-clamped]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-2-()-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-1-2-()]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape0-sample_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-10-clamped]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->c-2]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-3-()-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-1-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Bernoulli]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-3-2-1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-10-unclamped]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[91]
0.00s call     tests/infer/test_valid_models.py::test_dim_allocation_error[TraceEnum_ELBO-True]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[i->i]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-1-2-()]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-10-clamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-2-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-()-()-9]
0.00s call     tests/nn/test_module.py::test_constraints[shape20-constraint_20]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-()-()-9]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.55-0.44]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-2-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-3-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-3-(2, 3)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-2-clamped]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape5]
0.00s call     tests/poutine/test_properties.py::test_idempotent[trace-normal_normal_model]
0.00s call     tests/distributions/test_one_hot_categorical.py::TestOneHotCategorical::test_support
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-3-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-2-(5,)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(2,)-(6,)-9]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[3]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-0-2-()]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-2-2-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-3-(2, 3)]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-10-clamped]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->d-1]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-10-unclamped]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-5-unclamped]
0.00s call     tests/nn/test_module.py::test_decorator[2-AttributeModel]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[81]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[93]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-20-unclamped]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Beta]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-2-1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->b-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-3-(2, 3)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->a-2]
0.00s call     tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-Trace_ELBO]
0.00s call     tests/infer/mcmc/test_mcmc_util.py::test_model_with_subsample[5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-1-1]
0.00s call     tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-Trace_ELBO]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[3-2-()]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-1-2-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-3-2-7]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-()-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-1-1]
0.00s call     tests/distributions/test_relaxed_straight_through.py::test_bernoulli_shapes[probs0]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(4,)-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-2-(2, 3)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->ac-2]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->ac-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-3-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(4,)-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-3-(2, 3)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[95]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-2-2-()]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[94]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-1-1-()]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[w->]
0.00s call     tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-2-unclamped]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-2-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-3-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[2-2-()]
0.00s call     tests/distributions/test_spanning_tree.py::test_log_prob[1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-2-(5,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-6]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-0-2-()]
0.00s call     tests/ops/test_tensor_utils.py::test_dct[shape0]
0.00s call     tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs4]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-2-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-2-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-2-2-()]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[99]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-2-()]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-3-()-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-2-1-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-2-1-()]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-TraceEnum_ELBO]
0.00s call     tests/ops/test_tensor_utils.py::test_idct[shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[96]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-condition-bernoulli_normal_model]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[48]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[1-0-1-()]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->c-1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->ac-1]
0.00s call     tests/distributions/test_relaxed_straight_through.py::test_bernoulli_shapes[probs1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-2-1-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(3, 2)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-2-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-2-()-(4,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[1.0-1]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 1, 1)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[OneHotCategorical]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-2-(5,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-2-1-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-0-1-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-5]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-replay-bernoulli_normal_model]
0.00s call     tests/poutine/test_properties.py::test_idempotent[block-bernoulli_normal_model]
0.00s call     tests/distributions/test_stable.py::test_shape[batch_shape1-sample_shape0]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-0-2-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[0-1-1-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-1-1-()]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[90]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-4]
0.00s call     tests/ops/test_special.py::test_log_binomial_stirling[1e-08]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->a-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[87]
0.00s call     tests/ops/test_gamma_gaussian.py::test_pad[2-0-1-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-2-(2, 3)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[83]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Multinomial]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-1-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[98]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.5]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape6]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[86]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-1.0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[89]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.5]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[88]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-1.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-2-()-(3, 2)]
0.00s call     tests/ops/test_special.py::test_log_binomial_stirling[0.01]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[3-3-()]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Poisson_2D]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.05]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,x->]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-1-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->a-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.5]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[92]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape0-sample_shape1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.05]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[85]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-1-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-3]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[97]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.05]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-3-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[82]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[65]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-3-()-(3, 2)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-3-7]
0.00s call     tests/ops/test_indexing.py::test_value[()-()-(4, 1, 1)-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.05]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[2-3-()]
0.00s call     tests/nn/test_module.py::test_constraints[shape18-constraint_18]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.2]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->a-2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-20]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-3-(5,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[79]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[78]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 1, 1)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-6]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-1-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-3-()-(3, 2)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.5]
0.00s call     tests/distributions/test_lowrank_mvn.py::test_log_prob
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-2-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-1.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-3-()-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[VonMises-list-sample_shape2]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape2-event_shape2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[77]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[46]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-1-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-1.0]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(2,)-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-50]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-2-()-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-2-(5,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[VonMises-list-sample_shape0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.05]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-1-1-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.5]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[52]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape2-event_shape2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.2]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(2,)-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-1.0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[55]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[76]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.5]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[75]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-2-()-(3, 2)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.1]
0.00s call     tests/ops/test_contract.py::test_ubersum_total[ai->-i]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-3-()]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(2,)-(6,)-9]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[84]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[14]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.05]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-1-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.05]
0.00s call     tests/infer/test_enum.py::test_unsafe_bernoulli[sample_shape1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.2]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-()-()]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[5-()-(4, 5)-9]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[73]
0.00s call     tests/ops/test_contract.py::test_ubersum_jit[->-]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-2-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-50]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-w,x->]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.5]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(2,)-(3, 1)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[43]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[marginal-ab,bc->-]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-1.0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[70]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-1-()-(3, 2)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-50]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[74]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-2-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.2]
0.00s call     tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[()-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.5]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(3, 1)-()]
0.00s call     tests/distributions/test_stable.py::test_shape[batch_shape2-sample_shape0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-50]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(2,)-()]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-()-()]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-()-(2,)-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.05]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi->i]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-1.0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-50]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[67]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-2-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-1.0]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(2,)-()-(7,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[9]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[58]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[51]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-replay-normal_normal_model]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-2-(2, 3)]
0.00s call     tests/ops/test_indexing.py::test_value[()-(2,)-(2,)-(7,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-20]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.5]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape6]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[53]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.2]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-condition-normal_normal_model]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.2]
0.00s setup    tests/contrib/tracking/test_em.py::test_svi_multi
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-2-1-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-2-1-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-10]
0.00s call     tests/distributions/test_pickle.py::test_pickle[ContinuousBernoulli]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.1]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-()-(3, 1)-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.1]
0.00s call     tests/ops/test_tensor_utils.py::test_precision_to_scale_tril[batch_shape0-event_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[61]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[40]
0.00s call     tests/poutine/test_properties.py::test_idempotent[block-normal_normal_model]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.2]
0.00s call     tests/poutine/test_properties.py::test_idempotent[replay-bernoulli_normal_model]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[64]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-10]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[69]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-3-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-1-1-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.5]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-3-(5,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[59]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.05]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[28]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.2]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape3-sample_shape0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-1.0]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(2,)-(2,)-(7,)]
0.00s call     tests/ops/test_indexing.py::test_value[()-()-(2,)-(7,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-1.0]
0.00s call     tests/distributions/test_haar.py::test_haar_ortho[7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[66]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[DirichletMultinomial]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(2,)-(6,)-9]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[44]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-ij->ji]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-6]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-0-1-()]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-()-(6,)-9]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[1-2-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.5]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[72]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-2-1-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-2]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(3, 1)-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.2]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-0-1-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-50]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-20]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.5]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-3-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-3-(5,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-50]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-3-2-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-20]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[RejectionExponential]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,ab->ab-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-1-7]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-1-1-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-2]
0.00s call     tests/ops/test_special.py::test_log_beta_stirling[1e-08]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[LogNormal]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-10]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-3-2-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-10]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[45]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[8]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-50]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-3-()]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[56]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-3-()]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[3-()-(4, 5)-9]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-2-(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[68]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[50]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[5]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-,ab->ab-2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[71]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-1]
0.00s call     tests/ops/test_special.py::test_log_binomial_stirling[0.0001]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[31]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape6]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[2-()-(4, 5)-9]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-5]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[VonMises]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[49]
0.00s call     tests/ops/test_indexing.py::test_value[()-(3, 1)-(2,)-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-50]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[17]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 1, 1)-()]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape4]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-2-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[11]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-20]
0.00s call     tests/ops/test_tensor_utils.py::test_dct[shape2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-3-(2, 3)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[47]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-2-1-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-3-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-20]
0.00s call     tests/ops/test_gaussian.py::test_pad[2-1-1-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[63]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-()-()-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-1-1-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-5]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-5]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[23]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-5]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-3-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-2-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-1-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-1]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-0-1-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-10]
0.00s call     tests/ops/test_gaussian.py::test_pad[1-2-1-()]
0.00s call     tests/distributions/test_spanning_tree.py::test_partition_function[3]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape6]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-3-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-5]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[37]
0.00s call     tests/ops/test_gaussian.py::test_pad[0-2-1-()]
0.00s call     tests/ops/test_special.py::test_log_beta_stirling[1e-06]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-2-()]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[27]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[62]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-20]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[42]
0.00s call     tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-()-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-4]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-20]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-3-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-2-(2, 3)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-20]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-ij->ji]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-4]
0.00s call     tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-()-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-3-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-3-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-2-()]
0.00s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[1-size2]
0.00s call     tests/ops/test_gaussian.py::test_add[3-(3, 2)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[34]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[54]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-10]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[29]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-2-(5,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-10]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-10]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[60]
0.00s call     tests/ops/test_indexing.py::test_value[()-(2,)-()-(7,)]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape6]
0.00s call     tests/infer/test_jit.py::test_backward
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-2-7]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape0]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Beta]
0.00s call     tests/contrib/autoname/test_scoping.py::test_recur_multi
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-3-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-1-1]
0.00s call     tests/ops/test_indexing.py::test_value[()-(2,)-(3, 1)-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-10]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-1-()-(3, 2)]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-2-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-10]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[41]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-1-1-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-2]
0.00s call     tests/infer/test_jit.py::test_grad_expand
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-3-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-False-list]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-5]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[19]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[36]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[39]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-2-(5,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-5]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-4]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-1-()-(3, 2)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-tuple-sample_shape2]
0.00s call     tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape0-gelman_rubin]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-3]
0.00s call     tests/distributions/test_von_mises.py::test_log_prob_normalized[0.1]
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_three_layers_model_permuted
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape3-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-list-sample_shape2]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-6]
0.00s call     tests/distributions/test_spanning_tree.py::test_partition_function[1]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-1]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-6]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-1-1-7]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-4]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[LogNormal]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-4]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape2-sample_shape0]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape6]
0.00s call     tests/infer/test_valid_models.py::test_no_plate_enum_discrete_batch_error
0.00s call     tests/ops/test_gaussian.py::test_marginalize[2-1-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-3]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-5]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-()-()]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape3]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-Size-sample_shape2]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-2]
0.00s call     tests/poutine/test_properties.py::test_idempotent[replay-normal_normal_model]
0.00s call     tests/ops/test_special.py::test_log_binomial_stirling[1e-06]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-3-()-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-1-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-list-sample_shape1]
0.00s call     tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[TraceEnum_ELBO]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[26]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[3-1-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-2-1-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(4,)-()]
0.00s call     tests/ops/test_gaussian.py::test_marginalize[1-3-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[VonMises-tuple-sample_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[24]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-3-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-Size-sample_shape1]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-2]
0.00s call     tests/ops/test_gaussian.py::test_add[2-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-3-()-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-2-()-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-2-1-1]
0.00s call     tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[TraceTMC_ELBO]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-2-()]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[35]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-list-sample_shape0]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal[(2, 3)-(2, 3)]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape2-sample_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[16]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[13]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-1-()-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[20]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-3-(2, 3)]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[VonMises-Size-sample_shape0]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-()-()]
0.00s call     tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-5]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[VonMises-Size-sample_shape2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-2-()-(4,)]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape3-sample_shape0]
0.00s call     tests/params/test_module.py::test_random_module[outer]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-3-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[6]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-tuple-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-Size-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-2-1-7]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-do-normal_model]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[3-(3, 2)-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[VonMises-list-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Poisson]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[VonMises-tuple-sample_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[32]
0.00s call     tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape0-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-tuple-sample_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[21]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Binomial]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->-2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[30]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-3-()-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-1-()-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[38]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[4]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[22]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 2)-()-9]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-2-()-(4,)]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[sample-ab,bc->-]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-1-()-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[25]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-3-(5,)]
0.00s call     tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape0-_cummin]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-2-(2, 3)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 2)-()-9]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[18]
0.00s call     tests/infer/test_enum.py::test_unsafe_bernoulli[sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-2-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-3-(5,)]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(3, 1)-()]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-True]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[NaiveBeta]
0.00s call     tests/infer/test_jit.py::test_grad
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-()-(3, 1)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[ExtendedBinomial]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[True-p_mask3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-2-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[VonMises-Size-sample_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-1-()-(4,)]
0.00s call     tests/ops/test_special.py::test_log_beta_stirling[0.01]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-False-Size]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-3-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[12]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-3-(5,)]
0.00s call     tests/ops/test_special.py::test_log_beta_stirling[0.0001]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-3-(2, 3)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-False-tuple]
0.00s call     tests/ops/test_contract.py::test_ubersum_total[i->i-i]
0.00s call     tests/distributions/test_unit.py::test_expand[batch_shape2-sample_shape0]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-p_mask3]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[15]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape1-sample_shape2]
0.00s call     tests/contrib/test_util.py::test_get_indices_sizes
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-(2, 3)-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-2-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-2-()]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 2)-()-9]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-3-(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-2-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-2-(2, 3)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-True-list]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-2-(2, 3)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 1)-()-9]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[VonMises-tuple-sample_shape0]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape1-event_shape1]
0.00s call     tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[7]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(2,)-()-9]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 1)-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-3-(2, 3)]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-3-(5,)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-()-()-9]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-3-(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-3-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_features[10]
0.00s call     tests/poutine/test_properties.py::test_idempotent[trace-normal_model]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 2)-()-9]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[marginal-ai,ai->i-i]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-2-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-2-(2, 3)]
0.00s call     tests/distributions/test_one_hot_categorical.py::TestOneHotCategorical::test_support_non_vectorized
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-10-10]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-()-(2,)-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-2-(2, 3)]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Geometric]
0.00s call     tests/infer/test_valid_models.py::test_dim_allocation_error[Trace_ELBO-False]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-3-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-3-(5,)]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_4[Trace_ELBO]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-3-2-1]
0.00s call     tests/poutine/test_mapdata.py::test_replay[plate-20]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-3-2-7]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_4[TraceEnum_ELBO]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,w->]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-True-tuple]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-TraceEnum_ELBO]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,ab->ab-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-3-(2, 3)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(2,)-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-3-2-7]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[finite-,ab->ab-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-2-1-7]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[17]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-1-()-(4,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-1-()-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-2-1-1]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape2-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-True-Size]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape1-event_shape1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-2-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-2-1-7]
0.00s call     tests/ops/test_stats.py::test_cummin
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-3-(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-3-2-1]
0.00s call     tests/ops/test_gaussian.py::test_add[3-(4,)]
0.00s call     tests/distributions/test_von_mises.py::test_log_prob_normalized[1.0]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape2-sample_shape0]
0.00s call     tests/ops/test_indexing.py::test_value[()-(3, 1)-(3, 1)-()]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ShapeAugmentedBeta]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size2-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-3-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-3-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-2-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-2-()]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[1.0-10]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-3-(5,)]
0.00s call     tests/ops/test_indexing.py::test_value[()-(3, 1)-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-1-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-2-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-1-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-3-()]
0.00s call     tests/ops/test_gaussian.py::test_add[1-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-2-(2, 3)]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc->ac-2]
0.00s call     tests/distributions/test_unit.py::test_expand[batch_shape1-sample_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(2,)-()-9]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-2-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-3-2-1]
0.00s call     tests/ops/test_indexing.py::test_value[()-()-(3, 1)-()]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal[(4,)-(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-1-1-7]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 2)-()-9]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-1-()-(3, 2)]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape2-event_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-3-(5,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[5.0-5]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-1-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-2-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-3-()]
0.00s call     tests/contrib/gp/test_kernels.py::test_kernel_forward[WhiteNoise1]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-TraceTMC_ELBO]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape3-sample_shape1]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape1-event_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 1)-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-2-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-2-(5,)]
0.00s call     tests/ops/test_contract.py::test_ubersum_total[,i->-i]
0.00s call     tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape1]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-TraceTMC_ELBO]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-()-(6,)-9]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5-0.9]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 2)-()-9]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 2)-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-3-1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 2)-()-9]
0.00s call     tests/distributions/test_tensor_type.py::test_float_type
0.00s call     tests/ops/test_gaussian.py::test_marginalize[1-1-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-3-2-1]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7-0.5]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_2[TraceEnum_ELBO]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-1-1-7]
0.00s call     tests/infer/test_valid_models.py::test_iplate_variable_clash_error[TraceTMC_ELBO]
0.00s call     tests/distributions/test_von_mises.py::test_log_prob_normalized[30.0]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 1)-()-9]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(2,)-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-2-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-10-6]
0.00s call     tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-2-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-2-()]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-replay-normal_model]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape3-sample_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-3-2-7]
0.00s call     tests/poutine/test_mapdata.py::test_cond_indep_stack[iplate-5]
0.00s call     tests/poutine/test_poutines.py::test_enumerate_poutine[0--3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[10.0-1]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[True-True]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-()-()-9]
0.00s call     tests/distributions/test_mask.py::test_mask_noop[(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-3-2-7]
0.00s call     tests/ops/test_indexing.py::test_value[()-()-(2,)-()]
0.00s call     tests/ops/test_gaussian.py::test_add[2-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-3-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-2-1-7]
0.00s call     tests/ops/test_indexing.py::test_value[(3, 1)-()-()-()]
0.00s call     tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype0-event_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-5]
0.00s call     tests/poutine/test_poutines.py::UnconditionHandlerTests::test_undo_uncondition
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-1-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-6]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-()-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-2-1-1]
0.00s call     tests/infer/test_valid_models.py::test_dim_allocation_error[TraceEnum_ELBO-False]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-3-(3, 2)]
0.00s call     tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape1]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_4[TraceGraph_ELBO]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc->ac-1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-()-(6,)-9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-3-2-1]
0.00s call     tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-2-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-2-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-2-1-7]
0.00s call     tests/poutine/test_properties.py::test_idempotent[block-normal_model]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape1-sample_shape2]
0.00s call     tests/distributions/test_categorical.py::TestCategorical::test_support
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-3-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-w,w->]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-,w->]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-6-10]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-()-()-9]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape2-event_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-3-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-3-7]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[29]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-3-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-3-(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-1-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-3-2-1]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[2-True]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-False-NUTS-_empty_model]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-5-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-10-5]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape2-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape1-sample_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(2,)-()-9]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-3-()]
0.00s call     tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-()-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-3-(2, 3)]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape2-event_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-1-1-7]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape2-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ShapeAugmentedBeta]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-3-1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 1)-()-9]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-3-(5,)]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal[(2, 3)-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-2-(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-2-1-1]
0.00s call     tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype1-event_shape2]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,ixj->ij]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-3-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-1-1-1]
0.00s call     tests/poutine/test_poutines.py::test_enumerate_poutine[0--1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 2)-()-9]
0.00s call     tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype2-event_shape2]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 2)-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-2-(5,)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 2)-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-2-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-10-3]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal[()-(2, 3)]
0.00s call     tests/poutine/test_poutines.py::BlockHandlerTests::test_block_tutorial_case
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal[(4,)-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Uniform]
0.00s call     tests/contrib/tracking/test_hashing.py::test_aps_hash[100]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-1-1-1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 2)-()-9]
0.00s call     tests/ops/test_stats.py::test_pi
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-2-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-1-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-10-4]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-()-(6,)-9]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-4-10]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-1-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-False-Size]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 1)-()-9]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-()-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-3-7]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[23]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-5-2]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape3-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-3-(2, 3)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-6-6]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(2,)-()-9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-3-2]
0.00s call     tests/infer/test_valid_models.py::test_dim_allocation_error[TraceGraph_ELBO-False]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-3-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-True-Size]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-1-1-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,ixj->ji]
0.00s call     tests/params/test_module.py::test_module_nn[outest]
0.00s call     tests/distributions/test_pickle.py::test_pickle[LowRankMultivariateNormal]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-4-2]
0.00s call     tests/ops/test_indexing.py::test_value[()-()-()-(7,)]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[3-(3, 2)-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-2-1-7]
0.00s call     tests/contrib/tracking/test_hashing.py::test_aps_hash[10]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-2-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-True-tuple]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-3-2-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-2-()]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[27]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-2-1-1]
0.00s call     tests/contrib/tracking/test_hashing.py::test_aps_hash[1]
0.00s call     tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full_expose
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-2-10]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(2,)-(2,)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-3-2-7]
0.00s call     tests/distributions/test_von_mises.py::test_log_prob_normalized[0.3]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[25]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-3-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-3-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwj->ji]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape3-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-False-tuple]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-1-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-3-10]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-True-list]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-3-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-wi,xi->i]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-False-list]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-3-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-3-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-3-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-3-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-10-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-2-7]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[sample-ai,ai->i-i]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-3-()]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[28]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape0-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-2-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-1-7]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_mvn_shape[3-(2, 3)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-10-2]
0.00s call     tests/nn/test_module.py::test_constraints[shape2-constraint_2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-3-()]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional[0]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape2-sample_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-10-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-4-5]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape0-event_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-5-6]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[3-(2, 3)]
0.00s call     tests/poutine/test_poutines.py::TraceHandlerTests::test_trace_return
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-3-2-1]
0.00s call     tests/ops/test_stats.py::test_autocorrelation
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[HalfCauchy]
0.00s call     tests/ops/test_indexing.py::test_value[()-(2,)-(2,)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-1-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-2-1-1]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-True-HMC-_empty_model]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-1-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-5-10]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-2-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-2-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-2-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-1-7]
0.00s call     tests/ops/test_gaussian.py::test_add[1-(4,)]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-condition-normal_model]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-10-6]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-3-2-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-2-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-6-10]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape1-event_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-2-1-1]
0.00s call     tests/poutine/test_poutines.py::test_enumerate_poutine[0--2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-4-10]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-3-7]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape1-event_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-6-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-5-5]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-3-2-1]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-0.5]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-3-2-1]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[24]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-2-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-4]
0.00s call     tests/distributions/test_von_mises.py::test_log_prob_normalized[10.0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-3-(2, 3)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-True-initial_shape0-proposed_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[18]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-6-5]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-3-2-7]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[26]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-3-2-1]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_mvn_shape[3-(4,)]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[21]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[3-(3, 2)-(4,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-3-1]
0.00s call     tests/distributions/test_spanning_tree.py::test_partition_function[2]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape1-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[NaiveDirichlet]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-2-1-1]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[19]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-1-1-7]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[22]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-10-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-10-4]
0.00s call     tests/poutine/test_poutines.py::BlockHandlerTests::test_block_partial_hide
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[14]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-3-10]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-3-2-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-()-()]
0.00s call     tests/distributions/test_von_mises.py::test_log_prob_normalized[3.0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-2-1-7]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-()-()]
0.00s call     tests/distributions/test_von_mises.py::test_log_prob_normalized[100.0]
0.00s call     tests/nn/test_module.py::test_names
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[16]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-2-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-2-1-1]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-0.9]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-10.0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-10-10]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-3-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-10]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-10.0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-3-(5,)]
0.00s call     tests/test_generic.py::test_not_implemented[minipyro]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-1.0]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-()-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-2-1-7]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-()-()-()]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape1-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-2-1-1]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape2-sample_shape1]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-0.1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-6-4]
0.00s call     tests/infer/test_inference.py::SafetyTests::test_duplicate_obs_name
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape1-event_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-3-2-7]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape0-sample_shape2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-4-6]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-2-(2, 3)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-False-list]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[2-(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-2-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-10]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-1-7]
0.00s call     tests/ops/test_indexing.py::test_value[(2,)-(2,)-()-()]
0.00s call     tests/ops/test_indexing.py::test_value[()-(2,)-()-()]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-0.9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-2-1]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[20]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-10.0]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-2.0]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-1.0]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ShapeAugmentedDirichlet]
0.00s call     tests/nn/test_module.py::test_constraints[shape17-constraint_17]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-2.0]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-1.1]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-0.5]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-0.1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-1-7]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-1.0]
0.00s call     tests/distributions/test_pickle.py::test_pickle[AVFMultivariateNormal]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape2-event_shape2]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 1)-()-9]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-1.1]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape0-event_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-1-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-5-6]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape0-event_shape2]
0.00s call     tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype0-event_shape1]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ShapeAugmentedDirichlet]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-0.9]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-6-3]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-0.5]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-True-2200]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-3-(2, 3)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-8]
0.00s call     tests/poutine/test_poutines.py::test_decorator_interface_queue
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-2.0]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-1.1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 1)-()-9]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-2-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-True-initial_shape1-proposed_shape1]
0.00s call     tests/poutine/test_poutines.py::TraceHandlerTests::test_trace_full
0.00s call     tests/nn/test_module.py::test_constraints[shape1-constraint_1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-3-1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-,w->]
0.00s call     tests/ops/test_tensor_utils.py::test_idct[shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-()-()]
0.00s call     tests/ops/test_contract.py::test_contract_to_tensor[example1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-5-4]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[2-(3, 2)-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-8]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-3-1]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-1-1-1]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[13]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-1-1-7]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape3-sample_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-1-7]
0.00s call     tests/ops/test_stats.py::test_autocovariance
0.00s call     tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-True-initial_shape2-proposed_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-False-Size]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-False-tuple]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-3-6]
0.00s call     tests/pyroapi/test_pyroapi.py::test_generate_data_plate[pyro]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape1-event_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-3-2-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-4-6]
0.00s call     tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-0.1]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal[()-(4,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-2-6]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[15]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-8]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-3-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-3]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 1)-()-9]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-6-6]
0.00s call     tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype1-event_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-3-5]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[12]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-2]
0.00s call     tests/contrib/tracking/test_hashing.py::test_aps_hash[0.1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-3-(5,)]
0.00s call     tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype0-event_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[11]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-True-NUTS-_empty_model]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[10]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-3-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-1-1]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[DirichletMultinomial]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-2-(5,)]
0.00s call     tests/contrib/test_minipyro.py::test_generate_data_plate[pyro]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-True-initial_shape0-proposed_shape0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-6-2]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ShapeAugmentedGamma]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-3-6]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-10]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[2-(3, 2)-()]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[HalfCauchy]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-2-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-()-()]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[3-event_shape2]
0.00s call     tests/poutine/test_properties.py::test_idempotent[replay-normal_model]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 1)-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-1-7]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 1)-()-9]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-6-2]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[HalfCauchy]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[3-(4,)-()]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[4]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-()-(3,)]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-False-HMC-_empty_model]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape2-event_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-4-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-()-()]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[8]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-1-1]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[10-event_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-4-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-5-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-1-7]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[4-event_shape2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-3-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-True-initial_shape1-proposed_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-4]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-()-()-9]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-3-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-3-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-3-7]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[RejectionExponential]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Categorical-True-initial_shape2-proposed_shape2]
0.00s call     tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype2-event_shape1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 1)-()-9]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-2-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-True-initial_shape2-proposed_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-2-1]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape1-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-5-3]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-()-()]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-10]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-1-7]
0.00s call     tests/ops/test_stats.py::test_effective_sample_size
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-2-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-6-5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-()-()]
0.00s call     tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-1-1]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[2-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-1-7]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape2-event_shape2]
0.00s call     tests/poutine/test_poutines.py::BlockHandlerTests::test_block_partial_expose
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[3-(4,)-(3, 2)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-1-1]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[marginal-a,a->-]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-2-(3, 2)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-()-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-False-initial_shape0-proposed_shape0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-3-(5,)]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[map-ai,abij->i-ij]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-3-()]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-1-2]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-()-()]
0.00s call     tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-()-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-2-(2, 3)]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-()]
0.00s call     tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-TraceGraph_ELBO]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-3-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-2-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-2-1]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges471]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-(2, 1)-(3,)]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_mvn_shape[2-(2, 3)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-parallel-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-2-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-()-(3,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-2-7]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape1-event_shape2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-1-1]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[2-event_shape2]
0.00s call     tests/distributions/test_mask.py::test_mask_noop[(4,)]
0.00s call     tests/distributions/test_categorical.py::TestCategorical::test_log_prob_sum
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-3-2-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-2-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-3-(2, 3)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-6-4]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-2-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-8]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(2,)-()-9]
0.00s call     tests/distributions/test_coalescent.py::test_sample_is_valid[3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-2-5]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-7-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1207]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-5-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-5-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-3-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[cold-None-1-None-1-8]
0.00s call     tests/poutine/test_poutines.py::QueueHandlerMixedTest::test_queue_single
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-1-1]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape3-sample_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-()-()-9]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-False-initial_shape1-proposed_shape1]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-None-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-3-1]
0.00s call     tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype1-event_shape0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-()-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-5-4]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-3-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-6]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-True-initial_shape1-proposed_shape1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-2]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(2,)-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-2-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-3-5]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-(2, 3)-(2, 3)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-3-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-4-5]
0.00s call     tests/distributions/test_conjugate_update.py::test_gamma_poisson[()-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-sequential-parallel-None]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-3-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-parallel-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-()-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-parallel-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-1-1]
0.00s call     tests/distributions/test_haar.py::test_haar_ortho[4]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-3-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-3-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-3-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-3-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-3-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-parallel-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-2-4]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-3-4]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-parallel-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-True-initial_shape0-proposed_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(2,)-()-9]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-()-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-parallel-parallel-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size0-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-3-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-False-0-1000-False-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-3-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-3-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-8]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-2-(5,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-sequential-parallel-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-10]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-()-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-None-sequential-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-sequential-sequential-None]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape2-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-2-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-3-1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(2,)-()-9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-3]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-p_mask2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-3-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-3-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-(3,)-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-4-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-2-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-3-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-3-7]
0.00s call     tests/nn/test_module.py::test_constraints[shape3-constraint_3]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape1-sample_shape0]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-1-7]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape5]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-2-(2, 3)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-3-(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-(3,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-(2, 1)-(3,)]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_mvn_shape[1-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-2-1-7]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[3-(4,)-(4,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-3-7]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape4]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-3-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-3-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-2-(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-(3,)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-6]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-3-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-1]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ShapeAugmentedGamma]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-(3,)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-4-3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-3-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-True-initial_shape2-proposed_shape2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-False-unscaled]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-2-(5,)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-()-()-9]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-3-4]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-2-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-(3,)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-2-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-2-3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-2-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Categorical-True-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-()-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-3-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape2-event_shape2]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-False-1800]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-3-7]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(2,)-()-9]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-3-(5,)]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-()-()-9]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-2-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-3-1]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape2-event_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-3-2-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-8]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Categorical-True-initial_shape0-proposed_shape0]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwj->ij]
0.00s call     tests/ops/test_tensor_utils.py::test_dct[shape1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-4-3]
0.00s call     tests/ops/test_contract.py::test_contract_tensor_tree[example2]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape2]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-5-uniform]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-1-1]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-Trace_ELBO-full]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-()-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-5]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-2-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Categorical]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-1-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-1-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape2]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-p_mask3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-2-(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-3-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-2-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-True-initial_shape1-proposed_shape1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-6]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape5]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-3-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-1-7]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape1]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape0-sample_shape1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-3-7]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape4]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-3-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-ij->ij]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-5]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-3-7]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-p_mask2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-1-1]
0.00s call     tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype2-event_shape0]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-3-1]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[2-(4,)-(3, 2)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-2-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-(3,)-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-4]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-(2, 1)-(3,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-False-0-1000-False-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-3-2]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-3-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-3-1]
0.00s call     tests/infer/test_valid_models.py::test_variable_clash_in_model_error[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-(3,)-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-False-0-1000-False-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-1-7]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence
0.00s call     tests/poutine/test_poutines.py::LiftHandlerTests::test_random_module
0.00s call     tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-3-1]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[2-(4,)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-3-(3, 2)]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[True-p_mask2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-3-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-1-1]
0.00s call     tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full_hide
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-2-5]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-1-1]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[3-(4,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[full-()-3-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-1-7]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-()-()]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-4]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-None-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-3-2]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[1-(2, 3)]
0.00s call     tests/test_util.py::test_warn_if_inf
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-6]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape4]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-7]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.7-0.0-0.5]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-3-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-2-(3, 2)]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape5]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-True]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-1-7]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-do-trivial_model]
0.00s call     tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[TraceGraph_ELBO]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[LKJCorrCholesky]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-3-7]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-3-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape4]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-1-7]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape5]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-1-1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-True-0-1000-False-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges163]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-(2, 3)-(2, 3)]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-True-1800]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-3-7]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[map-a,abi->-i]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape4]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-()-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-3-1]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape1-event_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-1-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape5]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-3-2]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-w->]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Bernoulli]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-3-1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(2,)-()-9]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-None-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-5]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-False-sample_shapes1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-()-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-None-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-3-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-None-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-1-7]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-None-plate]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[same-()-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-1-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-(3,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-sequential-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-3-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-3-(4,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-3-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-3-7]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[2-(4,)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-3-(3, 2)]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape2-event_shape1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-nonreparam-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-3-7]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-nonreparam-unscaled]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-3-1]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-20-plate]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-1-7]
0.00s call     tests/poutine/test_poutines.py::LiftHandlerTests::test_random_module_prior_dict
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape2-event_shape1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-2-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-()-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-2-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-3-2]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag[mat_size0-2-batch_shape0]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape1-event_shape1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-3-7]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_mvn_shape[2-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-3-(5,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-1-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-3-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-1-7]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape2]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-()-(3,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-False-0-1000-False-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-10]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-1-1]
0.00s call     tests/ops/test_contract.py::test_contract_tensor_tree[example3]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-7]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape2]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape0]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape1]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape1-event_shape1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[10.0-1.0]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-3-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-2-(4,)]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-True-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-3-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape2]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-6]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-1-1]
0.00s call     tests/ops/test_contract.py::test_contract_tensor_tree[example1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-1-1]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape2-event_shape0]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-False-0-1000-False-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-()-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-()-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-False-0-1000-False-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-()-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-(3,)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-3-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape0]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape5]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[laplace-2-100]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_1_gate[0.1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-()-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-parallel-sequential-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-3-7]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Normal]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-1-1]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_1_gate[0.9]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-3-(5,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-parallel-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-parallel-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-3-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-parallel-sequential-None]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-()-()-9]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-()-(3,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-1-1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-1-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-ij->ij]
0.00s call     tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape0]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-3-1]
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-bi,abi->b-i]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-(3,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-sequential-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-2]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape1-event_shape2]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-(3,)-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-3-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-wi,wi->i]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-2-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-()-()]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-4]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_1_gate[1.1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-4]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-None-parallel-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-parallel-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-3-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[1.0-10.0]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-1-2]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-True-0-1000-False-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-1-7]
0.00s call     tests/ops/test_contract.py::test_naive_ubersum[i->,i-i]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-1-7]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_1_gate[2.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-sequential-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-(3,)-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-(3,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-1-1]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-EnergyDistance_noprior]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_1_gate[0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[5.0-5.0]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-3-1]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape2-event_shape1]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape2-event_shape1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-()-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-()-(3,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-True-0-1000-False-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LogNormal-True-initial_shape0-proposed_shape0]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-1-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-sequential-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-1-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-1-1]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-None-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-()-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-2-(5,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-None-parallel-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-3-7]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-20-plate]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[10-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-(3,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-3-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-4]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Beta]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-1-7]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[4-event_shape1]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[1-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-()-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-()-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-1-1]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-None-plate]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-()-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-3-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[3-2-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-()-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-parallel-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-1-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[1.0-1.0]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-False-2200]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-1-1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-,ab->ab-2]
0.00s call     tests/ops/test_contract.py::test_contract_tensor_tree[example0]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-(2, 1)-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-parallel-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-1-(3, 2)]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape0-event_shape2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-None-parallel-None]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape0-0.5-0.6-0.4]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-3-7]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape0-event_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-3]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[11-()-(7,)-9]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-None-iplate]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-0-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-(3,)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-1-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-3-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-True-0-1000-False-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-1-7]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[3-event_shape1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-(3,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-()-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-sequential-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-()-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-()-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-()-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-3-1]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape0-event_shape2]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-()-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-None-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-3-7]
0.00s call     tests/poutine/test_properties.py::test_idempotent[block-trivial_model]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-(3,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-()-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-()-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-sequential-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-()-(3,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-True-0-1000-False-1]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-parallel-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-3-7]
0.00s call     tests/distributions/test_pickle.py::test_pickle[MultivariateNormal]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-1-7]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_1_gate[10.0]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-(3,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-(2, 1)-(3,)]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-3-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-sequential-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-1-1]
0.00s call     tests/nn/test_module.py::test_constraints[shape19-constraint_19]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-(3,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-3-1]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape2-event_shape0]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-1-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-(3,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-(3,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-None-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-2-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-3-7]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape0-event_shape1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-()-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-sequential-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-1-7]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,ab->ab-2]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[2-event_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-3-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-3-7]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-4]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-parallel-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-(3,)-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-1-7]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape0-event_shape1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-()-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-(3,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-(3,)-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-sequential-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-3-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-2-(3, 2)]
0.00s call     tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[Trace_ELBO]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-()-(3,)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-True-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-sequential-sequential-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-sequential-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-3]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-(2, 1)-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-()-(3,)]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-1-1]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape0-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-()-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-(3,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-()-(3,)]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-0-3]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-3-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-sequential-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-1-7]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-EnergyDistance_prior]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-()-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-7-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-()-(3,)]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-3-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-(3,)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-parallel-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-(2, 1)-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-sequential-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-3-1]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-(3,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-()-()]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-3-1]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-EnergyDistance_noprior]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-(3,)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-None-sequential-None]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-2-3-MixtureOfDiagNormals]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-2-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-3-7]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-0-7]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Geometric]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-2-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-()-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LogNormal-True-initial_shape1-proposed_shape1]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-None]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-()-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-1-7]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape1-event_shape1]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-parallel-parallel-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-3-1]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-EnergyDistance_prior]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape1-0.5-0.6-1.1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-3-1]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-sequential-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-3-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges511]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-1-7]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[1-(3, 2)-()]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-0-2]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-()-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-()-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-sequential-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-sequential-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-7-3]
0.00s call     tests/poutine/test_poutines.py::ConditionHandlerTests::test_trace_data
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-(3,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-sequential-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-1-1]
0.00s call     tests/distributions/test_haar.py::test_haar_ortho[3]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-3-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-1-(4,)]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-0-1]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape1-event_shape1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-sequential-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-parallel-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-1-7]
0.00s call     tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-sequential-sequential]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-1-7]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[1-(3, 2)-(4,)]
0.00s call     tests/distributions/test_zero_inflated.py::test_zip_1_gate[1.0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-1-7]
0.00s call     tests/nn/test_module.py::test_nested
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-sequential-None-None]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-3-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges859]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape2-event_shape0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-()-(3,)]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape0-0.5-0.6-1.1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-3-7]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges860]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-(3,)-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-True-0-1000-False-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-1-2]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-4-3-GaussianScaleMixture]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-3-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-3-2-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-sequential-None-None]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-4-3-MixtureOfDiagNormals]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1243]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-2-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-None-parallel-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-1-1]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,ab->ab-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-3-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Geometric]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-sequential-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-3-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-1-1]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-4-3-GaussianScaleMixture]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-()-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-4]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-1-1]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-()-(3,)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-nonreparam-scaled]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-3-1]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_2[Trace_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-None-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-3-1]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape0-event_shape1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-(3,)-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-1-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-()-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-1-(4,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-1-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-w,x->]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-0-1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-3]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-False-scaled]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-(3,)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-parallel-parallel-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-0-3]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-(3,)-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-2-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-1-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Gamma]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LogNormal-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-7-7]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-2-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-7-3]
0.00s call     tests/poutine/test_counterfactual.py::test_counterfactual_query[True-False-False]
0.00s call     tests/nn/test_module.py::test_module_cache
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-1-1]
0.00s call     tests/distributions/test_delta.py::TestDelta::test_mean_and_var
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-None-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-3-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-2-()]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-2-3-MixtureOfDiagNormalsSharedCovariance]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-0-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-1-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-1-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-parallel-parallel-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-3-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[NaiveDirichlet]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-1-7]
0.00s call     tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_2[TraceGraph_ELBO]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-3-1]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_mvn_shape[1-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-1-7]
0.00s call     tests/ops/einsum/test_torch_log.py::test_einsum[infinite-,ab->ab-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-(3,)-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-True-0-1000-False-2]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-1-2]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-4-3-MixtureOfDiagNormalsSharedCovariance]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-3-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-(3,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-()-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-3-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-2-0]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-1-7]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-2-()-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-1-(3, 2)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[VonMises-True-initial_shape0-proposed_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[5-()-(7,)-9]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-7-7]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape0-event_shape1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-3]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-False-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(4,)-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-1-1]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[11-(6,)-()-9]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-4]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-3-7]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-nonreparam-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-1]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-2-3-MixtureOfDiagNormalsSharedCovariance]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-(5,)-(3, 2)]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape0-event_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-3-(4,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-3-7]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-3-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-1-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-0-0]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-3-2]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-2-3-MixtureOfDiagNormalsSharedCovariance]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-3-7]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape1-event_shape0]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-(3,)-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-1-1]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-4-3-GaussianScaleMixture]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(4,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-3-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-0-0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-1-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-(3,)-()]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-4-3-MixtureOfDiagNormalsSharedCovariance]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-3-7]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-3-()-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-False-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-3-1]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-4-3-MixtureOfDiagNormalsSharedCovariance]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-3-1]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-2-()-()]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-2-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-()-(5,)]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-2-3-GaussianScaleMixture]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-1-7]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[3-()-(7,)-9]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-1-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-1-2]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape1-event_shape0]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-nonreparam-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-1-7]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape0]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-1-1]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-2-3-GaussianScaleMixture]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-0-7]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-1-1]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape0-event_shape0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-3-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[VonMises-True-initial_shape1-proposed_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-2-7]
0.00s call     tests/contrib/test_minipyro.py::test_generate_data[pyro]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-1-1]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-7]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-1-1]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-2-3-GaussianScaleMixture]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-(3,)-()]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-nonreparam-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-1-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-3-1]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-1-0]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-3-1]
0.00s call     tests/distributions/test_ig.py::test_log_prob[0.5-2.5-2.5]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[11-(2, 3)-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-1-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-1-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-3-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-ji,iwj->ji]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-2-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-(3,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(4,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-3-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-3-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-1-0]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-sequential-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-3-7]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-4]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-1-1-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-False-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-2-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-1-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-2-0]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[VonMises-True-initial_shape2-proposed_shape2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-(3,)-()]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-False-unscaled]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-None-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-2-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-1-1]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[1-event_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-1-1]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Gumbel]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-1-7]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-1-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-1-1]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-3-0.37-0.77-0.33]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape1-0.5-0.6-0.4]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-1-2]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-4-3-MixtureOfDiagNormalsSharedCovariance]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-3-()]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-3-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-3-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-()-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-3-2]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-2-3-GaussianScaleMixture]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-1-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-ij,iwj->ij]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-False-sample_shapes0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-3-1]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-1-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-parallel-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-3-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-0-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-3-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-False-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-1-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-nonreparam-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-3-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-1-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-()-(3,)]
0.00s setup    tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[False-False-100-expected0]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-1-7]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape1-event_shape0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-3-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-parallel-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-1-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-3-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Poisson]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-1-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-1-1]
0.00s call     tests/infer/test_valid_models.py::test_variable_clash_in_model_error[Trace_ELBO]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-(5,)-()-()-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-None-parallel-None]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape0-0.5-0.6-1.1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-3-7]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Normal]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-3-1]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-1-2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-3-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-2-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-(5,)-()-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-4]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-()-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-(3,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-1-7]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape1-0.5-0.6-0.4]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-1-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[2-3-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-3-1]
0.00s call     tests/infer/test_valid_models.py::test_iplate_variable_clash_error[TraceEnum_ELBO]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-3-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-2-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-2-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(5,)-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-2-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-2-1-1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-2]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-3-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-1-1]
0.00s call     tests/optim/test_optim.py::test_clippedadam_lrd[1.0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-1-1-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-sequential-sequential-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-sequential-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-1-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-(3,)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-sequential-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-1-2]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-()-()-9]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-1-1]
0.00s call     tests/ops/test_gaussian.py::test_condition[3-1-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-(3,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-(3,)-()]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-1.0-(4,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-1-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[3-3-(4,)]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[sample-a,a->-]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-()-(3,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-(3,)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-None-parallel]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-None-parallel]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-None-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-(3,)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-sequential-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-3-1]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[OneHotCategorical]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape0-0.5-0.6-0.4]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-(3,)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-None-parallel-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-1-7]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-()-()-9]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-3-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-3-7]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-1-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[3-3-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-()-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-3-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-(5,)-()-()-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-1-7]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-2-3-MixtureOfDiagNormalsSharedCovariance]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-4-3-MixtureOfDiagNormals]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Bernoulli-True-initial_shape0-proposed_shape0]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[OneHotCategorical]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-3-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-()-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-1-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-3-()-()]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-2-2]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[2-()-(7,)-9]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-(5,)-()-()-(4,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-2-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-3-7]
0.00s call     tests/distributions/test_pickle.py::test_pickle[RelaxedOneHotCategorical]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-1-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-2-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-sequential-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-3-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-2-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(2, 3)-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-1-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-()-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-3-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-2-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-(3,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-2-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-sequential-parallel-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-1-1]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[5-(2, 3)-()-9]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-3-1]
0.00s call     tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-1]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-4-3-GaussianScaleMixture]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-1-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-1-7]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape1-event_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-1-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-3-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-False-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-1-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-1-1]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape1-0.5-0.6-1.1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-3-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-1-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-None-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-2-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-(3,)-()]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-False-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-3-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[warm-None-10-None-10-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-i,wi->i]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-1-1]
0.00s call     tests/distributions/test_pickle.py::test_pickle[VonMises]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-1-1-7]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-3-0.37-0.77-0.33]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-2-uniform]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-3-7]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape0-event_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-3-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(2, 3)-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-3-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-(5,)-()-()-(3, 2)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-3-1]
0.00s call     tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape0-event_shape0]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-True-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-1-1]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-Adam-optim_kwargs0-py]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-None-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-2-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-2-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-1-7]
0.00s call     tests/contrib/test_minipyro.py::test_generate_data_plate[minipyro]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-2-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-1-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[NaiveDirichlet]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-3-2-7]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-Adam-optim_kwargs0-jit]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-()-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-()-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-2-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-1-1]
0.00s call     tests/ops/test_gaussian.py::test_condition[2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-()-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-3-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-1-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-nonreparam-unscaled]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-3-1]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-Adam-optim_kwargs0-py]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(4,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-(2, 1)-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-1-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-1-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-parallel-300]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-2-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-3-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-2-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-()-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-3-1]
0.00s call     tests/optim/test_optim.py::test_clippedadam_lrd[5.0]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-3-7]
0.00s call     tests/infer/test_valid_models.py::test_iplate_variable_clash_error[TraceGraph_ELBO]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-Adam-optim_kwargs0-jit]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-2-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-1-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-2-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-False-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-2-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-3-7]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-ClippedAdam-optim_kwargs1-jit]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-(3,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-3-7]
0.00s call     tests/ops/test_gaussian.py::test_condition[1-3-()-()]
0.00s call     tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_log_prob[True-relu]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_diagnostics[1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-3-7]
0.00s setup    tests/distributions/test_tensor_type.py::test_float_type
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-1-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-2-()]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-()-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-1-1]
0.00s call     tests/distributions/test_unit.py::test_expand[batch_shape0-sample_shape2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-1-1]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-3-7]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[1-event_shape1]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-2-()]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[cold-10-1-None-3-5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-(5,)-()-()-(4,)]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-3-0.37-0.51-0.33]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[jit-exponential-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Dirichlet]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-()-(3,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-(5,)-()-()-()-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-3-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-3-1]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceGraph_ELBO-full]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-2-1]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[svi-14-1-7-1-1]
0.00s call     tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-()-()-9]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-()-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-()-(3, 2)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-3-7]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-4-0.37-0.77-0.33]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(5,)-2]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[1-(4,)-(3, 2)]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-2-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-nonreparam-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-2-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-1-7]
0.00s call     tests/poutine/test_poutines.py::LiftHandlerTests::test_random_module_warn
0.00s call     tests/ops/test_gaussian.py::test_condition[1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-2-(2, 3)]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-ClippedAdam-optim_kwargs1-py]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-()-(5,)]
0.00s call     tests/contrib/autoname/test_scoping.py::test_nested_traces
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-1-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[2-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-(5,)-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-2-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-(5,)-()-()-(4,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-3-1]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[jit-normal-2]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-3-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-(2, 3)-(2, 3)]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape2-event_shape0]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-3-0.37-0.77-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-2-()]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0--1.0--1.0]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-1-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Cauchy]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(3, 2)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-scaled]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-3-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-3-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-1-7]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-4-uniform]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[warm-None-1-None-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(3, 2)-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[1-(4,)-(4,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-5]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-3-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Bernoulli-True-initial_shape2-proposed_shape2]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[jit-laplace-1]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-ClippedAdam-optim_kwargs1-py]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-1-1]
0.00s call     tests/distributions/test_pickle.py::test_pickle[RelaxedOneHotCategoricalStraightThrough]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-2-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-1-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-3-()]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[python-bernoulli-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-i->i]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-()-(3,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-2-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-True-initial_shape0-proposed_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[5-(6,)-()-9]
0.00s call     tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[Trace_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-3-(2, 3)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-(5,)-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-2-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-nonreparam-scaled]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[warm-7-1-7-1-7]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[LogNormal]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-()-2]
0.00s call     tests/infer/test_valid_models.py::test_iplate_variable_clash_error[Trace_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-()-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-()-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-()-(3,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-2-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-False-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-2-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[1-(4,)-()]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape0-event_shape0]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-3-()]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[cold-None-10-None-10-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-(5,)-(3, 2)]
0.00s setup    tests/distributions/test_spanning_tree.py::test_make_complete_graph[cpp-3-expected_grid1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Geometric-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(5,)-3]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-6]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-3-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-(5,)-()-()-()-(4,)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-nonreparam-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-3-(2, 3)]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[svi-None-10-None-10-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-(5,)-()-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-1-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-sequential-sequential]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape0]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[2-2-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-4-0.37-0.77-0.33]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[2-()-(4,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-2-(2, 3)]
0.00s setup    tests/poutine/test_counterfactual.py::test_counterfactual_query[True-False-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-None-parallel]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[jit-studentt-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(2, 3)-2]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional_whiten[4]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[warm-10-1-None-3-5]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype0-size0]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-3-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-True-scaled]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(5,)-3]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape2-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-()-2]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[python-normal-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-3-(5,)]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional[5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-1-1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-False-0-1000-False-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-False-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(5,)-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-False-0-1000-False-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-3-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-()-()]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-5-plate]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-0-0]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-None]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-(5,)-()-()-()-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-2-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-None-sequential]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-2-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-2]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape2-event_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[3-(2, 3)-()-9]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-sequential-parallel]
0.00s call     tests/distributions/test_unit.py::test_expand[batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-3-(5,)]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional[4]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-2-(2, 3)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-2-(2, 3)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-3-(5,)]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-None-5-10-1-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-parallel-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-None-sequential]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-2-(2, 3)]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-5-5-plate]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-None-parallel]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-i->i]
0.00s call     tests/infer/test_inference.py::SafetyTests::test_duplicate_names
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-(5,)-(4,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(5,)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(2, 3)-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-sequential-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-None-sequential]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-()-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-()-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[3-2-(4,)]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_mvn_shape[1-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(2, 3)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(2, 3)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(2, 3)-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(2, 3)-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(2, 3)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(5,)-2]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[1-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-2-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-()-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-1-7]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-6]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-(5,)-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-4-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-()-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(5,)-3]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(5,)-2]
0.00s call     tests/optim/test_optim.py::test_clippedadam_lrd[3.0]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(5,)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-()-2]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_mvn_shape[2-()]
0.00s setup    tests/ops/test_gaussian.py::test_cat[3-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(5,)-3]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[cold-None-5-10-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-2-3]
0.00s setup    tests/infer/test_util.py::test_psis_diagnostic[5-0.95-krange1-500000]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-True-scaled]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-sequential-sequential]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-3-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-2-(5,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-3-(2, 3)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-1]
0.00s setup    tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-1-7]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[2-(6,)-()-9]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-(5,)-()-()-()-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-4]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-3-0.37-0.51-0.33]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[sample-a->-]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceGraph_ELBO-subsample]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(2, 3)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(2, 3)-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-2-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-True-0-1000-False-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-nonreparam-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(2, 3)-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-2-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-3-()]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[pyro-py]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-1-2]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-4-0.37-0.77-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(2, 3)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-()-2]
0.00s setup    tests/contrib/gp/test_models.py::test_inference[VariationalGP_y1D]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-2-(2, 3)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-2-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-2-2]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceGraph_ELBO-full]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-parallel-parallel]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-3-(2, 3)]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[svi-None-5-10-1-5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-1.0]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(5,)-3]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-None-10-None-10-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-2-(5,)]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[3-(6,)-()-9]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-()-()-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(5,)-3]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-(5,)-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-(5,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-1-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-sequential-sequential]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(5,)-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-()-(4,)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-nonreparam-unscaled]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(2, 3)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(2, 3)-3]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape1-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(5,)-3]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-6]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(2, 3)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(2, 3)-3]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_MultiClass_1D]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-3-(2, 3)]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional[2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-3-(5,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-1-1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-False-0-1000-False-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-nonreparam-scaled]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.39-0.44]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-()-3]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-2-()-()]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-ClippedAdam-optim_kwargs1-jit]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-None-parallel]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(2, 3)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(5,)-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(5,)-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-(5,)-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-None-parallel]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-3-()]
0.00s call     tests/poutine/test_poutines.py::test_trace_score_parts_err_msg
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-3-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-3-2-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-5]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-scaled]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-3-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(5,)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-()-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-()-2]
0.00s call     tests/ops/test_tensor_utils.py::test_block_diag[mat_size1-1-batch_shape0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-1-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-3]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-(5,)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-parallel-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(5,)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(2, 3)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-()-2]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[normal-1-200]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-(5,)-()-()-()-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-2-(2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-(5,)-()-()-()-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-3-(2, 3)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-1-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-3-(5,)]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-10-1-None-3-5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-()-()]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-3-0.37-0.77-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(5,)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-()-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-parallel-sequential]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(5,)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-()-3]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[jit-studentt-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-None-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-parallel-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-parallel-300]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(2, 3)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(5,)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(2, 3)-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-(5,)-()-()-()-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-3-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-()-3]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional_whiten[6]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[2-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-(5,)-()-()-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(5,)-2]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_generate_data[pyro]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(2, 3)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(5,)-3]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-(2,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-(5,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-(5,)-()-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-3-0.37-0.77-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-()-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-()-3]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape0-event_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[2-(2, 3)-()-9]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(2, 3)-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-3]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-False-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(2, 3)-3]
0.00s call     tests/ops/test_tensor_utils.py::test_repeated_matmul[1-size0]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape3]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-3-()]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-()-(3,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-3-()]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[jit-laplace-2]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[jit-normal-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-(5,)-()-()-()-()-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-True-0-1000-False-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(2, 3)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(5,)-2]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-0.0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-3-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[3-()-(4,)]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[3-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-()-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-sequential-parallel-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-2-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-1-1]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-4-0.37-0.77-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(5,)-2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-2-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(2, 3)-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-()-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-()-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(2, 3)-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(2, 3)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(5,)-2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-2-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-1-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-True-0-1000-False-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-unscaled]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-2-(2, 3)]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional_whiten[0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-(5,)-()-()-()-()-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-sequential-sequential]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-2-(5,)]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-(2,)]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-parallel-sequential]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-4-random]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-()-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(2, 3)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-()-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(5,)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-3-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(2, 3)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-()-3]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape1-event_shape0]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-3-0.37-0.77-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-()-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-()-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-3-()]
0.00s call     tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-Trace_ELBO-subsample]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-parallel-None]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional_whiten[3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-3-7]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-3-0.37-0.77-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(5,)-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-1-3]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.39-0.44]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-sequential-parallel-None]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[2-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-1-3]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-3-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-2-(5,)]
0.00s call     tests/distributions/test_ig.py::test_log_prob[1.7-3.0-4.0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-(5,)-()-()-()-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-1-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-2]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-False-0-1000-False-2]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-4-0.37-0.77-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-parallel-sequential]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(5,)-3]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[cold-14-1-7-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-(5,)-(4,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-True-0-1000-False-2]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-4-0.37-0.77-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-sequential-None]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-(5,)-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-2-2]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-3]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-unscaled]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-True-scaled]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-None-None-None]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_mvn_shape[3-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-(5,)-()-()-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-2-2]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-nonreparam-scaled]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-scaled]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-3-0.37-0.77-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-3-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-2-(5,)]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional[1]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(2, 3)-2]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[svi-7-1-7-1-7]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[cold-7-1-7-1-7]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-(2,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-False-0-1000-False-2]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-wi->i]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-0.0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-1-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-(5,)-()-()-()-()-(3, 2)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-2]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-parallel-parallel]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-3-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-1-1]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-4-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-sequential-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-2-(2, 3)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Bernoulli-True-initial_shape1-proposed_shape1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-False-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(5,)-2]
0.00s call     tests/distributions/test_haar.py::test_haar_ortho[1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-(5,)-()-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-()-(5,)-()-()-()-(4,)]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-4-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-4-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-()-()]
0.00s call     tests/distributions/test_ig.py::test_log_prob[1.7-2.5-4.0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-(5,)-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-2-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-4]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-3-0.37-0.51-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-()-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-sequential-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-parallel-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(4,)-()]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-Size-sample_shape2]
0.00s call     tests/distributions/test_unit.py::test_expand[batch_shape0-sample_shape1]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-4-0.37-0.77-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-3-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-()-(4,)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3--0.5]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-True-scaled]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-4-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-3-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-4-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-parallel-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-parallel-300]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(2, 3)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(5,)-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(5,)-2]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[3-()-(3, 2)]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-()]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-0.0]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-3-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-2-(2, 3)]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.39-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-sequential-None]
0.00s call     tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape0-event_shape0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-(5,)-()-()-()-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-(5,)-()-()-()-()-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(5,)-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-(5,)-()-(4,)]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-4-0.37-0.77-0.33]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional[3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-None-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-parallel-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-2-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-()-2]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-None]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-False-unscaled]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-True-scaled]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-True-unscaled]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional[6]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-1-1-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-()-(4,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-True-0-1000-False-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-3-2-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-4-sparse]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(2, 3)-3]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-()-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-1-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[3-(5,)-()-()-()-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-1-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-3]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-4-0.37-0.51-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-None-None]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-(2, 1)-(3,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-(5,)-()-()-()-(4,)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-True-scaled]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-3-0.37-0.51-0.33]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-False-0-1000-False-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-parallel-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(4,)-()]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[jit-exponential-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-unscaled]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-None-None]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-2.0-0.9-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-()-3]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-(5,)-()-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-parallel-300]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional_whiten[1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-2-3]
0.00s setup    tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-4-0.37-0.51-0.33]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-None-1-None-1-8]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-True-unscaled]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-()-(3,)]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[exponential-2-200]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[BetaBinomial]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(4,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-parallel-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(3, 2)-()]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(3, 2)-()]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[normal-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-(5,)-(4,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-4]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-sequential-parallel]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-True-unscaled]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-nonreparam-unscaled]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-sequential-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-3-2-7]
0.00s setup    tests/infer/test_util.py::test_psis_diagnostic[5-0.95-krange1-250000]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(4,)-()]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional_whiten[5]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Geometric-True-initial_shape1-proposed_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-3-2-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-None-parallel-None]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-14-1-7-1-1]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-2.0-0.5-0.5]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-parallel-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-parallel-None-None]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-3-7]
0.00s call     tests/distributions/test_ig.py::test_log_prob[0.5-3.0-2.5]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[studentt-2-100]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[laplace-1-100]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[exponential-2-100]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-condition-trivial_model]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-parallel-10000]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-2-1-1]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0-0.9]
0.00s setup    tests/contrib/gp/test_conditional.py::test_conditional_whiten[2]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-6]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-False-scaled]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-None-parallel-None]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-2.0--0.5-0.5]
0.00s call     tests/distributions/test_ig.py::test_log_prob[0.5-3.0-4.0]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-3-(2, 3)]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_logistic_regression[0.0855-None-4-False-True-False]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-7-3]
0.00s call     tests/distributions/test_ig.py::test_log_prob[0.5-2.5-4.0]
0.00s call     tests/distributions/test_categorical.py::TestCategorical::test_support_non_vectorized
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-()]
0.00s setup    tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-False-scaled]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-True-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-()-()]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-unscaled]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.39-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-parallel-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-3-2]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-7-1-7-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-7-3]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape3]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-1-1-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-sequential-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(2, 3)-2]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[map-ai,ai->i-i]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-None-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-parallel-2000]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-True-initial_shape1-proposed_shape1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-6]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-4]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-nonreparam-scaled]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-2-1-7]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-4]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-()-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-None-None-None]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-()-(3,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[svi-10-1-None-3-5]
0.00s call     tests/poutine/test_counterfactual.py::test_counterfactual_query[True-True-True]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-3-1]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Poisson_2D]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-scaled]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.55-0.44]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-2-1-7]
0.00s call     tests/distributions/test_ig.py::test_log_prob[1.7-3.0-2.5]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-(5,)-()-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-parallel-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(2, 3)-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-5]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_logistic_regression[None-1-None-True-False-False]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-()-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-2-1-7]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[2-()-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-3-2-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-parallel-parallel]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-0.5--0.5-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-2-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-3-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-3-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-Size-sample_shape2]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-parallel-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-4]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-scaled]
0.00s setup    tests/contrib/gp/test_models.py::test_gplvm[GPRegression_y2D]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-(2, 3)-(2, 3)]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_logistic_regression[None-1-None-True-True-False]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(4,)-(5,)]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-(2,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-4]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-2.0-0.0-1.5]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape4-event_shape4]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-2-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-Size-sample_shape0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-3]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-nonreparam-scaled]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[2-sequential-None-None]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-1-1-7]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.7-0.9-0.5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-False-unscaled]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.7-0.9-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(4,)-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_logsumexp[3-(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-None-False-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(3, 2)-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Geometric-True-initial_shape2-proposed_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-2-1-1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-5]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Binary_1D]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-0-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape1-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-()-(5,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape0-value_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-list-sample_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(4,)-(5,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-2-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-5]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[normal-2-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-6]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-2-1-7]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_logistic_regression[None-1-None-True-True-True]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.55-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(3, 2)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-3-2-1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape1-value_shape1]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype1-size0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-4]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-10]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[jit-bernoulli-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-2-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-3-2-1]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-1-()-(4,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-True-unscaled]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.7-0.5-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(4,)-(5,)]
0.00s call     tests/poutine/test_properties.py::test_commutes[trace-replay-trivial_model]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.7-0.9-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(3, 2)-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-3-1]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.7-0.0-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(4,)-()]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-6]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-True-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(4,)-(5,)]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[laplace-1-200]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[exponential-1-100]
0.00s call     tests/ops/test_contract.py::test_ubersum_total[i->-i]
0.00s call     tests/distributions/test_ig.py::test_log_prob[1.7-2.5-2.5]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-0.0]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-10]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-3-7]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-2.0-0.9-1.5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-2-()]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-7]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-2.0--0.5-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(3, 2)-(5,)]
0.00s setup    tests/infer/test_util.py::test_psis_diagnostic[5-0.5-krange0-250000]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(2, 3)-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(4,)-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-4]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-()-2]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[a]
0.00s call     tests/distributions/test_pickle.py::test_pickle[MixtureOfDiagNormals]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-True-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-()-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-4]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-2.0-0.0-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(4,)-()]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-1-200]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-nonreparam-unscaled]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-scaled]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(3, 2)-()]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Poisson_1D]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-2-1]
0.00s call     tests/poutine/test_properties.py::test_idempotent[trace-trivial_model]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(4,)-(5,)]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-3-7]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[studentt-2-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-unscaled]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-2.0-0.0-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(4,)-(5,)]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[laplace-2-200]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-tuple-sample_shape2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-5]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-True-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(3, 2)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-1]
0.00s setup    tests/infer/test_util.py::test_psis_diagnostic[5-0.5-krange0-500000]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.7-0.5-0.5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(2, 3)-2]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[laplace-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(3, 2)-()]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceGraph_ELBO-subsample]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(4,)-(5,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-4]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-1-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-()-()]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[exponential-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(5,)-2]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-()-()-(5,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-6]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.7-0.5-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(4,)-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-2-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(3, 2)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-()-()]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_MultiClass_1D]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-2.0]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-2.0-0.9-1.5]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-2.0-0.9-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Beta-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.7-0.9-0.5]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(3, 2)-()]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[studentt-1-1]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[laplace-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-tuple-sample_shape1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-()-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-1-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-2-3-MixtureOfDiagNormals]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-3-2]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-2.0--0.5-1.5]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-2-()]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-unscaled]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-2.0-0.5-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(3, 2)-(5,)]
0.00s setup    tests/infer/test_util.py::test_psis_diagnostic[1-0.95-krange1-250000]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-w,w->]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-list-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-7-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwj->ji]
0.00s setup    tests/infer/test_util.py::test_psis_diagnostic[1-0.95-krange1-500000]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.55-0.44]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-sequential-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-()-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-2-1-1]
0.00s call     tests/pyroapi/test_pyroapi.py::test_generate_data_plate[minipyro]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-6]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-2.0-0.5-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-()-(5,)]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-7-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(4,)-(5,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-4]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-5]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-4-3-MixtureOfDiagNormals]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-iwj->ij]
0.00s call     tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_shared_cov_log_prob[3]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[3-2-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(5,)-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-2-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(4,)-(5,)]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[studentt-2-200]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-tuple-sample_shape1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-1]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-2.0-0.5-0.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[warm-None-5-10-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-False-scaled]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-scaled]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.39-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-()-()]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-()-()]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[3-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-()-()]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[python-laplace-1]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-2-200]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-3]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-nonreparam-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-()-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-4]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-False-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(2, 3)-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-1-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-list-sample_shape2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-1-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-()-(5,)]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-()-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-2-1-7]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[normal-1-100]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Beta-True-initial_shape2-proposed_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-2-1-7]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-nonreparam-scaled]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(4,)-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-2-1-7]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[map-ab,bc->-]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(4,)-()]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Poisson_2D]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-0-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-3-2-1]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-(2,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(2, 3)-2]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-2-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-4]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-True-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(4,)-(5,)]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[python-bernoulli-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-2-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-4]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-3-2]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalSparseGP_y1D]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[studentt-1-200]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Beta-True-initial_shape1-proposed_shape1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-6]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-True-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(3, 2)-(5,)]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Binary_2D]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-1-()]
0.00s call     tests/distributions/test_pickle.py::test_pickle[StudentT]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(3, 2)-(5,)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-True-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(5,)-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-1-7]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 1)-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-1-1-7]
0.00s setup    tests/contrib/gp/test_models.py::test_inference[VariationalGP_y2D]
0.00s call     tests/distributions/test_pickle.py::test_pickle[MixtureOfDiagNormalsSharedCovariance]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-()-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-2-1-7]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-2.0--0.5-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-()-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-()-(5,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-1-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-2-1-1]
0.00s call     tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_log_prob[3]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-False-unscaled]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.7--0.5-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-3-2-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(3, 2)-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-1-1-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Dirichlet]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-False-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(5,)-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-1-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Gamma]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-4]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.7-0.0-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(4,)-()]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(4,)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-2-1-1]
0.00s call     tests/distributions/test_pickle.py::test_pickle[FisherSnedecor]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-Size-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-tuple-sample_shape2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-2]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-1-1-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-()-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_poisson[svi-None-1-None-1-8]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-1-1]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_logistic_regression[0.0855-None-4-False-False-False]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-17]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-1-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-3-()]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceTMC_ELBO-full]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(4,)-(5,)]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[1.0-1]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-3-2-(3, 2)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(2, 3)-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-2-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-list-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-Size-sample_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-3-2-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-None-parallel]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-3-2-7]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-2-100]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-list-sample_shape0]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-3-1]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-3-2-(3, 2)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-4]
0.00s setup    tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[plate-3-7000-0.0008]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.7-0.0-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-()-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-sequential-parallel]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(4,)-(5,)]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[normal-2-200]
0.00s call     tests/poutine/test_counterfactual.py::test_counterfactual_query[True-True-False]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-1-0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-4]
0.00s setup    tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[True-None-1-False]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-3-2]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-4-3-MixtureOfDiagNormals]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-list-sample_shape2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-()-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-1-1-7]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[1-()-()]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[LowRankMultivariateNormal]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(4,)-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-2-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-1-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-0-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-True-scaled]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(5,)-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-tuple-sample_shape0]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-None-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-2-2]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-True-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-list-sample_shape1]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Binary_2D]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ExtendedBetaBinomial]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(4,)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-2-1-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-2-0]
0.00s call     tests/contrib/gp/test_kernels.py::test_active_dims_overlap_ok
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-parallel-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(3, 2)-(5,)]
0.00s call     tests/infer/test_valid_models.py::test_enum_sequential_in_model_error
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-2-1-7]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-3-2-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-()-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-1-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-unscaled]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-2.0-0.0-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-()-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-1-1-7]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[python-studentt-2]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-1-1-()]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(5,)-2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Stable-list-sample_shape1]
0.00s setup    tests/contrib/gp/test_models.py::test_gplvm[GPRegression_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-2-1-1]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[exponential-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-4]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-parallel-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-1-1-1]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_model_with_subsample[10]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-sequential-parallel-None]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-4]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-3-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-1-1-1]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[exponential-1-200]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-2-sparse]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(3, 2)-(5,)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-nonreparam-unscaled]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-2-1-7]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[normal-2-100]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-1-1-1]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape0-event_shape0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-True-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-3-2]
0.00s setup    tests/contrib/gp/test_models.py::test_gplvm[SparseGPRegression_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-3-2-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape0-value_shape2]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-3-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-()-(5,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(3, 2)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-()-3]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[GPRegression_y1D]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges60]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-1-1-7]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-sequential-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-3-2-1]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[1-0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(3, 2)-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-1-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-0-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-5]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-True-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(3, 2)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-2-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-0-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-3-2-7]
0.00s call     tests/distributions/test_pickle.py::test_pickle[GaussianScaleMixture]
0.00s call     tests/distributions/test_pickle.py::test_pickle[BetaBinomial]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-3-uniform]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-3-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-2-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-3-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-6]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-False-scaled]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(4,)-()]
0.00s setup    tests/contrib/gp/test_models.py::test_gplvm[VariationalSparseGP_y1D]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Poisson_2D]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[1-()-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-3-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(4,)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-3-2-1]
0.00s call     tests/poutine/test_properties.py::test_idempotent[replay-trivial_model]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Weibull]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Poisson-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-2-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-3-3]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_MultiClass_2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-2-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-2-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-3-2-1]
0.00s call     tests/infer/test_valid_models.py::test_variable_clash_in_model_error[TraceEnum_ELBO]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-()-()-(5,)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(3, 2)-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-1-1-7]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[python-laplace-2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-4]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-nonreparam-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 2)-(6,)-9]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_MultiClass_2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-3-2-7]
0.00s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-4-3-GaussianScaleMixture]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-()-()-(5,)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-()]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-3-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-5]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-True-scaled]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-2-2]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_whiten[GPRegression_y1D]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Binary_2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-3-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-3-2-7]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-0.0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(3, 2)-(5,)]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalSparseGP_y2D]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Poisson_2D]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-2-7]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-2-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-3-2-1]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-1-100]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-1-1-7]
0.00s call     tests/contrib/test_minipyro.py::test_generate_data[minipyro]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-False-unscaled]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-True-unscaled]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-nonreparam-scaled]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Poisson-True-initial_shape1-proposed_shape1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-4]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-2-3-MixtureOfDiagNormals]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-3-2-7]
0.00s call     tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape0-event_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-1-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-4]
0.00s setup    tests/distributions/test_tensor_type.py::test_double_type
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(4,)-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-2-1-7]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_simple[warm-14-1-7-1-1]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-()]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(4,)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-1-1-()]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.7-0.5-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-4]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-2-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-1]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-3-2-(3, 2)]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.7--0.5-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalGP_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-1-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-()-(5,)]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-2-3-MixtureOfDiagNormals]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(2,)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(4,)-(5,)]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_MultiClass_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Cauchy]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(4,)-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-list-sample_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-3-2-7]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-()-()-(5,)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-3]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-()-(6,)-9]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-False-Size]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-6]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist2]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-3-3]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-3-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-1-1]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-tuple-sample_shape2]
0.00s setup    tests/contrib/gp/test_models.py::test_gplvm[VariationalSparseGP_y2D]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-()-()-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-6]
0.00s call     tests/distributions/test_reshape.py::test_reshape[2-5]
0.00s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-2-3-GaussianScaleMixture]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-()-()-(5,)-(4,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-3]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-True-True]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(4,)-()]
0.00s setup    tests/contrib/gp/test_models.py::test_gplvm[VariationalGP_y2D]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[2-1-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-2-1]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-()-()-(5,)-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-3-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-1]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-1-1-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(4,)-()]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_MultiClass_2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-1]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(3, 2)-()]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0--1.0-0.0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-6]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-parallel-parallel]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(4,)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-1-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-5]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-3-2]
0.00s setup    tests/contrib/gp/test_models.py::test_gplvm[SparseGPRegression_y2D]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_MultiClass_2D]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-2-3]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(2,)-()-()]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.39-0.44]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-3-3]
0.00s setup    tests/contrib/gp/test_models.py::test_forward[VariationalSparseGP_y2D]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_MultiClass_2D]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Poisson_1D]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_MultiClass_2D]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Poisson_2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-1-1-7]
0.00s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-4-3-GaussianScaleMixture]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LogNormal-list-sample_shape0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceEnum_ELBO-subsample]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-3-2-7]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-0-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-4]
0.00s setup    tests/infer/test_util.py::test_psis_diagnostic[1-0.5-krange0-500000]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-3-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-parallel-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-3-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[1-2-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-3-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-3-2-1]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-0-1]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-(2,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-2]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-1-2-1]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9-0.5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-list-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 1)-(6,)-9]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Poisson_1D]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-2-2]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-3-2-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LogNormal-Size-sample_shape0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-4]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_MultiClass_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-3-2-7]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-2-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-2]
0.00s setup    tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-True-18-expected5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(4,)-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-3-2-1]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[python-normal-2]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[1-3-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-5]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(3, 2)-()]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[jit-bernoulli-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-5]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[1-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-True-initial_shape2-proposed_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-3-2]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(3, 2)-()]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_MultiClass_2D]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-True-tuple]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-6-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-1]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-3-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(4,)-(5,)]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Binary_2D]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-Size-sample_shape1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-1-7]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[studentt-1-100]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-2-2]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[1-2]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-None-parallel]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5-0.9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-1-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-6]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-3-2-(3, 2)]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-2-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalSparseGP_y1D]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-()-()]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Binary_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-1-1]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[python-exponential-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-()-(5,)]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_local_param_ok[minipyro-py]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-centered3-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-2-1-1]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[python-studentt-1]
0.00s call     tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-2-3-GaussianScaleMixture]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-True-Size]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LogNormal-list-sample_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-3-2-1]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-None-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(3, 2)-()]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0--1.0-1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalSparseGP_y2D]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Binary_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-2-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-5]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-7-5-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-sequential-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(3, 2)-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[3-1-(3, 2)]
0.00s call     tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[TraceGraph_ELBO]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_constraints[pyro-jit]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalGP_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-1-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LogNormal-list-sample_shape2]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-()-()-()-()]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5-0.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape1-df_shape0]
0.00s setup    tests/contrib/gp/test_models.py::test_gplvm[VariationalGP_y1D]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-1-2]
0.00s setup    tests/contrib/gp/test_models.py::test_forward[VariationalGP_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(4,)-()]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Poisson_1D]
0.00s call     tests/distributions/test_reshape.py::test_reshape[2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-2-2]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-2]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-3-7]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-0.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape1-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-11]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-False-list]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(3, 2)-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-False-tuple]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Exponential]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.25]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-2]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Binary_1D]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-2-1-7]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[pyro-py]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-3-5-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-0-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-()-(5,)]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-()-()-9]
0.00s call     tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-3-7]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-unscaled]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-12]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-5]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-True-True]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-8]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-1-1-7]
0.00s call     tests/contrib/gp/test_kernels.py::test_active_dims_disjoint_ok
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(3, 2)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-3-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-3]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-True-True]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-()-(5,)]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_MultiClass_1D]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LogNormal-tuple-sample_shape0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-6]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[nonreparam-RenyiELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-0-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(3, 2)-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[2-1-(4,)]
0.00s call     tests/infer/test_valid_models.py::test_variable_clash_in_model_error[TraceTMC_ELBO]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Poisson-True-initial_shape2-proposed_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LogNormal-tuple-sample_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-3-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-1-1-7]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-12]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-10]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-1-1-7]
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-,bi,abi->b-i]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-2]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-10]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Binary_1D]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-3-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-1]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape1-df_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape3-sample_shape1]
0.00s setup    tests/contrib/gp/test_models.py::test_hmc[GPRegression_y1D]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-2-1-1]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-0.0]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(4,)-()]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-Size-sample_shape2]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Binary_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-1-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-3-3]
0.00s call     tests/distributions/test_pickle.py::test_pickle[ExtendedBetaBinomial]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-(2,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-sequential-parallel]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.25]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-()-()]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(2,)-(6,)-9]
0.00s setup    tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[GPRegression_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-1-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LogNormal-Size-sample_shape1]
0.00s setup    tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-True-200-expected3]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(3, 2)-()]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-10]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-centered3-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-3]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_MultiClass_2D]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-1-1]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-(2,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-2]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist2]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5--0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape2-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(3, 2)-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-2-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-1]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-16]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-1-1]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_serialization[python-exponential-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Binary_1D]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LogNormal-Size-sample_shape2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-()-(5,)]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.0-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-5]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape2-event_shape2]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-3-2-7]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-None]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape2-df_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LogNormal-tuple-sample_shape1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-14]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-2-1-7]
0.00s call     tests/poutine/test_counterfactual.py::test_counterfactual_query[False-True-False]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-()-()-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-1-1-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-sequential-None-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(3, 2)-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-True-list]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-14]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-Size-sample_shape2]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 1)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-1-1-7]
0.00s call     tests/distributions/test_reshape.py::test_reshape[2-4]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(4,)-()]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-list-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0--1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-()-()-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-0-7]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-()-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-5]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-()-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-3-2-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-3-1]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[pyro-jit]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-(5,)-()-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-()-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-3-2-7]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(3, 2)-(5,)]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-1.0-0.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-3-2-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_add[3-()]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[MultivariateStudentT]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-1-1]
0.00s setup    tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-True-280-expected4]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-14]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-()-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-()-(4, 5)-9]
0.00s setup    tests/contrib/gp/test_models.py::test_model[GPRegression_y1D]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Poisson_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-3-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-tuple-sample_shape1]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[minipyro-py]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-3-2-(4,)]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9--0.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-1-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-None-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.75]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-()-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-2-1-7]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-3]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Binary_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-3-2-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape3]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-10]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(2,)-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-2-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-Size-sample_shape1]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-1-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-1]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Poisson_2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-2]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-1.0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(4,)-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-1-1-1]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[minipyro-py]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-()-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-3-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-6]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_sequential_consistent
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(4,)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-2-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-13]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-1-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_add[2-()]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-1.0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(4,)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-2-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-19]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-2-7]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-3-1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-1.0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-()-()]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-4000]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-2-1-7]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_local_param_ok[pyro-jit]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-3]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-3-2-()]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-True-False]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(2,)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 2)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-1-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-7-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-3-7]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.75]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-12]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[2-3-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-2-1-7]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-False-False]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-()-()]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-()-()-9]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(4,)-(5,)]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-2-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-0-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-3-()-()]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-None]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-3-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-3]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-parallel-sequential-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-6]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-2]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.55-0.33]
0.00s setup    tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[plate-None-7000-0.0008]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 2)-(6,)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0--1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-3-2-7]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_plate_ok[minipyro-jit]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-4]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape2-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-()-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-2-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Normal-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-()-()]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-Size-sample_shape0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-1-1-1]
0.00s call     tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-None]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-3]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-subsample-nonreparam]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[1-None-parallel-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-True-True]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 1)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(2,)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(2,)-(4, 5)-9]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape1-value_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-2-1-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-1-uniform]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-()-(4,)]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-1.0-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-6]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape0-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 2)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-2-1-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape4]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Normal-True-initial_shape1-proposed_shape1]
0.00s setup    tests/infer/test_gradient.py::test_plate[TraceEnum_ELBO-reparam]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape0-df_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(2,)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(2,)-()-9]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Poisson_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-3-2-7]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-w->]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1-1.0]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-centered3-(3, 2)]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.6-(3, 2)]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-False-True]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-1-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-2-1-7]
0.00s call     tests/distributions/test_delta.py::TestDelta::test_batch_log_prob
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-3]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-True-Size]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-2-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-3]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-None-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape3-event_shape3]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape2-event_shape2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-2-7]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-()-()-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1--1.0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-1]
0.00s setup    tests/contrib/gp/test_models.py::test_forward[VariationalSparseGP_y1D]
0.00s setup    tests/contrib/gp/test_models.py::test_forward[SparseGPRegression_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-3-2-7]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_plate_ok[pyro-jit]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(4,)-(7,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.25]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-1-2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(2,)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-()-(4, 5)-9]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Binary_2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-2-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-list-sample_shape0]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.55-0.33]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-False-True]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.75]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-()-()]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(2,)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 1)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-()-(4, 5)-9]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-4000]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-tuple-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-7]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-()-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-2-1-1]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_constraints[minipyro-py]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[pyro-jit]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-6]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.5]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape3-event_shape3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-2-1-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[3-1-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-3-2-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-1-2000]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-False-True]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9--0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape2-df_shape2]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape2-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 2)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-()-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-1-1-7]
0.00s call     tests/ops/test_gaussian.py::test_rsample_shape[1-()-(4,)]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_constraints[minipyro-jit]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape0-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-12]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 2)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-1-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(3, 2)-()]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0-1.0-1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_forward[GPRegression_y2D]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_MultiClass_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-3-2-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-3]
0.00s setup    tests/infer/test_jit.py::test_svi_enum[1-sequential-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-parallel-parallel]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-True-False]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-16]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-1-1-1]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_local_param_ok[minipyro-jit]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-1.0-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-5]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 2)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 1)-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0--1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-1-1-7]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-None-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-17]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-2-7]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[minipyro-jit]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-2-1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-()-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_reshape.py::test_reshape[2-3]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-0.5]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.6-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-1]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_MultiClass_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-3-2-1]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_delta_smoke[None]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_plate_ok[minipyro-py]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape0-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-1-3]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-()-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 2)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(2,)-(6,)-9]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Poisson_2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-1-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-Size-sample_shape0]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_local_param_ok[pyro-py]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-()-()]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(2,)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-()-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-2-1-1]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_plate_ok[pyro-py]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-5]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-True-False]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape1-df_shape2]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape1-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-1-2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-()-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 2)-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-3-2-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape0-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 2)-(4, 5)-9]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-4000]
0.00s setup    tests/contrib/gp/test_models.py::test_forward[GPRegression_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-2-7]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-20-5-plate]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Normal-True-initial_shape2-proposed_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 1, 1)-()]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape0-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-()-()]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 2)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(2,)-(4, 5)-9]
0.00s setup    tests/contrib/gp/test_models.py::test_hmc[VariationalGP_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-2-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-True-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-3]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-6]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.25]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape0-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-2-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-True-initial_shape1-proposed_shape1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-7]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-2]
0.00s setup    tests/infer/reparam/test_haar.py::test_normal[(4, 2)--2-True]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.25]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-5]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 1)-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0--1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-2-1-7]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-1-2]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.6-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-parallel-parallel]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.75]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-2]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape4-event_shape4]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-10]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0--1.0--1.0]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Binary_2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-2-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-sequential-sequential-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.5]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-2.0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-1-2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 2)-(6,)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0-1.0--1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_model[VariationalGP_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-2-1-7]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.0-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-3]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-True-False]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-5]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-10]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-2]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(4,)-(5,)]
0.00s setup    tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalGP_y2D]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Poisson_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-3-2-1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-1-2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-3-2-7]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[minipyro-jit]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-13]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-2-1-7]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-1-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_8[plate-plate-1-1]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-False-True]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape2-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-16]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape4-event_shape4]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 1)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-1-1-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-3]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.75]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-()-()]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 1)-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-3-2-7]
0.00s call     tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_shared_cov_log_prob[1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-3-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-3]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.25]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-18]
0.00s setup    tests/contrib/gp/test_models.py::test_inference[SparseGPRegression_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-2-1-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[1-2-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-True-Size]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-6]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-True-True]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-()-(4, 5)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0-1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-1-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-()-(5,)-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-2-1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 2)-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-2-1-7]
0.00s call     tests/distributions/test_reshape.py::test_reshape[2-1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-6]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-None-parallel]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-1.0]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-15]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-()-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-()-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-3-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-7-7]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[0-0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-4]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.25]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.75]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-1-2]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-10]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 2)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 2)-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-1-1-1]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 3, 2)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[1-parallel-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-1-2000]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-True-True]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0-0.5]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 1)-()-9]
0.00s setup    tests/contrib/tracking/test_em.py::test_em_nested_in_svi[True]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0--1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-2-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_add[1-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-3-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-3]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-3-2-()]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-False-False]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.25]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape0-df_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape1-value_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 1)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-6]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-0-2000]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-None-None-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9--0.9]
0.00s setup    tests/distributions/test_mvn.py::test_shape[3-cov_shape1-loc_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 2)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-3-2-1]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-centered3-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-3]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-2.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 2)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 1)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-6]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.25]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5--0.9]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape1-value_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-1-3]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-()-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(2,)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(2,)-(4, 5)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0--1.0-0.0]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Poisson_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-2-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-19]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-10]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(2,)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-2-1-7]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-None-plate]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-()-(5,)-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-1]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-parallel-True-3]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.25]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.25]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-()-()]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-True-tuple]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-5]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-3-1-1]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.39-0.33]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.75]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9--0.9]
0.00s setup    tests/distributions/test_rejector.py::test_exponential_elbo[1.0-2.0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-16]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-3]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Binary_2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-True-list]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-(5,)-()-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-1.0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 2)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(2,)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 2)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-3-2-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(3, 2)-()]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-2-1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-()-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0--1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-1-1-7]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-False-sample_shapes0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-5]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-()-()-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-3]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-18]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 1)-(4, 5)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0--1.0--1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_hmc[VariationalSparseGP_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-2-1-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-,w->]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-1-1]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-None-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-4]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-18]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-()-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-1-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-()-(7,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-4]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(2,)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 1)-(6,)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0-1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-2-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-10]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-2]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-3-2-(4,)]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-9]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(2,)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-2-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-sequential-parallel]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(4,)-()]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 2)-(6,)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0--1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-1-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-()-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-3-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-parallel-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.25]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-15]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-2]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-3]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 1)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(2,)-(4, 5)-9]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-10]
0.00s setup    tests/contrib/gp/test_models.py::test_model[VariationalGP_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-1-1-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-1]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-()-(7,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-17]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-()-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-()-(6,)-9]
0.00s setup    tests/contrib/gp/test_models.py::test_model[VariationalSparseGP_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-1-1-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-4]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0--0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape0-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-18]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 1)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 2)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-()-(4, 5)-9]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-ssmgp-3-1.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-2-1-7]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(3, 2)-GreaterThan(lower_bound=0.0)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-3-2-1]
0.00s setup    tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-14]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 1)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(2,)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-()-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-()-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(2,)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 1)-(4, 5)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0--1.0-1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_model[VariationalSparseGP_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-3-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-7-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-3-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-3]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-7-2-1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-14]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-2]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 2)-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0--1.0--1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_inference[GPRegression_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-2-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-()-()-()-(4,)]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.6-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-7]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 2)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 1)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-1-1-1]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-1.0-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-6]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-1-2-1]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.55-0.44]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-6]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 1)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 2)-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-3-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-0-7]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Multinomial]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 3, 2)-(7,)]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.0-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-2]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.55-0.33]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5--0.5]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 1)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-()-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 2)-(4, 5)-9]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-glgssm-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-3-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-2-1-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-parallel-300]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0-0.9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 2)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(2,)-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0-1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-2-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-3-1]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-Size-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-2-1-7]
0.00s call     tests/distributions/test_pickle.py::test_pickle[RelaxedBernoulli]
0.00s call     tests/distributions/test_pickle.py::test_pickle[InverseGamma]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-6]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-6]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 1)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-()-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(2,)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 1)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 1)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-2-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-1]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape0-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-13]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(2,)-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0-1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-2-1-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-1]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(4,)-()]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-False-True]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.75]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-19]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-3]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 1)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-()-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-3-2-1]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(3, 2)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-5]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-1.0-(3, 2)]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.0-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-11]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-10]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-sequential-parallel]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-3]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(2,)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(2,)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-3-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-0-3]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-()-(5,)-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(4,)-(4,)]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-centered3-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-1]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-10]
0.00s setup    tests/contrib/gp/test_models.py::test_forward[SparseGPRegression_y1D]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-1-1]
0.00s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_nonreparameterized_N_is_3
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-4]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[nonreparam-ReweightedWakeSleep]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-()-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 2)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 1)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 2)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(2,)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 2)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(2,)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-2-1-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[1-3-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-6]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 2)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(2,)-(4, 5)-9]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-2-0.5]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_MultiClass_1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-1-1-7]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(3, 2)-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-2-3-(3, 2)]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.75]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5--0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape2-df_shape0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape1-df_shape1]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-()-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-15]
0.00s setup    tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[SparseGPRegression_y1D]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-1-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-6]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-3-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0-0.5]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape1-value_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-16]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 2)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(2,)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-()-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 1)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 2)-(6,)-9]
0.00s setup    tests/contrib/oed/test_linear_models_eig.py::test_nmc_eig_linear_model
0.00s setup    tests/contrib/gp/test_models.py::test_hmc[GPRegression_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-2-1-7]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[MultivariateNormal]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-1-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-()-(7,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-3]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5--0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape2-df_shape2]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape2-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-1-2]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 1)-(6,)-9]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-100]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-4000]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0-1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-3-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-0-7]
0.00s call     tests/ops/test_contract.py::test_einsum_linear[->-]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[0-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-4]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(3, 2)-(7,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-3]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-None-False-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9--0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape1-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-6]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-1-1-7]
0.00s call     tests/poutine/test_poutines.py::test_trace_log_prob_err_msg
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-()-(5,)-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-0-3-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-6]
0.00s setup    tests/infer/reparam/test_haar.py::test_normal[(2, 5)--1-True]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-1.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-None-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5-0.5]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-16]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-()-(5,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 1)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-()-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-2-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-7-3]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1239]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-1-(3, 2)-(3, 2)]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-None-(3, 2)]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0-0.9]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-13]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 2)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-1-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-(5,)-()-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-3]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-17]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-True-initial_shape2-proposed_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0-1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-1-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-()-(5,)-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-1]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5-0.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape1-df_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-()-()-9]
0.00s setup    tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Binary_2D]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-1]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-centered3-(4,)]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.0-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-2]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5-0.5]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-13]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape4-event_shape4]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-()-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 1)-(4, 5)-9]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(4,)-(7,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-None-None-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape1-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-14]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-()-(4, 5)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-0.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-4]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-None-(4,)]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-3-5-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-1-2000]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-None-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-False-True]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-3-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-3-2-1]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Stable]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-2-random]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(3, 2)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-6]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-0-2000]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9--0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape1-df_shape1]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape2-df_shape2]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape2-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-4000]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-2-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-7-3]
0.00s call     tests/distributions/test_conjugate.py::test_log_prob_support[dist0-None]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-2]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-0-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9--0.9]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-10]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-2-1]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-()-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-3]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.39-0.33]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.75]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 1)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(2,)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 2)-()-9]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-4000]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-3-2-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-1-(3, 2)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-2]
0.00s setup    tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-sequential-1-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-parallel-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5--0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape2-df_shape2]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape1-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-19]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 2)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-3-2-7]
0.00s call     tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-20-plate]
0.00s setup    tests/ops/test_indexing.py::test_value[()-()-(3, 1)-()]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-3-2]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-3-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-3]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0-0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape1-df_shape0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape1-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-17]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Categorical-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Uniform-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 1)-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-1-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-0-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-3-2-1]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(4,)-(4,)]
0.00s setup    tests/nn/test_autoregressive.py::AutoRegressiveNNTests::test_jacobians
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9-0.9]
0.00s setup    tests/distributions/test_mvn.py::test_shape[1-cov_shape0-loc_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-3-2-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[2-2-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-True-initial_shape0-proposed_shape0]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Uniform]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-3-2-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-6]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-1-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-1-2000]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(2,)-(6,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-1-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-100]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-2]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-2]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-2-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-6]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-()-(5,)-()-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-sequential-sequential-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape1-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-()-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-6]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.55-0.33]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5--0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape1-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-15]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-10]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{}-0-8-6]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-5]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-1-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-2-()-(4,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-1-2]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.6-()]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.6-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-3]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.55-0.33]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9-0.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape0-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-Size-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-1-1-1]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape0-df_shape0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape2-df_shape0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape2-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-8]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-4000]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[1-parallel-sequential-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9-0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape1-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-11]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-()-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(2,)-(6,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 1)-()-9]
0.00s setup    tests/contrib/gp/test_models.py::test_inference[GPRegression_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-0-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-1-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-(5,)-()-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-3]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5--0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape2-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-1]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[BetaBinomial]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-full-reparam]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0--0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape0-df_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-17]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 1)-(4, 5)-9]
0.00s setup    tests/contrib/gp/test_models.py::test_model[GPRegression_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-2-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-()-(6, 5)]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-None-parallel]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-9]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 1)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-1-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-2-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-1]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-1.0-(3, 2)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-1-1-(3, 2)]
0.00s setup    tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[False-True-50-expected1]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.39-0.44]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-parallel-None-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-13]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 1)-(4, 5)-9]
0.00s setup    tests/contrib/gp/test_models.py::test_hmc[SparseGPRegression_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-2-1-1]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(3, 1)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0--0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape0-df_shape1]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape1-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-tuple-sample_shape0]
0.00s setup    tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalSparseGP_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-6]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-3-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-4]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-3-1-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_8[iplate-plate-1-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-parallel-parallel]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-sequential-parallel-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-parallel-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.75]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0-0.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape2-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-Size-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0--1.0--1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_forward[VariationalGP_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-1-1-1]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-5]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[3-2-()]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_constraints[pyro-py]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-4]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape2-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-12]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-Size-sample_shape0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-3-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[HalfCauchy]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-1-1]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9-0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape1-df_shape2]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape0-df_shape2]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape0-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-11]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 2)-(6,)-9]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-10]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-()-()-(5,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-()-(6, 5)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-1-1]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.55-0.44]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-sequential-parallel-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9-0.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape0-df_shape0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape2-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-10]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-3-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0-1.0-1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_hmc[VariationalSparseGP_y1D]
0.00s setup    tests/contrib/gp/test_models.py::test_inference[SparseGPRegression_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-4]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-17]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-list-sample_shape0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-1-1-1]
0.00s call     tests/poutine/test_nesting.py::test_nested_reset
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Gamma-True-initial_shape1-proposed_shape1]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-sequential-None-None]
0.00s setup    tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[iplate-8-100-0.018]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9--0.5]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[0.2-1.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape0-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-6]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-Size-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0-1.0--1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0--1.0-1.0]
0.00s call     tests/distributions/test_categorical.py::test_batch_log_dims[dim=1-probs0]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-(5,)-()-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-()-()-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-parallel-parallel]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-None-parallel-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-True-False]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9-0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape0-df_shape1]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape2-df_shape1]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape1-df_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape0-value_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-15]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-1]
0.00s setup    tests/contrib/gp/test_models.py::test_hmc[SparseGPRegression_y1D]
0.00s setup    tests/contrib/gp/test_models.py::test_model[SparseGPRegression_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-6]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[None-True-True-True]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-parallel-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-parallel-300]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5-0.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape1-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-()-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0-1.0--1.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-7-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-4]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5-0.9]
0.00s setup    tests/distributions/test_mvn.py::test_shape[1-cov_shape1-loc_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0-1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-1-1-1]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-3-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-sequential-sequential-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.75]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5--0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape2-df_shape0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape2-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-12]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-list-sample_shape0]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-4000]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-7-3]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(3, 2)-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-5]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-()-(5,)-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_expand[1-(5,)-()-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-3-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-4]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-None-None-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape2-df_shape0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape1-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(3, 2)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-19]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-()-()-(5,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-0-2000]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-sequential-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-None-sequential-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-False-False]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9-0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape0-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-15]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-4000]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-3-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-False-Size]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-1-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-10]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-()-(5,)-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-()-(5,)-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-2]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist2]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-parallel-None-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-9]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[3-3-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-5]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-1-5-1]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.55-0.33]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-False-False]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.75]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape2-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-17]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-list-sample_shape2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-6]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-()-(4,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-sequential-sequential-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-False-False]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.25]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9--0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape2-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-11]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-list-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0-1.0--1.0]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-3-lgssm]
0.00s setup    tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalSparseGP_y2D]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-True-Size]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-4]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-parallel-True-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-parallel-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-True-False]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5--0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape1-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Cauchy-tuple-sample_shape0]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-4000]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-3-1]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Binomial]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-3]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.75]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9--0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape1-df_shape2]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape0-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0--1.0--1.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-False-tuple]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-3-2-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(2,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-3-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-0-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-0-2000]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9-0.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape0-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-tuple-sample_shape1]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-()-()-(5,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-(5,)-()-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(3, 2)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-None-parallel-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape2-df_shape0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape1-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-4]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(2,)-(6,)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0-1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-2-1-1]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-7-1-2]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.39-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-sequential-parallel-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(4,)-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(2,)-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-2-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-2-()-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-parallel-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-sequential-parallel-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-False-True]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-14]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0--1.0-1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_whiten[SparseGPRegression_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-1-1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape1-value_shape1]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(3, 2)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s call     tests/contrib/autoname/test_scoping.py::test_mutual_recur
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-3-(4,)-(4,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-sequential-sequential-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-True-False]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-False-False]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.0]
0.00s setup    tests/distributions/test_tensor_type.py::test_conflicting_types
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5-0.9]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape0-df_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-13]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Bernoulli-True-initial_shape2-proposed_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-7-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-3-()-()]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(3, 2)-Real()]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-True-tuple]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[1-1-()]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-2-3-(4,)]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5-0.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape0-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-3]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-10]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0-1.0-1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-1.0-0.0]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-1-0.5]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-True-list]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-6]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape0-df_shape0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape1-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-2-7]
0.00s call     tests/distributions/test_categorical.py::test_batch_log_dims[dim=3-probs0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-1-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-4]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(2,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-()-(7,)]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[0.5-False-True-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-sequential-sequential-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-5]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-5-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-3-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0-1.0-1.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-2-7]
0.00s setup    tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-4]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-parallel-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-parallel-300]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5-0.9]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape0-value_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-14]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape2-event_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0--1.0--1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_model[SparseGPRegression_y2D]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-0-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-3-2-1]
0.00s setup    tests/ops/test_gaussian.py::test_expand[2-(5,)-()-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-3-(3, 2)-(4,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-1]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.39-0.44]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.55-0.33]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5-0.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape1-df_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-6]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-3-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-10]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-6]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-1-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-None-sequential-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-parallel-sequential-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-False-False]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape2-df_shape1]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape0-df_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(4,)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-14]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-Size-sample_shape2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape4-event_shape4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-2-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-1-(3, 2)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-2-True]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-0-2000]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-None-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-parallel-None-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-10]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape4-event_shape4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-list-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0--1.0-1.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-3-6]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-()-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-()-(7,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-1]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceTMC_ELBO-subsample]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-parallel-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-parallel-10000]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5--0.5]
0.00s setup    tests/distributions/test_rejector.py::test_exponential_elbo[1.0-1.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-12]
0.00s call     tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-5]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-2-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-4]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-parallel-parallel-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-True-True]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.75]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-9]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-3-lgssmgp]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-3-7]
0.00s call     tests/pyroapi/test_pyroapi.py::test_generate_data[pyro]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-3-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-3]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-2-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_cat[1-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-3-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-6]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[None-True-True-False]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-1-1-2]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-7-5-1]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-3-2-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-sequential-parallel-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-False-False]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-13]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-Size-sample_shape0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-2-7]
0.00s call     tests/distributions/test_pickle.py::test_pickle[RelaxedBernoulliStraightThrough]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-10]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-None-sequential-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-True-False]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-13]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_expand[3-(5,)-()-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-3-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-parallel-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-4]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape3-event_shape3]
0.00s setup    tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-2-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-2-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-1-2000]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-parallel-sequential-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-tuple-sample_shape1]
0.00s setup    tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalGP_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-3-2-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-6]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-0.5]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-parallel-None-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape2-df_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-list-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0--1.0-1.0]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-2-(3, 2)-()]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-sequential-False-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.25]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9-0.0]
0.00s setup    tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape0-df_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-4]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape3-event_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-Size-sample_shape0]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape2-value_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-True-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Exponential]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-2-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(3, 2)-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-19]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(4,)]
0.00s setup    tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-False-150-expected2]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.55-0.44]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.75]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-12]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape4-event_shape4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-list-sample_shape2]
0.00s call     tests/distributions/test_pickle.py::test_pickle[OneHotCategorical]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9--0.5]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape2-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-10]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape4-event_shape4]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-2-clamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-3-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-3]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-3-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-None-None-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-False-True]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5--0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-list-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0-1.0-1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_whiten[GPRegression_y2D]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-6]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-2-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-1]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.39-0.33]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9-0.0]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-4]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape3-event_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-tuple-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-0-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-True-initial_shape2-proposed_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-4]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-3-()-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(3, 2)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-None-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-13]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-Size-sample_shape1]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-glgssm-1-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-2-1-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[1-2-()]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-6]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-parallel-True-None]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-sequential-False-None]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-3-2-2]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_restrictions[False-sequential-1-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-sequential-None-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-19]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-1-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-7]
0.00s setup    tests/infer/reparam/test_haar.py::test_normal[(4, 2)--2-False]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-7-5-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-None-sequential-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9-0.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-1-1-7]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-3-1-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-parallel-sequential-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-sequential-None-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-True-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-False-list]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-1]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.39-0.33]
0.00s setup    tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[range-3-100-0.018]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-True-True]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9-0.9]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Poisson-False-initial_shape0-proposed_shape0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-2-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-7-7]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ExtendedBetaBinomial]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-10]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_cat[2-(4, 7, 6)--1-(2, 1, 3)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-6]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5--0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Dirichlet-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-10]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0--1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-3-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-10]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 3, 2)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_8[plate-iplate-1-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-None-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-9]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-7-7]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-()-(6, 5)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-1-1-(4,)]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[map-10]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-9]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-Size-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0-1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-3-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-0-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5--0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-8]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0--1.0-0.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-3-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-()-(7,)]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_feasible-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-10]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Gamma-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-10]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-1-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-0-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-6]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-4]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-None-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5-0.9]
0.00s setup    tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-tuple-sample_shape0]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-2-lgssm]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalGP_y1D]
0.00s setup    tests/contrib/gp/test_models.py::test_inference[VariationalSparseGP_y2D]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-0-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-6]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.55-0.44]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-0-2000]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-16]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape3-event_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-list-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-20-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0-1.0--1.0]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-glgssm-1-3]
0.00s setup    tests/contrib/gp/test_models.py::test_hmc[VariationalGP_y2D]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-3]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.55-0.44]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-1-None]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-19]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape4-event_shape4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Bernoulli-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-list-sample_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-4]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-parallel-False-None]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-sequential-True-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-parallel-300]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_beta[0.5-1.0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape2-value_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-1]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape3-event_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-list-sample_shape1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-5-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-2-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-3-(3, 2)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-1]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-parallel-True-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-0-2000]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-18]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-3-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0--1.0-1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[SparseGPRegression_y1D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-3-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-3-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-8]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceTMC_ELBO-full]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-tuple-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-0-3]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-6]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-Trace_ELBO-full]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.39-0.44]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-parallel-10000]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Stable-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-Size-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0-1.0--1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0-1.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-0-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-4-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(4,)-(3, 2)]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-3-1-1]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.39-0.44]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-7]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-5-unclamped]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[SparseGPRegression_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-2-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-7-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-7-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-1-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-2-(4,)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-4]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-0-2000]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9-0.0]
0.00s setup    tests/distributions/test_mvn.py::test_shape[1-cov_shape0-loc_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-19]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape2-event_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar_full_mass': 4}-7-8-6]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-3-()-()]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape1-value_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-5]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-2-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(4,)-(6, 5)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-3]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.39-0.44]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-0-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-1-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-1.0]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0-0.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-16]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Categorical-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-10]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0-1.0--1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-0.0-0.0]
0.00s setup    tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[SparseGPRegression_y2D]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-4-10]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-None-sequential-None]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-list-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0-1.0--1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_whiten[SparseGPRegression_y2D]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(3, 2)-Real()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-1-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-2-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-3-(3, 2)-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-None-None-None]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-12]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape4-event_shape4]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape3-event_shape3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0--1.0--1.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-0-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(3, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-1-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-3-()-()]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-1-1-(4,)]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-2-2]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-parallel-False-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[1-parallel-None-None]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Dirichlet-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-20-clamped]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape0-value_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-10-5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-1-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_cat[2-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0-0.0]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-2.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-8]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Bernoulli-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-Size-sample_shape1]
0.00s setup    tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-2-NcvContinuous]
0.00s setup    tests/contrib/gp/test_models.py::test_inference[VariationalSparseGP_y1D]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-2-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-4]
0.00s setup    tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9-0.9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-19]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0--1.0--1.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-8]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape0-value_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[5-(6,)-(4, 5)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0--1.0--1.0]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalGP_y1D]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-7-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(4,)-(6, 5)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-5]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-7-1-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0-1.0-1.0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-2-1-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-3-(4,)-(4,)]
0.00s setup    tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape2-True]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-1-None]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-False-initial_shape1-proposed_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-3-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-7]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-2-()-(4,)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-2-3-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceTMC_ELBO-full]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-1-2000]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.0--0.5-0.5]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape2-value_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-6]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-tuple-sample_shape2]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-3-lgssm]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-3-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-1-1-7]
0.00s call     tests/contrib/test_minipyro.py::test_local_param_ok[minipyro-jit]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-4]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-sequential-None]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.25]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape1-value_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-7]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-Size-sample_shape2]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-3-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-0-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-1-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-6-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-3-(4,)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-1-1-()]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[0.1-False-False-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-1-None]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-tuple-sample_shape0]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-3-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-0-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(3, 1)-(7,)]
0.00s setup    tests/infer/reparam/test_haar.py::test_uniform[(2, 5)--1-True]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-sequential-True-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-1-2000]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Normal-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Beta-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-tuple-sample_shape2]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-100]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-1-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(4,)-()]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-1-1-(4,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-parallel-parallel-None]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-18]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-0-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-0-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-6]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(4,)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-3]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-1.0]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Geometric-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-20-unclamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-3-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-2]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[2-3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(4,)-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-0-None]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Uniform-True-initial_shape2-proposed_shape2]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-True-sample_shapes0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-4]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-()-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-2-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-()-()]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[GPRegression_y2D]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-True-sample_shapes1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-10]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-3-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-0-None]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-0.5-0.0-0.5]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-12]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-Size-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-3-2-1]
0.00s setup    tests/params/test_module.py::test_module_nn[outest]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-10-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-()-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[2-None-sequential]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Cauchy-True-initial_shape0-proposed_shape0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-8]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-3-5-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-Size-sample_shape0]
0.00s setup    tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[GPRegression_y2D]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-2-7]
0.00s call     tests/distributions/test_reshape.py::test_reshape[2-0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-()-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-1-(4,)-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-()-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-3-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-4]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-0.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-1-None]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape1-value_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-tuple-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-1.0--1.0]
0.00s setup    tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-2-(3, 2)-(4,)]
0.00s setup    tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape1-True]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[sequential-loss-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-list-sample_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-2-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-5]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-0-3-(3, 2)]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-list-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[2-(2, 3)-(7,)-9]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-ssmgp-3-1.5]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape1-value_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-()-(7,)]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-tuple-sample_shape1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-1-1-1]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-False-sample_shapes1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-10]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-6]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[2-sequential-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[2-None-None-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-parallel-None]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-Size-sample_shape1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-1-()-()]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-1-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-10]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-3-2-(4,)]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[None-True-False-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-1-2000]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-tuple-sample_shape0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-6]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(4,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-1-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[2-parallel-sequential]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-2.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-3-1]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(4,)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-6]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-5-unclamped]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-3]
0.00s call     tests/contrib/test_minipyro.py::test_local_param_ok[minipyro-py]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(3, 2)-()]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-3]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceEnum_ELBO-full]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape2-event_shape1]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape3-event_shape3]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-20-clamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-1-1-7]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-20-iplate]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-6]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(4,)-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-1-2]
0.00s setup    tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[plate-8-7000-0.0008]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-10]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-20-unclamped]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-7-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-8]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-4]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-2.0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[11-(2, 3)-()-9]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 2}-7-4]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-10-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-6-5]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(4,)-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_stable[LatentStableReparam-(4,)]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[(4,)-(2, 3, 1)-[1, 2]--2]
0.00s setup    tests/infer/test_abstract_infer.py::test_nesting
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape1-value_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-list-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-5-clamped]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-0-3]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_model_error_enum_dim_clash[HMC-kwargs0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(3, 1)-(3, 1)-()]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-list-sample_shape2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-2-1]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-3-()-(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceGraph_ELBO-full]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-0.0--1.0]
0.00s call     tests/ops/test_indexing.py::test_value[()-()-()-()]
0.00s setup    tests/optim/test_multi.py::test_optimizers[<lambda>0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-1-()-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-3-1]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-list-sample_shape1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-2-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[3-1-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-1-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-4]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-1-2-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_logsumexp[3-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-()-(4,)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8--0.5]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape1-value_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Poisson-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-list-sample_shape1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape2-value_shape1]
0.00s setup    tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-6]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(4,)-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99--0.5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(4,)-(5,)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-1.0-1.0]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-1-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-6]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(4,)-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7-1.0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-10-clamped]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-0-7]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-3-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-()-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-1-3-(4,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_size_error[naive_ubersum-ab,bc->-shapes1-b]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-1-1-(4,)]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_restrictions[False-None-2-True]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.0-0.9-1.5]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape0-event_shape1]
0.00s setup    tests/distributions/test_empirical.py::test_sample_examples[sample0-weights0-expected_mean0-expected_var0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-tuple-sample_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-6]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-3-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[2-2-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-4]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-3-2-()]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-Size-sample_shape1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-0-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-1-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-6]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-1-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-3-(3, 2)]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-tuple-sample_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-7]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[()-(6,)-[2, 1, 3]--1]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-2-3-(4,)]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-False-NUTS-_empty_model]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Beta-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-Size-sample_shape1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-6]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-()-(3, 2)]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-7-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-4]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-()-(6, 5)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-1-2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape2-value_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-5-unclamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-5]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(2,)-(3, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[1-(4,)-(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceTMC_ELBO-subsample]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape2-value_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-5-unclamped]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s call     tests/poutine/test_poutines.py::test_trace_log_prob_sum_err_msg
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-0-3-()]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-3-5-2]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[0.5-1.0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape1-value_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape2-value_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[VonMises-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-list-sample_shape0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape1-value_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-5]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-1-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-0-3-()]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[5-()-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-3]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape0-value_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-10-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-()-(4, 3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-2-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-list-sample_shape1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-3-2-1]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-()-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(2,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_cat[1-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[marginal-a,abi->-i]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-0.5--0.5-1.5]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape2-value_shape0]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape2-event_shape0]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape0-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-list-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-7-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-3]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-()-(6, 5)]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_beta[0.5-0.5]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape2-value_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-20-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-1-1]
0.00s call     tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-TraceEnum_ELBO]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape1-value_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-4-6]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-3-1-2]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-0.5-0.9-1.5]
0.00s setup    tests/distributions/test_mvn.py::test_shape[3-cov_shape0-loc_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-5-unclamped]
0.00s call     tests/distributions/test_categorical.py::test_batch_log_dims[dim=2-probs0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-()-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-1-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[1-2-(4,)]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape2-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape1-value_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-list-sample_shape0]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-3-1.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-7-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-1-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-3-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-8]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-()-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(3, 2)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-5]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-10-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0--1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-1-1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape2-value_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-1-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-2]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-2]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-3-5-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-1]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[5-()-(7,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-0-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-2-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-4]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-()-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(3, 1)-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(3, 2)]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Gamma-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-list-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-5-unclamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-3-1]
0.00s call     tests/contrib/autoname/test_named.py::test_named_object
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-3-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-3-()-(4,)]
0.00s setup    tests/infer/test_jit.py::test_svi[JitTraceMeanField_ELBO-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[2-()-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-2-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[2-2-()-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_condition[1-2-()-()]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-2]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-()-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-1-1-(3, 2)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Categorical-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-list-sample_shape1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-7-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-()-()]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-3-2-1]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-1-2-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-full-nonreparam]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_conjugate_update.py::test_beta_binomial[()-()]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-0.0-0.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-1-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape2-value_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-5]
0.00s setup    tests/ops/test_special.py::test_log_beta_stirling[0.02]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-()-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(2,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_cat[2-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_cat[1-(4, 7, 6)--1-(2, 1, 3)]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Gamma-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-5-clamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-5]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[1-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-3-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-1]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-7-2-2]
0.00s setup    tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[Trace_ELBO-reparam]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-tuple-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-0.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-()-(2,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[2-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_cat[2-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-2-(3, 2)-(4,)]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-3-2-1]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.17-0.33]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[10.0--1.5707963267948966]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-Size-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0--1.0--1.0]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-2-lgssm]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-7-7]
0.00s call     tests/distributions/test_delta.py::test_shapes[False-3-3]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-20-5-plate]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-6]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-2-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-0-2-()]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-1]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-14]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Uniform-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-5-unclamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-3-7]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(3, 2)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-1-1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(3, 2)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-1-1-(4,)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-4]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-7-2-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-None-None]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-10-clamped]
0.00s setup    tests/contrib/oed/test_linear_models_eig.py::test_laplace_linear_model
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-6-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(3, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-1]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Cauchy-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-2-clamped]
0.00s setup    tests/contrib/oed/test_linear_models_eig.py::test_posterior_linear_model
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-5-4]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(3, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-Trace_ELBO-subsample]
0.00s setup    tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-None-2-True]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-5-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-()-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(2,)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-1-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_logsumexp[3-(3, 2)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-tuple-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-1.0--1.0]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-glgssm-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-7]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-()-(7,)]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-Trace_ELBO-subsample]
0.00s setup    tests/distributions/test_mvn.py::test_shape[3-cov_shape1-loc_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-10-unclamped]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-1-1.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-0-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape2-value_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-8]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(2,)-(2,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-4]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-5.0]
0.00s setup    tests/distributions/test_mvn.py::test_shape[5-cov_shape1-loc_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape0-value_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-20-clamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s call     tests/infer/mcmc/test_valid_models.py::test_model_error_enum_dim_clash[NUTS-kwargs1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-10-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-()-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-()-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-()-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-3-()-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3-1.0]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[(3, 2)-(4, 2)-[1, 3]--2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-4]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-0.5-0.5-0.5]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape2-value_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape0-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape0-value_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-2-clamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-1-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-4-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-2-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(2,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-()-(7,)]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-1.0]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-1-1-1]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LogNormal-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Exponential-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0--1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-1-1]
0.00s call     tests/distributions/test_spanning_tree.py::test_make_complete_graph[python-4-expected_grid2]
0.00s call     tests/distributions/test_one_hot_categorical.py::test_support_dims[dim=1-probs0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-6]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-2-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-3-()-()]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[()-(4, 2)-[1, 3]--2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-1]
0.00s setup    tests/infer/reparam/test_haar.py::test_normal[(2, 3, 1)--2-False]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-1-1-2]
0.00s setup    tests/distributions/test_mvn.py::test_shape[1-cov_shape1-loc_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape2-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Cauchy-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-Size-sample_shape1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-1-1]
0.00s call     tests/distributions/test_reshape.py::test_independent_entropy
0.00s call     tests/contrib/autoname/test_scoping.py::test_basic_scope
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-6]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-6]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape1-value_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape0-value_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-2-unclamped]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-4000]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{}-7-8-6]
0.00s call     tests/contrib/autoname/test_named.py::test_named_dict
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,a,ab,b,b,b,b->a-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-7]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(3, 2)]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-0.5]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-1-1-2]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-1-5-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[1-None-sequential-None]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.3-1.5707963267948966]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape1-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape0-value_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-2-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-0.0-0.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[1-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-3-()-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-3]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape1-value_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-tuple-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0--1.0-0.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-3-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-5-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-5]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[(4,)-(2, 5)-[2, 3]--1]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-7-5-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-parallel-parallel]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape2-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-Size-sample_shape1]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-3-2.5]
0.00s setup    tests/contrib/oed/test_linear_models_eig.py::test_lfire_linear_model
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-0-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-3-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-100]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-5-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-3]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-3-2-()]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-1-1-()]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-Trace_ELBO-subsample]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-7-2-1]
0.00s setup    tests/distributions/test_stable.py::test_variance[1.7-0.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-4-sparse]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-5.0]
0.00s setup    tests/distributions/test_mvn.py::test_shape[3-cov_shape0-loc_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape1-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape2-value_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-tuple-sample_shape0]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Exponential]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-3-1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape1-value_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-0-3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-3-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-3]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-1-uniform]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape0-value_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-()-(5,)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-1.0-0.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-5-10]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_cat[2-(4, 7, 6)--1-(2, 1, 3)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-4]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceGraph_ELBO-subsample]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Categorical-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[11-()-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-1-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(3, 2)-()]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-1-2-2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape0-value_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Normal-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-list-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[2-(6,)-(4, 5)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0--1.0-0.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-7-7]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(5,)-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-5-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-3-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-4-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-5]
0.00s setup    tests/ops/test_special.py::test_log_beta_stirling[0.05]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 1, 1)-()]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape1-event_shape1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-7-7]
0.00s call     tests/distributions/test_delta.py::test_shapes[True-3-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-4-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-5-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-1-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-5]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape0-value_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-list-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-10-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-0.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-7-3]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ac-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-6-10]
0.00s setup    tests/ops/test_special.py::test_log_beta_stirling[0.0001]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-1-(3, 2)-(3, 2)]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-7-1-1]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-3-1-2]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-7-1-1]
0.00s setup    tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceEnum_ELBO-reparam]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_sample-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_rejector.py::test_exponential_elbo[0.5-1.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LogNormal-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[3-(2, 3)-(7,)-9]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-1-1.5]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Periodic]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-8]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-3-2-(4,)]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape2-value_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(4,)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[3-(2, 3)-()-9]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-10]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0--1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'num_quant_bins': 2}-7-8-6]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-7-3]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[2-1-()-()]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape0-value_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-2]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(3, 2)-(6, 5)]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape2-value_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape1-value_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Poisson-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-3]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-5-clamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-7-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-1.0]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Gamma]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(3, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-()-()-()]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[:,:]-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(4,)]
0.00s setup    tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape0-True]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-()-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-0-3]
0.00s call     tests/contrib/autoname/test_named.py::test_named_list
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-3-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-2-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-1-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-0-1-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-5]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-0.5]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceGraph_ELBO-full]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceEnum_ELBO-full]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape1-value_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-list-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-()-()-9]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-100]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Constant]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(3, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(2,)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(()))[...]-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-3-(3, 2)-()]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-1-5-2]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-7-1-2]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_median-Trace_ELBO]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[5.0-2.0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape2-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape0-value_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Stable-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-list-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-2-unclamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-3-1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape1-value_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-4-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 1, 1)-(7,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-6]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[2-sequential-None]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.01-0.0]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-0.5-0.9-0.5]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Bernoulli-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Dirichlet-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-list-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-()-(5,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-2-7]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[3-1-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-10-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-6-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-7]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-3-()-(3, 2)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-5]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-7-1-1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape2-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape1-value_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-list-sample_shape2]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(3, 2)-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-0.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar_full_mass': 4}-0-8-6]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar': True}-0-8-6]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-0-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-2-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceGraph_ELBO-subsample]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-0.5]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[1.0-5.0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape0-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape2-value_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Beta-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-2-clamped]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-3-1.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-0-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-5]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-0-1-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-4]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-7-5-2]
0.00s setup    tests/distributions/test_stable.py::test_variance[2.0-1.5]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.0-0.9-0.5]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape0-value_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Bernoulli-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-10-unclamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-0-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-0-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-3-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-3-()-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-2-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(4,)-()]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[(3, 2)-(2, 3, 1)-[1, 2]--2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-6]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-1-1-(3, 2)]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-1-1-1]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-3-5-1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape2-value_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[11-(6,)-(7,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[3-(6,)-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-0.0-1.0]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-3-lgssmgp]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-7-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-3-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-6]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(4,)-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0--1.0]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-1-1-(3, 2)]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-2-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-1.0-0.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-7-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-8]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-1-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[3-2-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-3-()-(4,)]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[()-(2, 3, 1)-[1, 2]--2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-4]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-0.0]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-10-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-0.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-2-1-7]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-20-5-iplate]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,b,bc->ac-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-()-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(3, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-2-()-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-2]
0.00s setup    tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[iplate-None-100-0.013]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[2.0-5.0]
0.00s setup    tests/distributions/test_rejector.py::test_exponential_elbo[1.0-0.5]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-list-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-20-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-0-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape1-value_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-10-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-5-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-7]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-1-3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-2]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceTMC_ELBO-subsample]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.0--0.5-1.5]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape1-value_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Categorical-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(3, 2)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[5-()-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[3-()-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-3-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-w->]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-2-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-2-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-2-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-0-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[(4,)-(4, 2)-[1, 3]--2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-2]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-4]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-list-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-5-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-7-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_enumerate_support[3]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_mean_field_ok[minipyro]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-6]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-3-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-2-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-1]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-1-1-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceTMC_ELBO-subsample]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-0.5-0.5-1.5]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-2.0]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-5.0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape1-value_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape1-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape0-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Exponential-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-20-clamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-1-7]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape4-event_shape4]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-3-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-3-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-10-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-1-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-3]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-3]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-7-5-2]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.0-0.5-0.5]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-0.5]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-1.0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape0-value_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-1.0-0.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0--1.0-1.0]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Cosine]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'num_quant_bins': 2}-0-8-6]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape2-value_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-6-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-6-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-7]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[2-3-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-0-3-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-4]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-1-1-1]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-list-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-2-unclamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-7-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-5-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-2-(3, 2)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-6]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-0.5]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-20-unclamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-6-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(2,)-(3, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-3-(4,)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-3]
0.00s setup    tests/infer/reparam/test_haar.py::test_normal[(6,)--1-True]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceGraph_ELBO-full]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-0.5--0.5-0.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_make_complete_graph[cpp-4-expected_grid2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0--1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-7-3]
0.00s call     tests/distributions/test_mask.py::test_mask_noop[()]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-2-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_stable[StableReparam-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-1]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[1-1-5-2]
0.00s setup    tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[Trace_ELBO-nonreparam]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape0-value_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-list-sample_shape1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-5-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-2-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-8]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(3, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_cat[1-(4, 7, 6)--1-(2, 1, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_cat[3-(4, 7, 6)--1-(2, 1, 3)]
0.00s setup    tests/infer/reparam/test_stable.py::test_stable[StableReparam-(2, 3)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-5]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-2.0]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-5]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[2-sequential-sequential-None]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape2-value_shape2]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape1-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Geometric-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Cauchy-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(4,)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-0-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape2-value_shape1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape1-value_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-10-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-()-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-()-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-()-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-1-1-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-4]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape2-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[2-(6,)-(7,)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-0.0-0.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-0.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-0-3]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(3, 2)-Real()]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->b-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(3, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(2,)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-()-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(3, 2)-(6, 5)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-2]
0.00s setup    tests/infer/reparam/test_haar.py::test_normal[(2, 3, 1)--2-True]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-3-2-(4,)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-1-1-(3, 2)]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-1.0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-5-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-10-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-2-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-2-3-()]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[0.5-5.0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[2-()-(7,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-0-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-0-3]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(4,)-GreaterThan(lower_bound=0.0)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-1-1-7]
0.00s setup    tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-4]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-()-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-()-(6, 5)]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-0.5]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-1-1-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceGraph_ELBO-subsample]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape2-value_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Poisson-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-list-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(3, 2)-(5,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-0-3]
0.00s setup    tests/contrib/autoguide/test_mean_field_entropy.py::test_guide_entropy[mean_field_guide-args0-expected_entropy0]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Stable-True-initial_shape1-proposed_shape1]
0.00s setup    tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(3, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-2-3-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-()-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-3-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-3]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-1-1-(4,)]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-7-2-1]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-1-2-2]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-parallel-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-None-None]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-list-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[3-()-(7,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-0-3]
0.00s call     tests/infer/test_valid_models.py::test_plate_broadcast_error[Trace_ELBO]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape2-value_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-3-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-8]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(3, 1)-(2,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(4,)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-6]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-1-2-2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape0-value_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-2]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-()-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(3, 2)-(5,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[5-(2, 3)-(7,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-6]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-1-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_cat[3-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-2]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[1.0-1.5707963267948966]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[2-(6,)-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-3]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-ssmgp-2-2.5]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-7-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-1-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(3, 2)-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-4]
0.00s setup    tests/distributions/test_mvt.py::test_log_prob[2-(4,)]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-list-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-20-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-0.0--1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0--1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-0-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(3, 1)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-1-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-1-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-2-2-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-5]
0.00s setup    tests/infer/test_gradient.py::test_plate[Trace_ELBO-nonreparam]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[1-sequential-sequential-None]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape0-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[2-(2, 3)-(4, 5)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-1-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-3-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(2,)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-1-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-3]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-Trace_ELBO]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-3-2-2]
0.00s setup    tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-None-2-False]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-0-0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape0-value_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape2-value_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LogNormal-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-20-clamped]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-dmgp-1-1.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape4-event_shape4]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-3-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-6-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-6-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-3-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-6]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-5.0]
0.00s setup    tests/distributions/test_mvt.py::test_log_prob[1-(4,)]
0.00s setup    tests/distributions/test_mvn.py::test_shape[5-cov_shape0-loc_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape2-value_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-2-unclamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-7-3]
0.00s setup    tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_logsumexp[3-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-2-3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-3-(4,)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-6]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-7-2-2]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-0.5-0.0-1.5]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-0.5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(3, 2)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(4,)-(5,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-2-1]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-None]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-7-3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-10-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-5]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape5]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-1-1-(4,)]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-0.2]
0.00s setup    tests/distributions/test_mvn.py::test_shape[1-cov_shape2-loc_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-Size-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0--1.0-0.0]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-2-lgssm]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-2-lgssmgp]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-True-initial_shape0-proposed_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-3-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-6]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-None-batch_shape4]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-()-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-()-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-2-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-1-()-()]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[(3, 2)-(6,)-[2, 1, 3]--1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-4]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-Trace_ELBO-subsample]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-2]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-4]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape1-value_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-20-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0--1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape2-value_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-2-7]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-1-1-()]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-Trace_ELBO-subsample]
0.00s setup    tests/infer/test_gradient.py::test_plate[TraceGraph_ELBO-nonreparam]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-0-2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape0-value_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape1-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-20-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0--1.0--1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0--1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-1-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-1-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_stable[StableReparam-()]
0.00s setup    tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape1-False]
0.00s setup    tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-parallel-1-False]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape1-value_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[11-()-(7,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[5-(6,)-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-7-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-0-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-7-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-6-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-2-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[1-3-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-0-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(4,)-()]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-2-3-(3, 2)]
0.00s setup    tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[range-None-100-0.011]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape2-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape0-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-20-clamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar': True}-7-8-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 1, 1)-(7,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-3]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape0-value_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-20-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-0.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-10-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-8]
0.00s setup    tests/ops/test_indexing.py::test_value[()-()-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-1-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-0-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-2-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-1-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[()-(2, 5)-[2, 3]--1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape2-value_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape0-value_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape1-value_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(3, 2)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-7-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape1-value_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-3-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_logsumexp[2-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-3-(4,)-()]
0.00s setup    tests/ops/test_arrowhead.py::test_utilities[0]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-1-1-(4,)]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist0]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[30.0--1.5707963267948966]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape0-value_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-10-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-0.0-1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0--1.0-0.0]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-2-lgssmgp]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-3-7]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-None-iplate]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(3, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-()-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[3-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-1-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-6]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[2-1-5-2]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape0-value_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[11-(6,)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[3-(6,)-(7,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-7-3]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape1-value_shape0]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape0-value_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-2-1-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-5]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-3]
0.00s setup    tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[range-8-100-0.01]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.0-0.5-1.5]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape2-value_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape1-value_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[3-(6,)-(4, 5)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[3-()-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-7-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[1-1-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-0-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-3-(3, 2)-(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-Trace_ELBO-full]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[1.0--1.5707963267948966]
0.00s setup    tests/distributions/test_empirical.py::test_sample_examples[sample1-weights1-expected_mean1-expected_var1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-10-unclamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-0-3]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(3, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-3-()-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-3-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-1-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-3-(4,)]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.7--0.5-0.5]
0.00s setup    tests/distributions/test_mvt.py::test_rsample[1-3.9]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape1-value_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape1-value_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-tuple-sample_shape0]
0.00s setup    tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-3-NcvContinuous]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-0.0-0.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-7-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-3]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-1-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-1-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-2-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01-0.5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-4]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape2-value_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-13]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-list-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-5-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(3, 2)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(3, 2)-(5,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-7-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-7-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Exponential-True-initial_shape2-proposed_shape2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->b-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-6]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(2,)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[2-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-3]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.0-0.0-0.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-5-sparse]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Gamma-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(4,)-(5,)-9]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-3-lgssmgp]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-0-3]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-10]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-()-(3, 1)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-1-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-1]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceGraph_ELBO-full]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-2]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-5]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-subsample-reparam]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-5-clamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-2-1]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[3-1-()]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-5]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-None-batch_shape0]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(2,)-()-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-2-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-1-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-5]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-3]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-Trace_ELBO-subsample]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-1-sparse]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape0-value_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape0-value_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[VonMises-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[11-()-()-9]
0.00s setup    tests/contrib/oed/test_linear_models_eig.py::test_marginal_linear_model
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Exponential-True-initial_shape0-proposed_shape0]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-5-5-iplate]
0.00s setup    tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-6]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(3, 2)-(6, 5)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-3]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-9]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Geometric-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-list-sample_shape1]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-()-(5,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[5-(2, 3)-()-9]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-dmgp-3-1.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-0-7]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Pareto]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-5-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-8]
0.00s setup    tests/ops/test_special.py::test_log_beta_stirling[0.01]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-0-1-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_cat[1-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-6]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_restrictions[True-None-1-False]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[30.0-1.5707963267948966]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-8]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LogNormal-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[5-(6,)-(7,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s call     tests/distributions/test_one_hot_categorical.py::test_support_dims[dim=3-probs0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->c-1]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-()-(4, 3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-3-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[3-1-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-2]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[parallel-True-None]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceGraph_ELBO-full]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceTMC_ELBO-full]
0.00s setup    tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[iplate-3-7000-0.0008]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-0-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[11-(6,)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-50]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-()-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-3-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(4,)-(6, 5)]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-1]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape1-event_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-list-sample_shape0]
0.00s setup    tests/distributions/test_conjugate.py::test_mean[dist2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[2-(2, 3)-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0--1.0--1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-8]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(3, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[3-1-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-0-2-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-0-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(3, 2)-(6, 5)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-5]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[10.0-1.5707963267948966]
0.00s setup    tests/distributions/test_stable.py::test_mean[S0-1.0-0.0-1.5]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[2.0-2.0]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Binomial-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(4,)-(5,)-9]
0.00s setup    tests/contrib/oed/test_linear_models_eig.py::test_marginal_likelihood_linear_model
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-0-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape2-value_shape2]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-3-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-2-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-2-()]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[(3, 2)-(2, 5)-[2, 3]--1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-3]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[2-()-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-()-(3, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-2-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-2]
0.00s setup    tests/infer/reparam/test_haar.py::test_normal[(6,)--1-False]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[rsvi-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape1-value_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Beta-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[5-(2, 3)-(4, 5)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-3-3]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-ssmgp-2-2.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-0-7]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-2-2]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[1-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-1]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-3-2-()]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceGraph_ELBO-subsample]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-2-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(4,)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(4,)-()-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-3-3]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-7-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape1-value_shape0]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape2-value_shape1]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-5-iplate]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(4,)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-1]
0.00s setup    tests/distributions/test_rejector.py::test_exponential_elbo[0.5-0.5]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Multinomial-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(4,)-()-9]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-10]
0.00s setup    tests/contrib/oed/test_linear_models_eig.py::test_vnmc_linear_model
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-1-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 1, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-3-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-3-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-3-(3, 2)-(3, 2)]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-3-random]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Binomial-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-()-(5,)-9]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Matern32]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_mean_field_ok[pyro]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 3, 2)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-()-(4, 1, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-()-(2,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(3, 2)-()]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-2-3-(4,)]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-18]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-2-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(3, 2)-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-()-(5,)-9]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(3, 2)-(5,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-3-7]
0.00s setup    tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_log_prob[True-relu]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-4-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-8]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(3, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-()-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-1-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-2-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_cat[3-(4, 7, 6)--1-(2, 1, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-0-3-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8-0.5]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-1-1-()]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_util.py::test_psis_diagnostic[1-0.5-krange0-250000]
0.00s setup    tests/infer/test_discrete.py::test_plate_smoke[2-map]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-0.5--0.5-1.5]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-0.2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-10-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[3-(2, 3)-(4, 5)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-0-3]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-()-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-()-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-()-(2,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-3-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[3-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-2-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-2-2-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-5]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.03-1.5707963267948966]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[5.0-5.0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-tuple-sample_shape1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-8]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_cat[1-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-3-(3, 2)-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_stable[LatentStableReparam-(2, 3)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-5]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.01-1.5707963267948966]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Multinomial-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[11-(2, 3)-(4, 5)-9]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-lcmgp-3-1.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-1-1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape1-value_shape2]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape1-value_shape1]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(3, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[2-1-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[2-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-3-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-1-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_cat[3-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-6]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-Trace_ELBO-full]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-0.5-0.5-0.5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Dirichlet-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-list-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(3, 2)-(5,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-7-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-6]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(2,)-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-2-()-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[2-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-0-1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-3-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-1-()-(4,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_size_error[naive_ubersum-ab,bc->-shapes0-]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-1-1-(4,)]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[VonMises-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-5-clamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-()-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-6]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-2-3-()]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceEnum_ELBO-full]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-Size-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-2-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0--1.0-0.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-3-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-2-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-3-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-2-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-1-(4,)-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-6]
0.00s setup    tests/infer/reparam/test_haar.py::test_normal[(2, 5)--1-False]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-2-clamped]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-lcmgp-3-1.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-2-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape0-value_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Stable-True-initial_shape0-proposed_shape0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->ac-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(3, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-2-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-0-3-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-2]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[reparam-RenyiELBO]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Stable-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-2-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-1-2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-1-7]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->c-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-8]
0.00s setup    tests/ops/test_special.py::test_log_beta_stirling[1e-06]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(3, 2)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-2-3-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceEnum_ELBO-full]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-()-()-9]
0.00s setup    tests/distributions/test_coalescent.py::test_simple_smoke[11-(2, 3)-(7,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(3, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-1-()-(4,)]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-1.0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceTMC_ELBO-full]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-5.0]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[0.5-2.0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0--1.0-0.0]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-3-lgssm]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape1-value_shape1]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-None-batch_shape3]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-()-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-()-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 3, 2)-()]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-1.0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Normal-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-list-sample_shape2]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-10]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-10-clamped]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-1-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.5]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(4,)-GreaterThan(lower_bound=0.0)]
0.00s call     tests/contrib/oed/test_ewma.py::test_ewma_log_with_s
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[full-()-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-()-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-()-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-3-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[3-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-2-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-2-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(3, 2)-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01-1.0]
0.00s setup    tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-2]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceTMC_ELBO-full]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-3-uniform]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-0.2]
0.00s setup    tests/distributions/test_mvn.py::test_shape[1-cov_shape1-loc_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-6]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-Size-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-1-1]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-3-2.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-3-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-7-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape1-value_shape0]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-2-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-1-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-1-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-2-1-(3, 2)]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-1.0]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-1.0]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[nonreparam-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_mvn.py::test_shape[3-cov_shape1-loc_shape2]
0.00s setup    tests/distributions/test_mvn.py::test_shape[1-cov_shape2-loc_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Uniform-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-2-clamped]
0.00s setup    tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-3-NcpContinuous]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[WhiteNoise1]
0.00s call     tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_log_prob[False-relu]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-()-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(3, 1)-(2,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-()-(7,)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceTMC_ELBO-subsample]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist1]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[1-None-None-None]
0.00s setup    tests/distributions/test_mvn.py::test_shape[3-cov_shape2-loc_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-5-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(4,)-()-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-0-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-0-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-30]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(2,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-1-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-3-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-3-()-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-5]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.44-0.41]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-0.2]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-10-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-0.0-0.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s setup    tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_rsample[True-relu]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Exponential-True-initial_shape1-proposed_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-()-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-1-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[2-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[1-1-()]
0.00s setup    tests/infer/test_enum.py::test_svi_step_smoke[parallel-single]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-0.5]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Bernoulli-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-1.0--1.0]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-dmgp-2-1.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-()-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[2-3-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-1-()-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[1-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-2-1-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-5]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_mean-Trace_ELBO]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.3-0.0]
0.00s setup    tests/distributions/test_rejector.py::test_exponential_elbo[0.25-1.0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-11]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-tuple-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.1]
0.00s call     tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_log_prob[1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-()-(2,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-2-()-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3--1.0]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-2-3-(4,)]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist3]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[0.2-2.0]
0.00s setup    tests/distributions/test_mask.py::test_mask[1-1-1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-Size-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-2-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(2,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-()-()]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-Trace_ELBO-full]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-4-uniform]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-list-sample_shape1]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[WhiteNoise0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-2-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-0-2-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-1]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-False-HMC-_empty_model]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-True-NUTS-_empty_model]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-8]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape4-event_shape4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-20-unclamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-()-()-9]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-4]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-()-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(3, 1)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(3, 1)-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-3-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_cat[3-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-4]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-2.0]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-1-1-(4,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-13]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-list-sample_shape1]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-3]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-10-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(3, 2)-(5,)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-1.0-0.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape2-value_shape0]
0.00s call     tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-1]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(3, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-1-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_logsumexp[1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-2-(3, 2)-(3, 2)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-2-3-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.17-0.41]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-4-random]
0.00s setup    tests/distributions/test_mvn.py::test_shape[5-cov_shape2-loc_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-Size-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-3]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-lcmgp-2-2.5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-8]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-()-()]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(2,)-(2,)-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(2,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-1-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-2-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(4,)-(4,)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-1-1-(3, 2)]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceTMC_ELBO-full]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceTMC_ELBO-subsample]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-tuple-sample_shape2]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-3-lgssmgp]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-2-2-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[1-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-3-(3, 2)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-1-1-()]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-2-3-()]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-sequential-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[sequential-differentiable_loss-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-20-unclamped]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.0001]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-5-plate]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-8]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-3-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-1-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-3]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-5]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-Trace_ELBO-full]
0.00s setup    tests/distributions/test_mvn.py::test_shape[5-cov_shape2-loc_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-10-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-1.0--1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0--1.0-1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-0.0--1.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape1-value_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Stable-True-initial_shape2-proposed_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-8]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-()-()]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(2,)-(2,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-1-3-()]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-2-3-()]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-2-3]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-2-unclamped]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape2-value_shape0]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[3-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(4,)-(6, 5)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-6]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-2]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-1]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-1-1-(3, 2)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-full-reparam]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[2-None-parallel]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-4000]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-1-0.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-1-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape1-value_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-8]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-1-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-2-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(4,)-(6, 5)]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-1.0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-tuple-sample_shape2]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-2-lgssmgp]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-0-3]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->ac-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_reshape[1-(6,)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-3-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(4,)-(4,)]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[100.0--1.5707963267948966]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-2-random]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Poisson-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-tuple-sample_shape1]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-dmgp-2-1.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-1-1]
0.00s call     tests/ops/test_welford.py::test_welford_dense[1-1]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(4,)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-5-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-5]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-2-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-1-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-1-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-2-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-1-3-()]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceEnum_ELBO-full]
0.00s setup    tests/infer/test_gradient.py::test_plate[TraceGraph_ELBO-reparam]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Geometric-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Exponential-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-list-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-0-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-5]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-1-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-2-()-()]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-2-3-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[None-differentiable_loss-1]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-1.0]
0.00s setup    tests/distributions/test_mvn.py::test_shape[5-cov_shape1-loc_shape0]
0.00s setup    tests/distributions/test_mask.py::test_mask[0-0-0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Geometric-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Exponential-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-10]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-1-2.5]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Brownian]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape0-value_shape1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->b-2]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(2,)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-2-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(3, 2)-()]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_restrictions[False-None-2-False]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[100.0-1.5707963267948966]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-1-sparse]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-tuple-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-1-3]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-2-lgssmgp]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-3-lgssm]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-3]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoDiagonalNormal]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-6]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-1-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-2-(4,)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-subsample-nonreparam]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[10.0-0.0]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[1.0-0.0]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.03-0.0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-list-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-5-clamped]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-lcmgp-2-2.5]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-3-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-2-()]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-1-1-()]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceEnum_ELBO-full]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-Trace_ELBO-full]
0.00s setup    tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-2]
0.00s setup    tests/infer/test_discrete.py::test_distribution_masked[sample]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.3--1.5707963267948966]
0.00s setup    tests/distributions/test_rejector.py::test_exponential_elbo[0.25-2.0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-17]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-20-clamped]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-10]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape0-value_shape1]
0.00s call     tests/distributions/test_conjugate.py::test_log_prob_support[dist1-None]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_generate_data_plate[minipyro]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-8]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_logsumexp[2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-1-()-()]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-0.5]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-Trace_ELBO-subsample]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[1-sequential-parallel-None]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-0.5-0.9-0.5]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-10-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0--1.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-2-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape1-value_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-False-Size]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-2-1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-1-2-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceEnum_ELBO-full]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-4-uniform]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-6]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Stable-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-Size-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-7-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.1]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-None-iplate]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-7]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-1-(3, 2)]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-1-1-(3, 2)]
0.00s setup    tests/infer/test_gradient.py::test_plate[DiffTrace_ELBO-reparam]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-5-sparse]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-5-random]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-2-sparse]
0.00s setup    tests/distributions/test_mvn.py::test_shape[5-cov_shape0-loc_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-Size-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0--1.0-1.0]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[RBF]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-5]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-()-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-0-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-0-2-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-1]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.0-0.0-0.5]
0.00s setup    tests/distributions/test_mvn.py::test_shape[5-cov_shape1-loc_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-4000]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-2-0.5]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-0-7]
0.00s call     tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-TraceTMC_ELBO]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-4]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-0-1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-()-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-0-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-0-1-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-5]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-1.0]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-parallel-1]
0.00s setup    tests/infer/test_smcfilter.py::test_smoke[10-1-5-1]
0.00s setup    tests/infer/test_jit.py::test_dirichlet_bernoulli[JitTraceEnum_ELBO-True]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[5.0-1.0]
0.00s setup    tests/distributions/test_mvn.py::test_shape[3-cov_shape2-loc_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-tuple-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-3-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-3-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-7-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-8]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-1-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-1-(4,)-()]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_restrictions[False-parallel-1-False]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-1-uniform]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-18]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Geometric-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-20-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-1.0-1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0--1.0-1.0]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-1-2.5]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Linear]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-1-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-4-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-8]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-8]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[3-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-0-2-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-4]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-2-3-(4,)]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-5-uniform]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[2.0-1.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[NaiveBeta]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-4]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-0.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-1-1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,ab->ab-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-8]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-1-()-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[1-(4,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_split.py::test_normal[(4,)-(6,)-[2, 1, 3]--1]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-4]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-sequential-1]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceTMC_ELBO-full]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[rsvi-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.0--0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-0.5-0.0-0.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-1-random]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-1-random]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-6]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Cauchy-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-list-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-5-unclamped]
0.00s setup    tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-2-lgssm]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-3-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-8]
0.00s setup    tests/ops/test_gaussian.py::test_cat[2-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-1-(4,)]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-tuple-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-1.0-0.0]
0.00s call     tests/infer/test_valid_models.py::test_plate_stack_sizes[sizes2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[same-()-4-4]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 1, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-2-3-()]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-4-sparse]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-3-random]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-2-uniform]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-10-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-1.0-1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0--1.0--1.0]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-7-3]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape0-value_shape0]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(4,)-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-()-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-1-1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-2-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-Trace_ELBO-full]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[reparam-Trace_ELBO]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape2-event_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-5-unclamped]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-dmgp-3-1.5]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-7-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-7-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-5]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[2-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[1-()-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-1-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-2-3-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-3]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-2-sparse]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Exponential-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-list-sample_shape1]
0.00s call     tests/distributions/test_reshape.py::test_reshape_reshape[0-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-7]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-1-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-1-2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-1-1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-1-2-(4,)]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.5]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.1--1.5707963267948966]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-5-sparse]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-1-random]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-False-sample_shapes0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-0.0-0.0]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-dmgp-1-1.5]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-0-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-50]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(2,)-(2,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[3-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-1-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-2-2-()]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[100.0-0.0]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-3-uniform]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-4-sparse]
0.00s setup    tests/distributions/test_mvn.py::test_shape[5-cov_shape0-loc_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Gamma-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-20-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-1-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-1.0-0.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-3-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape2-value_shape1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape0-value_shape2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->b-1]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-1-()-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[2-(3, 2)-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-2]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[sequential-True-None]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.17-0.41]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[3.0-1.5707963267948966]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-19]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-Size-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-1-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-1-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-1-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.05]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(3, 1)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-1-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-2-(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceGraph_ELBO-subsample]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.0-0.5-1.5]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[0.2-5.0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Dirichlet-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Normal-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Categorical-Size-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-5-unclamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-2-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-3-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-3-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0--1.0-0.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-0.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-7-7]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-4]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-3-(3, 2)-(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[2-sequential-parallel-None]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-5-random]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-4-uniform]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-5-random]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-0.2]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[1.0-2.0]
0.00s setup    tests/distributions/test_rejector.py::test_exponential_elbo[0.5-2.0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape4]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape4-event_shape4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Multinomial-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-True-initial_shape0-proposed_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-1-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-0.0-0.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0--1.0-1.0]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[1-3-()]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-,ab->ab-1]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[1-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-1-2-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-()-(7,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[1-sequential-None-None]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_sample-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.0--0.5-1.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-2-uniform]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-2.0]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape2-event_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Uniform-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-tuple-sample_shape0]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-100]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-2-7]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-3-7]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->d-1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->ac-2]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_median-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-0.5-0.5-1.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-3-sparse]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Categorical-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LogNormal-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Exponential-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-list-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-2-2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-20]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[2-1-()]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-3]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[2-2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[1-1-()]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[3-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-()-()]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-1.5]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-2-3-()]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.1-0.0]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.0-0.9-0.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-1-sparse]
0.00s setup    tests/distributions/test_mvn.py::test_shape[5-cov_shape2-loc_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape0-event_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-1-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-14]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-True-initial_shape0-proposed_shape0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-2-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ac-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-6]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-0-2-()]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_diagnostics[2]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceGraph_ELBO-subsample]
0.00s setup    tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[DiffTrace_ELBO-nonreparam]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-0.5-0.0-1.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-2-random]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-3-uniform]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-2-uniform]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Normal-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-Size-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-()-()]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-1]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-2-sparse]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-3-uniform]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Binomial-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Stable-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Beta-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-2-unclamped]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-0-3]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape2-value_shape2]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape1-value_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-2]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-None-batch_shape0]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(2,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[3-3-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceGraph_ELBO-full]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[2-None-parallel-None]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.0-0.0-1.5]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-2.0]
0.00s setup    tests/distributions/test_mvn.py::test_shape[3-cov_shape0-loc_shape0]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape0-event_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-list-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-2-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-1.0-0.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-1.0--1.0]
0.00s setup    tests/contrib/oed/test_linear_models_eig.py::test_dv_linear_model
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-1-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-1.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-1-1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ac-2]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-2-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-1-1-(4,)]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[3.0-0.0]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-5-uniform]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-5-sparse]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-1-random]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-5-uniform]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Binomial-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Stable-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LogNormal-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(4,)-(5,)-9]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-3-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.05]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-2-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-()-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[2-()-()]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-full-nonreparam]
0.00s setup    tests/distributions/test_mvn.py::test_shape[1-cov_shape2-loc_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-11]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Dirichlet-tuple-sample_shape0]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-2-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-0.0--1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-0.0-1.0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-2-1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-3-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-1.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-7]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[None-loss-10]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_mean-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[3.0--1.5707963267948966]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-3-sparse]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-list-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-2-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-0.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-1.0]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-2-3-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-0-1-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-2-1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_logsumexp[3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-1-1-(3, 2)]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-2-3]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist2]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[1-parallel-parallel-10000]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-2-sparse]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-1-uniform]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-1]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Stable-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Normal-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-tuple-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-3-2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-1-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-0-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.5]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-6]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-()-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-1-(4,)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-1-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-0-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-1-1-(4,)]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-2-1]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-4-sparse]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-8]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-Size-sample_shape1]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-2-7]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape0-value_shape1]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape3-sample_shape0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->-2]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-0-2-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-None-None]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[VonMises-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-True-sample_shapes0]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-2-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-1.0]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape2-value_shape2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->b-1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->a-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-8]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape3]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_logsumexp[3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-0-1-(4,)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-4]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-2-2]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceGraph_ELBO-subsample]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-subsample-nonreparam]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-2-random]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-2-random]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Poisson-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[VonMises-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-list-sample_shape2]
0.00s call     tests/distributions/test_reshape.py::test_reshape[1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-3-1]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-None-batch_shape4]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-1-3-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-0-3-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-2-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-3-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[2-1-()]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_beta_binomial_static_sample
0.00s setup    tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[DiffTrace_ELBO-reparam]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.03--1.5707963267948966]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape1-event_shape0]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape1-event_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Beta-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Uniform-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(4,)-(5,)-9]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-3-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-1-2]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-1-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-7-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-50]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[1-1-(3, 2)]
0.00s call     tests/ops/test_contract.py::test_einsum_linear[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_generate_data_plate[pyro]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-20-iplate]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc->ac-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-10]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(2,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-3-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-0-2-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-0-1-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1--1.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[sequential-differentiable_loss-1]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-1-uniform]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-5-random]
0.00s setup    tests/distributions/test_mvn.py::test_shape[3-cov_shape2-loc_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Binomial-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-tuple-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-20-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-1-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-2-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-2-1]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-20-plate]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-2-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-1-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(4,)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-2-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-1-2-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4--0.5]
0.00s setup    tests/infer/test_predictive.py::test_posterior_predictive_svi_manual_guide[True]
0.00s setup    tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceGraph_ELBO-reparam]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-16]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Binomial-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-tuple-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-10-clamped]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Matern52]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape0-value_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-False-list]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-3]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->a-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-10]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape3]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-1-3-()]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-2.0]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[None-True-None]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_sample-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-3-sparse]
0.00s setup    tests/distributions/test_rejector.py::test_rejector[0.25-2.0]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape1-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-8]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Multinomial-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[VonMises-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-Size-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.5]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_condition[1-1-()-()]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-5-5-plate]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-None-plate]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-2-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-1-3-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-1-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-1-3-(3, 2)]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-3-random]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape2-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-12]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-False-Size]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-2-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-0.0-1.0]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[RationalQuadratic]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-7-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-1-7]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-20-plate]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-1-2-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-0-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(4,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[1-parallel-parallel-None]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_median-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.1-1.5707963267948966]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-3-sparse]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape0-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Dirichlet-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-True-list]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-2-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.2]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(4,)-Real()]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->ac-2]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-1-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-1-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-1-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-1-1-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[2-parallel-sequential-None]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-4-random]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-1-sparse]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-2-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-True-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Poisson-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-list-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-1-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-1-2]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize3]
0.00s setup    tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-1-7]
0.00s call     tests/distributions/test_pickle.py::test_pickle[MaskedMixture]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->b-2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->-1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,a,ab,b,b,b,b->a-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-3]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-0-3-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-2-3-(3, 2)]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-full-reparam]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.0-0.5-0.5]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Multinomial-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-False-initial_shape2-proposed_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-2-2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,ab->ab-1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ad-1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->b-1]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-1-3-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-2-1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-2-3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-1-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-1-2-(4,)]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-3-random]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-4-random]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-12]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Multinomial-False-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-False-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-Size-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.05]
0.00s call     tests/distributions/test_reshape.py::test_reshape[0-0]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape2-sample_shape0]
0.00s call     tests/contrib/randomvariable/test_random_variable.py::test_tensor_ops
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-None-plate]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-0-3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-0-1-(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[parallel-True-3]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape1-event_shape0]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-7]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-False-initial_shape2-proposed_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-2-3]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Polynomial]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-1-1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,ab->ab-2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->ac-1]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(2,)-()-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-()-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[3-2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-2-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-0-3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-1-3-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceTMC_ELBO-subsample]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_feasible-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-18]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-True-Size]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-10-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-0.0-0.0]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape0-value_shape1]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(4,)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-3-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-0-1-()]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[parallel-True-3]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-Size-sample_shape2]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-10-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-3-2]
0.00s setup    tests/contrib/oed/test_finite_spaces_eig.py::test_marginal_likelihood_finite_space_model
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-2]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-2-3-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-2-1-()]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_sample-TraceGraph_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_sample-TraceEnum_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_mean-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_independent.py::test_independent[normal-3-batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-20-clamped]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape2-value_shape2]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-None-plate]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape5]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(2,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-0-1-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-2]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide[AutoGuideList-None]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_mean-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.7--0.5-0.5]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape1-event_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Cauchy-True-initial_shape2-proposed_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-False-list]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-1-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-7-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[10.0-1.0-0.5]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->a-1]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-0-3-(4,)]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s setup    tests/infer/test_svgd.py::test_conjugate[IMQSteinKernel-univariate]
0.00s setup    tests/distributions/test_rejector.py::test_gamma_elbo[1.0-1.0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-False-initial_shape1-proposed_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-list-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-7-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.5]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape0-value_shape0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->b-1]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_logsumexp[2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-1-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[1-3-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-3-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-2-3-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-2-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-1]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-parallel-inf]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-1.0-0.9-1.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_make_complete_graph[python-2-expected_grid0]
0.00s setup    tests/distributions/test_ig.py::test_log_prob[0.5-2.5-2.5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-list-sample_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-20-clamped]
0.00s setup    tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-2-clamped]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-4]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-0.0-0.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.05]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-()-(2,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-1-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[3-1-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[1-3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[1-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-2-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-1-2-()]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-True-HMC-_empty_model]
0.00s setup    tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceEnum_ELBO-nonreparam]
0.00s setup    tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-parallel-parallel]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[30.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_normal[0.5-21]
0.00s setup    tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-3]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-2-1-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-1-3-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-2-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-1-2-()]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[sequential-False-None]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[0-2]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape2-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-3]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape4-event_shape4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-False-tuple]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-5]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-20-plate]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[1-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-2-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-1-2-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-0-3-(4,)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[sample-a,abi,bcij,bdik->-ijk]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_nonreparameterized_N_is_7
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[rsvi-ReweightedWakeSleep]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[rsvi-Trace_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-False-10]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.44-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[2-parallel-None-None]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-8]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-Size-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.5]
0.00s setup    tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-20-iplate]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->a-2]
0.00s setup    tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(2,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-2-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[3-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-0-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-1-3-(3, 2)]
0.00s setup    tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-1]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[parallel-False-None]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape3-event_shape1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-3-7]
0.00s setup    tests/ops/test_indexing.py::test_value[(2,)-(2,)-(3, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[3-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-()-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[3-1-(4,)]
0.00s setup    tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_diag_normal_guide[True]
0.00s setup    tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceGraph_ELBO-nonreparam]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(3, 2)-Real()]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-False-Size]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-4]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-3-7]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[2-3-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[1-1-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-0-2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-0-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-()-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-1-3-(4,)]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-list-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-,ab->ab-1]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-2-2-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-2-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-2-3-(4,)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[sample-ai,abij,acik->i-ijk]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-14]
0.00s setup    tests/distributions/test_distributions.py::test_expand_error[Gamma-True-initial_shape2-proposed_shape2]
0.00s setup    tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize0]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape0-value_shape2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,b,bc->ac-1]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(3, 2)-(6, 5)]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_gradient.py::test_plate[TraceEnum_ELBO-nonreparam]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[2-None-sequential-None]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_feasible-Trace_ELBO]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-4]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoLowRankMultivariateNormal]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape2-value_shape0]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape3]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-()-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[1-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-2-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[2-3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-2-1-(3, 2)]
0.00s setup    tests/distributions/test_ig.py::test_log_prob[0.5-3.0-4.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-14]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-False-sample_shapes0]
0.00s setup    tests/pyroapi/test_pyroapi.py::test_generate_data[minipyro]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-10]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[1-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-2-2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-1-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-1-()]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.01]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-()]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape2-value_shape1]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[2-1-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[1-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-2-3-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-2-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-2-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-1-3-()]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_mean-Trace_ELBO]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-False-Size]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.10]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-0.0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->d-1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->c-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-10]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[3-(4,)-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_mean[S-0.5-0.9-1.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-1]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-False-list]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.11]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.05]
0.00s setup    tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-1-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Normal]
0.00s setup    tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 1, 1)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-()-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[3-2-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-0-3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-1-2-(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-parallel-inf]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.17-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-18]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Gamma-list-sample_shape2]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-1000]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-()]
0.00s setup    tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[8]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->c-1]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-2-2-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-1-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[0-0-2-()]
0.00s setup    tests/infer/reparam/test_unit_jacobian.py::test_normal[(2, 1, 3)]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-True-tuple]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape1-value_shape2]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Dirichlet]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-3]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape0]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[3-3-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[3-1-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[2-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-()-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[2-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-2-3-()]
0.00s setup    tests/infer/test_gradient.py::test_plate[DiffTrace_ELBO-nonreparam]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_median-Trace_ELBO]
0.00s setup    tests/distributions/test_von_mises.py::test_log_prob_normalized[30.0]
0.00s setup    tests/distributions/test_ig.py::test_log_prob[1.7-3.0-2.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-18]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-2-1]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[1-(3, 2)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-1-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-()-(6, 5)]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-full-0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-3]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-False-sample_shapes1]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-1000]
0.00s call     tests/distributions/test_pickle.py::test_pickle[HalfCauchy]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape2-value_shape0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc->ac-2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,b,bc->ac-2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->ac-1]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(4,)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_condition[2-2-()-(3, 2)]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-2]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_sample-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_mean-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape3-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-13]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-True-list]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-4000]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->c-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-2-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-2-1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-0-1-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_reshape[1-(6,)-(3, 2)]
0.00s setup    tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape1-sample_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-3-7]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,b,bc->ac-2]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape0]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-2-2-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-1-(4,)]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-2.0]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s setup    tests/infer/test_autoguide.py::test_scores[AutoDiagonalNormal]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-18]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.5]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->ac-2]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_logsumexp[2-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-2-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[2-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-0-1-(3, 2)]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[rsvi-RenyiELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_median-TraceGraph_ELBO]
0.00s setup    tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-3-NcvContinuous]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.05]
0.00s call     tests/infer/test_valid_models.py::test_plate_wrong_size_error
0.00s call     tests/distributions/test_reshape.py::test_reshape[1-3]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(4,)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-3-1]
0.00s setup    tests/ops/test_gaussian.py::test_condition[3-1-()-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-2-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-1-3-(4,)]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[3.0-0.5]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.44-0.41]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-True-list]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-1.0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ac-1]
0.00s setup    tests/ops/test_stats.py::test_fit_generalized_pareto[0.8-0.2]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-2-1-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-0-1-(4,)]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-1]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_median-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_feasible-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-True-sample_shapes0]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-100]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-(2,)]
0.00s call     tests/distributions/test_pickle.py::test_pickle[LogisticNormal]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[2-(4,)-(3, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-0-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(3, 2)-(7,)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[marginal-ab,bc->-]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-2-True]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[2-parallel-parallel-None]
0.00s setup    tests/infer/test_enum.py::test_differentiable_loss[None-single]
0.00s setup    tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-1-expand_shape2]
0.00s call     tests/infer/test_valid_models.py::test_variable_clash_in_guide_error[TraceGraph_ELBO]
0.00s setup    tests/ops/test_gaussian.py::test_condition[1-3-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-()-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[1-1-(3, 2)]
0.00s setup    tests/infer/reparam/test_haar.py::test_uniform[(4, 2)--2-True]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-11]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-True-sample_shapes0]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-1000]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-1.0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->a-2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc->ac-2]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-2-2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-2-3-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_size_error[ubersum-ab,bc->-shapes1-b]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99-0.0]
0.00s setup    tests/infer/reparam/test_haar.py::test_uniform[(6,)--1-True]
0.00s setup    tests/infer/test_predictive.py::test_shapes[False]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.55-0.33]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-True-sample_shapes1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{}-7-4]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.05]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-30]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ad-1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->b-1]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape0]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(4,)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[1-2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[2-0-2-()]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-1-1-()]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s setup    tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape0]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_feasible-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_median-TraceEnum_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_mean-Trace_ELBO]
0.00s setup    tests/distributions/test_ig.py::test_log_prob[0.5-3.0-2.5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-True-list]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-1000]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-1.0]
0.00s call     tests/infer/test_valid_models.py::test_variable_clash_in_guide_error[Trace_ELBO]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape0-value_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-True-tuple]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-2-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-3.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-16]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.05]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape0-value_shape2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->c-2]
0.00s setup    tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[1-3-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs11-dims11-2]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-subsample-reparam]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-True-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-5]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-False-sample_shapes1]
0.00s setup    tests/contrib/oed/test_finite_spaces_eig.py::test_nmc_eig_finite_space_model
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-0-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-20]
0.00s call     tests/infer/test_valid_models.py::test_variable_clash_in_guide_error[TraceEnum_ELBO]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape0-value_shape2]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape0-value_shape1]
0.00s setup    tests/ops/test_gaussian.py::test_pad[1-1-2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_pad[0-1-2-(3, 2)]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[1]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape2-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-Size-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(3, 2)-(7,)]
0.00s setup    tests/infer/test_jit.py::test_svi_enum[1-sequential-parallel]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-3-True]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[()-()-Real()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-15]
0.00s setup    tests/distributions/test_conjugate.py::test_variance[dist1]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.02]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.5]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape1-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(2, 3)-3]
0.00s call     tests/contrib/oed/test_ewma.py::test_ewma_log
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-None-batch_shape4]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[3-3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-0-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-0.5]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-True-sample_shapes1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-10]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[5.0-5-1.0]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[2-3-(4,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Poisson-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Categorical-tuple-sample_shape0]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_log_prob[1e-06]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-10]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape1-value_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_reshape[2-(5, 6)-(5, 3, 2)]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(4,)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-7]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-list-sample_shape2]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-1-7]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-full-nonreparam]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-dense-2]
0.00s setup    tests/distributions/test_mvt.py::test_log_prob[2-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-6]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-100]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-0-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.5]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->bc-2]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[3-3-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_condition[2-3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[1-1-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_elbo_plate[2-parallel-parallel-10000]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_median-TraceGraph_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_mean-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[3-6]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-50]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(3, 2)-(6, 5)]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-full-reparam]
0.00s setup    tests/distributions/test_mvt.py::test_log_prob[2-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-10]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-tuple-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-1.0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-2-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[2-3-()]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[2-()-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-subsample-0]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-subsample-reparam]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-14]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_smoke[enum]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.2]
0.00s call     tests/distributions/test_pickle.py::test_pickle[LogNormal]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape2-value_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-2-7]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[1-2-3-(4,)]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-False-None]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-None-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-False-1]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_mean-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[GaussianScaleMixture]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-2]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-100]
0.00s call     tests/ops/test_welford.py::test_welford_diagonal[1-1]
0.00s call     tests/distributions/test_reshape.py::test_reshape[1-0]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape1-value_shape0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_pad[2-1-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1-1.0]
0.00s setup    tests/infer/reparam/test_haar.py::test_uniform[(6,)--1-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[3.0-0.75]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_mean-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(3, 2)-Real()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-10]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-True-Size]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-3-1]
0.00s setup    tests/distributions/test_rejector.py::test_rejector[0.5-2.0]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[2-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-19]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.05]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-()-(7,)]
0.00s setup    tests/ops/test_gaussian.py::test_logsumexp[2-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[2-1-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[sample-a,abi,bcij->-ij]
0.00s setup    tests/infer/test_jit.py::test_beta_bernoulli[JitTraceEnum_ELBO-True]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-0-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-True-tuple]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-4]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-1.0]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[11-()-()-9]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-1-()]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-,bi,abi->b-i]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_feasible-TraceEnum_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_feasible-Trace_ELBO]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[1-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-9]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-True-list]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-1.0-2.0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc->ac-1]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[3-3-()]
0.00s setup    tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_delta_guide[False]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_mean-Trace_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Binomial-list-sample_shape2]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-1000]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-50]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-True-list]
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-do-normal_model]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ad-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(4,)-()]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceTMC_ELBO-subsample]
0.00s setup    tests/distributions/test_rejector.py::test_exponential_elbo[0.25-0.5]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape0-event_shape2]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-True-Size]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.05]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape2-value_shape0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->bc-2]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-8]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-tuple-sample_shape1]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.05]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(2, 3)-2]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[2-()-(4,)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0-0.5]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoNormal]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_sample-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape1-event_shape0]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Poisson-list-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-1.0-1.0]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[1-1-(4,)]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(2, 3)-2]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-10]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-()-(6, 5)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0-0.5]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-3-False]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_feasible-TraceEnum_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_feasible-Trace_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-list-sample_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-3]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[3-1-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-3-(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-full-0]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-full-reparam]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-tuple-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-1.0]
0.00s call     tests/distributions/test_shapes.py::test_diag_normal_log_prob_shape
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(5,)-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-3]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[1-2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[2-2-(3, 2)]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_feasible-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_beta[4.0-1.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-4]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-4]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-()-(6, 5)]
0.00s setup    tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[sequential-True-None]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-tuple-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Cauchy-True-initial_shape0-proposed_shape0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-1.5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-6]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-subsample-reparam]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_sample-Trace_ELBO]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape3-event_shape0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-20]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape2-value_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-3-7]
0.00s setup    tests/ops/test_contract.py::test_ubersum_size_error[ubersum-ab,bc->-shapes0-]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-subsample-reparam]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-15]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-tuple-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-20]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape2-value_shape0]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal[(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[1-3-(3, 2)]
0.00s setup    tests/infer/test_jit.py::test_svi_enum[1-parallel-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_2[batch-1]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_mean-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(3, 2)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-False-Size]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-20]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges756]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceGraph_ELBO-subsample]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist3]
0.00s setup    tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_diag_normal_guide[False]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-full-reparam]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_median-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-16]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[VonMises-list-sample_shape1]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-1000]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-10]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Categorical-tuple-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.2]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoDiagonalNormal]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape2-value_shape2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->a-1]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[2-3-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-3-False]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-list-sample_shape1]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-1000]
0.00s setup    tests/contrib/oed/test_glmm.py::test_log_prob_matches[model10-lm_2p_10_10_1-design0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->b-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[3-3-(3, 2)]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_sample-Trace_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-5]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Poisson-list-sample_shape0]
0.00s setup    tests/distributions/test_conjugate.py::test_mean[dist4]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-100]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-10.0-0.5]
0.00s call     tests/distributions/test_pickle.py::test_pickle[GammaPoisson]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ad-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-5]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape4]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-aij->ai-ij]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01--0.5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Binomial-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Multinomial-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-Size-sample_shape2]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-1000]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-2]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Categorical]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-None-inf]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_mean-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape0-event_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-True-list]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-1.0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->b-2]
0.00s setup    tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape1-gelman_rubin]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[reparam-ReweightedWakeSleep]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_sample-TraceEnum_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_median-Trace_ELBO]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Stable-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Cauchy-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-tuple-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-5]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-None-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_1[batch-10]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-2-True]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-8]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-True-Size]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-100]
0.00s setup    tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-3-NcpContinuous]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-False-tuple]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[3-2-()]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[reparam-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_mvt.py::test_mean_var[()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-7]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Poisson-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-list-sample_shape1]
0.00s setup    tests/distributions/test_delta.py::test_shapes[True-1-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-1.0-0.5]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape0-value_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-10]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data3-expected_log_prob3]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[nonreparam-TraceGraph_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_sample-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Gamma-tuple-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-2]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_feasible-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_feasible-Trace_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-Size-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-1-1-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-1.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99-0.5]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-sequential-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-9]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-list-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-20]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape0-value_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-4]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape5]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[1-3-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(4,)-(7,)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-None-inf]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[reparam-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape1]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Dirichlet-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Beta-list-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-10.0-2.0]
0.00s call     tests/distributions/test_reshape.py::test_reshape[1-4]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-2-1]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_sample-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_mean-TraceEnum_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_mean-Trace_ELBO]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(4,)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[()-()-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-9]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Beta-Size-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.05]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-()-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_rsvi[parallel]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_mean-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape2-event_shape0]
0.00s setup    tests/distributions/test_ig.py::test_log_prob[1.7-2.5-2.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-15]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-Size-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.05]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->bc-1]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_feasible-Trace_ELBO]
0.00s setup    tests/distributions/test_mvt.py::test_log_prob[1-(3, 2)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-1]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-False-list]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-10]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[marginal-ai,abij->i-ij]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-2-True]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_median-Trace_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-10]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Multinomial-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-Size-sample_shape1]
0.00s setup    tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-2-NcpContinuous]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-4]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-0.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-20]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01-0.5]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape3-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Binomial-tuple-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.05]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->d-2]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[1-2-()]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-30-False]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-()-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(3, 2)-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-16]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-list-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-10]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal[(2, 3)-()]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[3-2-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(3, 2)-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7-1.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-20-False]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape1-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-11]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[VonMises-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[VonMises-tuple-sample_shape0]
0.00s setup    tests/distributions/test_delta.py::test_shapes[True-0-0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.2]
0.00s call     tests/distributions/test_reshape.py::test_reshape[0-1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape0-value_shape0]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[2-()-()-9]
0.00s call     tests/test_generic.py::test_not_implemented[pyro]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[2-2-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[marginal-a,abi,bcij,bdik->-ijk]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape2-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-7]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Binomial-tuple-sample_shape1]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_log_prob[1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.5]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape2-value_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-3-7]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->c-2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->c-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-10]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[marginal-a->-]
0.00s setup    tests/ops/test_contract.py::test_ubersum_collide_implemented[ubersum-False]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99--1.0]
0.00s setup    tests/infer/reparam/test_haar.py::test_uniform[(4, 2)--2-False]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_median-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-19]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-tuple-sample_shape1]
0.00s setup    tests/distributions/test_delta.py::test_shapes[True-1-0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_dense_smoke
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-1.0]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape1-value_shape0]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(4,)-Real()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-2-1]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape1]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[2-2-(4,)]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_median-TraceGraph_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_sample-Trace_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-4]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Dirichlet-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Normal-list-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-2-7]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->bc-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[3-size1]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape2]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape0]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-1-1]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-1-(2, 3, 5)]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-()-Real()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-11]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-list-sample_shape1]
0.00s setup    tests/contrib/oed/test_finite_spaces_eig.py::test_vnmc_finite_space_model
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-()]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-50]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-50]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag[mat_size2-3-batch_shape1]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize_shape[1-1-(4,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-,ai->a-i]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-18]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-True-sample_shapes1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-4]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-7-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-0-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-10]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-2]
0.00s call     tests/distributions/test_pickle.py::test_pickle[HalfNormal]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,a,ab,b,b,b,b->a-2]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0--0.5]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.27-0.33]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape2-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-7]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-tuple-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-,ab->ab-2]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(4,)-(6, 5)]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.55-0.33]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_median-TraceEnum_ELBO]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->b-2]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-,bi,abi->b-i]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs0-dims0-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Cauchy-tuple-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-2]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-None]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-1.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-2-7]
0.00s call     tests/contrib/autoname/test_scoping.py::test_only_withs
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-4]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[3-3-(3, 2)]
0.00s setup    tests/distributions/test_mask.py::test_mask[2-2-0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-Size-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-2]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-1.0]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-parallel-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-full-reparam]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-19]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-tuple-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-6]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0--1.0]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s setup    tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[parallel-False-None]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_sample-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape0-event_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape1-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-9]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Poisson-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-Size-sample_shape0]
0.00s setup    tests/distributions/test_conjugate.py::test_mean[dist1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-5]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[map-1]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_sample-TraceGraph_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_median-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_rejector.py::test_standard_gamma_elbo[1.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-12]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-list-sample_shape0]
0.00s setup    tests/distributions/test_delta.py::test_shapes[True-2-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-0.0--1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.05]
0.00s call     tests/nn/test_module.py::test_hierarchy_prior_cached
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-replay-bernoulli_normal_model]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-6]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1-0.0]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-subsample-nonreparam]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_median-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Bernoulli-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Gamma-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Exponential-tuple-sample_shape1]
0.00s setup    tests/distributions/test_delta.py::test_shapes[False-2-0]
0.00s setup    tests/contrib/oed/test_finite_spaces_eig.py::test_lfire_finite_space_model
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-1.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-2]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[2-2-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Geometric-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Geometric-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-list-sample_shape1]
0.00s setup    tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-1000]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[10.0-1.0-1.0]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[marginal-a->-]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[3-2-()]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs2-dims2-1]
0.00s setup    tests/infer/reparam/test_stable.py::test_symmetric_stable[()]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[1-sequential-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[parallel-loss_and_grads-10]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[None-loss-1]
0.00s setup    tests/distributions/test_ig.py::test_sample[2.5-4.0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Exponential-tuple-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-50]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-6]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-True-None]
0.00s setup    tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_delta_guide[True]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.0]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape1-event_shape1]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape1-event_shape0]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-1-(4, 5)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-10]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-True-list]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-3]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-0.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-10]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-3-1]
0.00s setup    tests/optim/test_optim.py::test_clip_norm[1.0-clip_value-SGD]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-2]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs4-dims4-1]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceEnum_ELBO-full]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-15]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-4]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-5]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[5-()-()-9]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-1-1]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[2-1-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-aij->aj-ij]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_model_error_stray_batch_dims[HMC-kwargs0]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(4,)-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/distributions/test_ig.py::test_log_prob[1.7-3.0-4.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-2]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-tuple-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0--1.0--1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(2, 3)-3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-4]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7-0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-15]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Binomial-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Binomial-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Gamma-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[DirichletMultinomial]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-0.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[10.0-1.0-2.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-1.5]
0.00s setup    tests/optim/test_optim.py::test_clip_norm[5.0-clip_norm-SGD]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape4]
0.00s setup    tests/infer/reparam/test_unit_jacobian.py::test_normal[(4, 5)]
0.00s setup    tests/infer/test_enum.py::test_svi_step_smoke[parallel-batch]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-1]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Beta-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-tuple-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-()-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-2-1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.01]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs6-dims6-3]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1-0.5]
0.00s setup    tests/distributions/test_mask.py::test_mask[0-2-1]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(3, 2)-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-14]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-False-list]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-1.0]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-None]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-20]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-1.5]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->a-1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->c-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-2]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[2-1-(3, 2)]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s setup    tests/infer/test_enum.py::test_differentiable_loss[None-batch]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape1-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-7]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Uniform-tuple-sample_shape2]
0.00s setup    tests/distributions/test_delta.py::test_shapes[False-3-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-1.0--1.0]
0.00s setup    tests/contrib/oed/test_finite_spaces_eig.py::test_posterior_finite_space_model
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-10]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-5]
0.00s setup    tests/infer/reparam/test_stable.py::test_subsample_smoke[LatentStableReparam-subsample]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceTMC_ELBO-full]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-subsample-inf]
0.00s setup    tests/infer/test_jit.py::test_svi_enum[2-sequential-parallel]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_sample-TraceGraph_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_sample-Trace_ELBO]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(3, 2)-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-0-expand_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-Size-sample_shape1]
0.00s setup    tests/distributions/test_delta.py::test_shapes[False-1-0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-0.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-10]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-50]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->c-1]
0.00s setup    tests/ops/test_gaussian.py::test_logsumexp[1-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-aij->ai-ij]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[None-differentiable_loss-10]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_median-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape0-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-18]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-Size-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-4]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-4]
0.00s setup    tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[4]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-1.0]
0.00s call     tests/distributions/test_one_hot_categorical.py::test_support_dims[dim=2-probs0]
0.00s setup    tests/optim/test_optim.py::test_clip_norm[1.0-clip_norm-SGD]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-5]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-None-batch_shape1]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-None-batch_shape0]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-None-batch_shape1]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4-0.0]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_mean-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Categorical-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-tuple-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar': True, 'num_quant_bins': 2}-7-12]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-20]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-10]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape3]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-bi,abi->b-i]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1--0.5]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-full-inf]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_median-Trace_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Geometric-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Beta-tuple-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-2]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-0.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape0-value_shape0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-2]
0.00s setup    tests/infer/reparam/test_stable.py::test_subsample_smoke[SymmetricStableReparam-subsample]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape3-event_shape0]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-0-(2, 3, 5)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-tuple-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-(2,)]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-None]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape2-value_shape2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.01]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,a,ab,b,b,b,b->a-2]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape0-event_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-5]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Exponential-tuple-sample_shape2]
0.00s setup    tests/distributions/test_delta.py::test_shapes[False-3-3]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-(2,)]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-1.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-5]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-False]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-13]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Binomial-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-tuple-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(2, 3)-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-6]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.00s setup    tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.0-()]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data3-expected_log_prob3]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_feasible-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-8]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Geometric-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[VonMises-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-list-sample_shape1]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-10]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-1.0--1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[5.0-5-0.5]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->a-2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->a-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-5]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape2]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[2-2-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-ai->a-i]
0.00s setup    tests/infer/reparam/test_haar.py::test_uniform[(2, 5)--1-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-3-True]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-4]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-tuple-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-()-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-2]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-None-batch_shape2]
0.00s setup    tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[None-False-None]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-full-0]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_median-Trace_ELBO]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape1-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-9]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Categorical-list-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-10]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.03]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->a-1]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4--0.5]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-sequential-inf]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-dense-3]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape3-event_shape1]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_mvn_shape[1-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-10]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape1]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Multinomial-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Uniform-tuple-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-tuple-sample_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-6]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape2]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_beta_binomial[False]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-subsample-reparam]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-3-False]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_feasible-TraceGraph_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_mean-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_mvt.py::test_rsample[1-9.1]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-0-(2, 3, 5)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-9]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[VonMises-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Dirichlet-list-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0--1.0--1.0]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-()]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{}-7-12]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape1-value_shape0]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape0-value_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-3]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7--0.5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-9]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Multinomial-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Cauchy-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-tuple-sample_shape0]
0.00s setup    tests/distributions/test_delta.py::test_shapes[True-3-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0--1.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-None]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 4}-7-12]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-0-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.3]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-None-1]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_feasible-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-6]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-0.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.05]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[5.0-5-2.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-6]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-None-inf]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[nonreparam-Trace_ELBO]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-subsample-reparam]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape2-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-6]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-list-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-0.0]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-()]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.05]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Cauchy-True-initial_shape2-proposed_shape2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.01]
0.00s setup    tests/optim/test_optim.py::test_clip_norm[3.0-clip_value-SGD]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-3]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-None-batch_shape2]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[3-1-(4,)]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs8-dims8-2]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-1.5]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-parallel-inf]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-parallel-1]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-parallel-1]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape1-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-16]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Cauchy-Size-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0--1.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-0.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[infinite-,ab->ab-2]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixyj,iyzj->ij]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-sequential-inf]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-sequential-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[parallel-loss_and_grads-1]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-10]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Normal-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LogNormal-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Uniform-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-2]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->c-2]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape2]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-1.5]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-0.5]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[parallel-loss-10]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape1-event_shape2]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape0-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-5]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-False-Size]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-1.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 2}-7-12]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-3.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_4[False-True-10]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[None-loss_and_grads-10]
0.00s setup    tests/distributions/test_von_mises.py::test_log_prob_normalized[0.1]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape2-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-7]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Uniform-list-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-4]
0.00s setup    tests/contrib/oed/test_finite_spaces_eig.py::test_dv_finite_space_model
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-1.0]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[sample-ai,abij->i-ij]
0.00s setup    tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist1]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_1[single-10]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_sample-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape3-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-19]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Geometric-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Exponential-list-sample_shape0]
0.00s setup    tests/distributions/test_delta.py::test_shapes[True-2-0]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-()]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-7-3]
0.00s call     tests/distributions/test_stable.py::test_variance[1.7-0.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.03]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-6]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_1[masked-1]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-flat-2]
0.00s setup    tests/distributions/test_mask.py::test_mask[0-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-4]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Normal-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Gamma-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-1.5]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-None-batch_shape5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(3, 2)-(6, 5)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4-1.0]
0.00s setup    tests/infer/reparam/test_haar.py::test_uniform[(2, 3, 1)--2-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[None-loss_and_grads-1]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide[AutoGuideList-init_to_mean]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_feasible-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape0-event_shape1]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape2-event_shape0]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-()-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-8]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-list-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(5,)-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-6]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(4,)-()]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-parallel-inf]
0.00s setup    tests/distributions/test_mvn.py::test_shape[1-cov_shape0-loc_shape0]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape1-event_shape0]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape2-event_shape1]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape0-event_shape0]
0.00s setup    tests/distributions/test_mask.py::test_mask[0-1-0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-11]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype2-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Bernoulli-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Exponential-Size-sample_shape2]
0.00s setup    tests/distributions/test_delta.py::test_shapes[False-2-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-4]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.05]
0.00s call     tests/distributions/test_spanning_tree.py::test_make_complete_graph[python-3-expected_grid1]
0.00s call     tests/distributions/test_reshape.py::test_reshape[0-3]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape4]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_4[False-True-1]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-1-(5,)]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-0-(4, 5)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-16]
0.00s setup    tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-1-expand_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Categorical-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[VonMises-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-list-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-1]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-()]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-0-3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(5,)-3]
0.00s call     tests/distributions/test_coalescent.py::test_simple_smoke[3-()-()-9]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.03]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-10]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[1-2-()]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-0.5]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-subsample-inf]
0.00s setup    tests/infer/test_enum.py::test_differentiable_loss[parallel-single]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape1-event_shape0]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-1-(4, 5)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-14]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-Size-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-2]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-(2,)]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-4]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[1-2-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[marginal-a,abi,bcij->-ij]
0.00s setup    tests/infer/reparam/test_unit_jacobian.py::test_normal[(6,)]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-1.5]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-subsample-inf]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[sample-10]
0.00s setup    tests/distributions/test_ig.py::test_log_prob[0.5-2.5-4.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-7]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Exponential-Size-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar_full_mass': 3, 'num_quant_bins': 2}-7-12]
0.00s call     tests/ops/test_contract.py::test_ubersum_jit[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-Size-sample_shape2]
0.00s setup    tests/optim/test_optim.py::test_clip_norm[5.0-clip_value-SGD]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape4]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-sequential-inf]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-Trace_ELBO-full]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide[AutoGuideList-init_to_median]
0.00s setup    tests/distributions/test_mask.py::test_mask[0-2-0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-8]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Normal-tuple-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.1]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[8-size2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag[mat_size2-3-batch_shape0]
0.00s setup    tests/infer/test_enum.py::test_svi_step_smoke[None-batch]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_sample-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-15]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Bernoulli-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Multinomial-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Cauchy-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Beta-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(5,)-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-6]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape1]
0.00s setup    tests/infer/reparam/test_haar.py::test_uniform[(2, 3, 1)--2-True]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-full-inf]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal_mvn[1-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-3]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Uniform-Size-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-4]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-6]
0.00s setup    tests/infer/test_enum.py::test_svi_step_smoke[sequential-batch]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Bernoulli-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-list-sample_shape0]
0.00s setup    tests/distributions/test_delta.py::test_shapes[True-3-0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-4]
0.00s setup    tests/contrib/test_hessian.py::test_hessian_mvn
0.00s setup    tests/perf/test_benchmark.py::test_benchmark[PoissonGamma::reparam=False_Trace]
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-replay-normal_model]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-10]
0.00s setup    tests/ops/test_ssm_gp.py::test_matern_kernel[1.5-2]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[2-3-(3, 2)]
0.00s setup    tests/infer/test_enum.py::test_differentiable_loss[sequential-single]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_median-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_ig.py::test_log_prob[1.7-2.5-4.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-9]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-False-sample_shapes0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-3]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs9-dims9-3]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01--1.0]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-full-0]
0.00s setup    tests/infer/test_jit.py::test_cond_indep_equality[x0-y0]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.27-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[1-sequential-None]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_median-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-11]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-Size-sample_shape2]
0.00s setup    tests/contrib/oed/test_glmm.py::test_log_prob_matches[model14-normal_inv_gamma_2_2_10_10-design4]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-None]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-None]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-5]
0.00s setup    tests/ops/test_gaussian.py::test_reshape[1-(5, 6)-(5, 3, 2)]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-16]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape3]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape3-event_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Dirichlet-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-Size-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-4]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-(2,)]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar': True, 'num_quant_bins': 2}-7-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-10]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7-0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_symmetric_stable[(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-Trace_ELBO-subsample]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape2-event_shape1]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape0-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-13]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Bernoulli-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-list-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-1]
0.00s setup    tests/contrib/oed/test_glmm.py::test_log_prob_matches[model-sigmoid_example-design6]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-5]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape2]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s setup    tests/infer/test_gradient.py::test_plate[Trace_ELBO-reparam]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape0-event_shape1]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape3-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-18]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Gamma-tuple-sample_shape0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(2, 3)-2]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape5]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[2-3-(3, 2)]
0.00s setup    tests/infer/reparam/test_neutra.py::test_neals_funnel_smoke[False]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-subsample-nonreparam]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[2-sequential-parallel]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape0-event_shape2]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-0-(4, 5)]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(4,)-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-15]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-True-sample_shapes1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(5,)-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Gamma-True-initial_shape0-proposed_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-5]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[sample-a,a->-]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0--1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4-0.0]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-subsample-nonreparam]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_feasible-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-17]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Dirichlet-Size-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-ij->ij]
0.00s setup    tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape2-autocorrelation]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(3, 2)-()]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-full-nonreparam]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_onehot_entropy_grad[1.0]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-1-(2, 3, 5)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-18]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-list-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-1]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-(2,)]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-10.0-1.0]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(4,)-Real()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-6]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[1-1-(4,)]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-1.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1-0.0]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_sample-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(4,)-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-5]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_shared_cov_log_prob[1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[VonMises-list-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar_full_mass': 3, 'num_quant_bins': 2}-7-4]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,ab->ab-2]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4--1.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-2-False]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-9]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Binomial-tuple-sample_shape2]
0.00s setup    tests/distributions/test_delta.py::test_shapes[False-3-0]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-0.0]
0.00s call     tests/distributions/test_reshape.py::test_reshape[0-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-3]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[2-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0-0.5]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape2-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-13]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Bernoulli-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Cauchy-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Dirichlet-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Beta-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Uniform-tuple-sample_shape1]
0.00s setup    tests/distributions/test_delta.py::test_shapes[False-1-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[MultivariateNormal]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[3-2-(4,)]
0.00s setup    tests/ops/test_gaussian.py::test_reshape[3-(6,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0-0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0-1.0]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-full-inf]
0.00s setup    tests/infer/test_enum.py::test_elbo_rsvi[sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[1-None-None]
0.00s setup    tests/infer/test_enum.py::test_svi_step_smoke[sequential-single]
0.00s setup    tests/infer/test_enum.py::test_svi_step_smoke[None-single]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape1-event_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-5]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Bernoulli-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Cauchy-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Beta-Size-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-2]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-(2,)]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-(2,)]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-0-batch_shape0-sample_shape0]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->d-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-3]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(4,)-()]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-()-(6, 5)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0--0.5]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-subsample-nonreparam]
0.00s setup    tests/distributions/test_rejector.py::test_rejector[0.5-0.5]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(4,)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-6]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape3-event_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-0-expand_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-list-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0--1.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-7-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-4]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape4]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_reshape[2-(6,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7--1.0]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape1-event_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-4]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-True-tuple]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-()-2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-3-2-7]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[map-ab,bc->-]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-18]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape3-event_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Stable-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Beta-tuple-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[1.0-0.0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-()-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-3-7]
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-condition-normal_normal_model]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-10]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-3]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[1-2-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3-0.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1-0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99--0.5]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-3-False]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[map-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-5]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[VonMises-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-tuple-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 4}-7-4]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_overdispersed_seir_smoke[{}-7-3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-()-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-3]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4-0.5]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-False-2200]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-mask4]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-Trace_ELBO-full]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-19]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-True-list]
0.00s setup    tests/distributions/test_delta.py::test_shapes[True-3-3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-1.0-0.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-4]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-3]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[sample-2]
0.00s setup    tests/distributions/test_mask.py::test_mask[1-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-11]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-True-Size]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-4]
0.00s setup    tests/contrib/autoguide/test_mean_field_entropy.py::test_guide_entropy[mean_field_guide-args1-expected_entropy1]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape0-value_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-1-1]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceGraph_ELBO-full]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_mean-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(3, 2)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-8]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-15]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-False-list]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-4]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-5]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[2-2-()]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-2.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7--0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-subsample-0]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_2[single-1]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide[AutoDelta-None]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape0-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-True-list]
0.00s call     tests/ops/einsum/test_adjoint.py::test_marginal[->]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-1-1]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape1]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-0.5]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_mean-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_feasible-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(3, 2)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-19]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-Size-sample_shape2]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape3-event_shape3]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-2-random]
0.00s setup    tests/ops/test_gaussian.py::test_reshape[2-(5, 6)-(5, 3, 2)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[marginal-ai,abij,acik->i-ijk]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_mean-Trace_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-2]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-4]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-2]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-0.0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(3, 2)-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[3-2-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[1-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-None-inf]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Normal-list-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-4]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-0-3]
0.00s setup    tests/contrib/test_minipyro.py::test_plate_ok[minipyro-py]
0.00s call     tests/ops/test_contract.py::test_ubersum[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[7-size2]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[7-size1]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[3-3-(3, 2)]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-3.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3-0.0]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-dense-1]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-100]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_sample-Trace_ELBO]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-0-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-17]
0.00s setup    tests/contrib/oed/test_glmm.py::test_log_prob_matches[lm_2p_10_10_1-model21-design1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_overdispersed_sir_smoke[{}-7-3]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(4,)-(6, 5)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99-1.0]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal-0.1]
0.00s setup    tests/distributions/test_mvt.py::test_log_prob_normalization[2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-2]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-Size-sample_shape1]
0.00s setup    tests/distributions/test_conjugate.py::test_variance[dist4]
0.00s setup    tests/distributions/test_conjugate.py::test_variance[dist3]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-100]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-4]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-()]
0.00s call     tests/distributions/test_spanning_tree.py::test_make_complete_graph[python-2-expected_grid0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-2-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.01]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(4,)-(7,)]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-1-(5,)]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-True-sample_shapes0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-1]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoDiagonalNormal]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.01]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-5]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[3-1-(3, 2)]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0-0.5]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[sequential-loss-1]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-18]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-False-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-Size-sample_shape1]
0.00s setup    tests/distributions/test_delta.py::test_shapes[False-2-2]
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-do-normal_normal_model]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-None-batch_shape3]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(4,)-(7,)]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-1.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-sequential-inf]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-subsample-0]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-subsample-0]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-3-True]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.27-0.33]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[parallel-differentiable_loss-1]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-dense-3]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_feasible-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[0-0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-12]
0.00s setup    tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-2-expand_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-Size-sample_shape0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-1.0]
0.00s setup    tests/ops/test_ssm_gp.py::test_matern_kernel[0.5-1]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99--1.0]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-subsample-0]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-full-nonreparam]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.55-0.43]
0.00s setup    tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_feasible-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-16]
0.00s setup    tests/distributions/test_delta.py::test_shapes[False-3-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-3]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape2-event_shape2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.3]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-None-batch_shape1]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-3.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99--0.5]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-True-2200]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide[AutoNormal-init_to_median]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(4,)-Real()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-11]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-Size-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-4]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-1]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoMultivariateNormal]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape3-event_shape3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-3]
0.00s setup    tests/ops/test_stats.py::test_fit_generalized_pareto[1.3-0.2]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[1-3-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_logsumexp[2-(4,)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1-1.0]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-Trace_ELBO-subsample]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-17]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Bernoulli-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[VonMises-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-list-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-4]
0.00s setup    tests/contrib/oed/test_glmm.py::test_log_prob_matches[model15-lr_10_10-design5]
0.00s setup    tests/contrib/oed/test_finite_spaces_eig.py::test_marginal_finite_space_model
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-0-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-1.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-mask2]
0.00s setup    tests/distributions/test_spanning_tree.py::test_make_complete_graph[cpp-2-expected_grid0]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-0-2]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_bernoulli_entropy_grad[1.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-18]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-False-Size]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwj->ij]
0.00s setup    tests/infer/reparam/test_transform.py::test_log_normal[(2, 3)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[sequential-False-None]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-subsample-0]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-full-reparam]
0.00s setup    tests/distributions/test_spanning_tree.py::test_make_complete_graph[python-3-expected_grid1]
0.00s setup    tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-0-(5,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-12]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LogNormal-Size-sample_shape0]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[3-3-()]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[3-1-(3, 2)]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape2-event_shape0]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_mvn_shape[3-(4,)]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-list-sample_shape1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwj->ji]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[3-1-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs10-dims10-2]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs3-dims3-2]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.27-0.43]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[1-None-sequential]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-flat-3]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Poisson-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Cauchy-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-list-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-7-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Cauchy-True-initial_shape1-proposed_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-1-1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-4-sparse]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->-1]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(3, 2)-(7,)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8-0.5]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-full-inf]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[2-parallel-parallel]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[2-1]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(3, 2)-Real()]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-()-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Multinomial-Size-sample_shape2]
0.00s setup    tests/distributions/test_conjugate.py::test_variance[dist2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0--1.0-0.0]
0.00s setup    tests/contrib/oed/test_glmm.py::test_log_prob_matches[model12-lm_2p_10_10_1_w12-design2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-1-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4-0.5]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-subsample-inf]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-40-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-3-True]
0.00s setup    tests/distributions/test_mvt.py::test_log_prob[1-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-11]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-3]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-7-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.03]
0.00s call     tests/test_util.py::test_warn_if_nan
0.00s setup    tests/optim/test_optim.py::test_clip_norm[1.0-clip_value-Adam]
0.00s setup    tests/ops/test_gaussian.py::test_reshape[3-(5, 6)-(5, 3, 2)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[marginal-a,a->-]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-1.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-50-False]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-dense-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-True-tuple]
0.00s call     tests/infer/test_valid_models.py::test_plate_broadcast_error[TraceEnum_ELBO]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[1-0.1]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(3, 2)-(7,)]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-2.0]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-3.0]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-subsample-reparam]
0.00s setup    tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-0-expand_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LogNormal-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-Size-sample_shape1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.1]
0.00s setup    tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-4]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape3]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4--1.0]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-full-0]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.55-0.43]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-flat-2]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_guide_list_x]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoMultivariateNormal-0.0001]
0.00s setup    tests/distributions/test_mask.py::test_mask[2-1-0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-3]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Dirichlet-Size-sample_shape1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-1-random]
0.00s setup    tests/ops/test_gaussian.py::test_logsumexp[1-(4,)]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs17-dims17-2]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs12-dims12-1]
0.00s setup    tests/distributions/test_stable.py::test_normal[0.5-1]
0.00s setup    tests/distributions/test_mvt.py::test_rsample[2-9.1]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[2-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-8]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Categorical-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Stable-Size-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-0.0-0.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-7-3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.03]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[1-3-()]
0.00s setup    tests/ops/test_contract.py::test_adjoint_marginal[a,abi,bcij,bdik->-ijk]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceTMC_ELBO-subsample]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-dense-2]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoNormal]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_beta[1.0-4.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Categorical-Size-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Stable-list-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-Size-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-0.0-0.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.03]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag[mat_size0-2-batch_shape0]
0.00s setup    tests/ops/test_gaussian.py::test_reshape[2-(6,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0--1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99-0.5]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-full-nonreparam]
0.00s setup    tests/distributions/test_mask.py::test_mask[1-1-0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-13]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Normal-tuple-sample_shape2]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-100]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.3]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[3-2-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(4,)-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-scale3]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-subsample-inf]
0.00s setup    tests/infer/test_jit.py::test_svi_enum[1-parallel-parallel]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-flat-3]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-17]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-list-sample_shape1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-1.5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-()-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[3-1-()]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-TraceTMC_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-full-inf]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-subsample-inf]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.55-0.33]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-4]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-False-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Geometric-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Multinomial-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LogNormal-tuple-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_overdispersed_seir_smoke[{'haar_full_mass': 2}-7-3]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-1.5]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.03]
0.00s setup    tests/ops/test_newton.py::test_newton_step[1-None-batch_shape5]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[sample-ai,ai->i-i]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[sample-a,abi->-i]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01-1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01--1.0]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-subsample-inf]
0.00s setup    tests/infer/test_jit.py::test_infer_discrete[map-1]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-full-nonreparam]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-2-False]
0.00s setup    tests/infer/test_discrete.py::test_plate_smoke[2-sample]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal_100-1e-08]
0.00s setup    tests/distributions/test_rejector.py::test_standard_gamma_elbo[2.0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-False-tuple]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-4]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape3-event_shape3]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3-0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1-0.5]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_predictive[False-100]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-False-1800]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=True-HMC]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceTMC_ELBO-full]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[1-None-parallel]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-flat-3]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-flat-1]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[3-(5,)]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-1-2]
0.00s setup    tests/distributions/test_mask.py::test_mask[1-0-0]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_mvn_shape[2-(2, 3)]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-False-tuple]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-0-7]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape4]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(3, 2)-(6, 5)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0-1.0]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-None-1]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-full-inf]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_1[masked-10]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.44-0.33]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape2-event_shape0]
0.00s setup    tests/distributions/test_mask.py::test_mask[2-0-0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Categorical-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-Size-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-0-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-7-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-1-1-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.1]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[2-1-()]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3--1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceGraph_ELBO-full]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoMultivariateNormal-1e-08]
0.00s setup    tests/distributions/test_von_mises.py::test_log_prob_normalized[0.01]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-1-0]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_mvn_shape[3-(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-16]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-True-Size]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-7-3]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_overdispersed_sir_smoke[{'haar_full_mass': 2}-7-3]
0.00s call     tests/distributions/test_kl.py::test_kl_independent_normal[()-()]
0.00s setup    tests/optim/test_optim.py::test_clip_norm[3.0-clip_norm-SGD]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixj->ji]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,wi->i]
0.00s setup    tests/ops/test_contract.py::test_ubersum[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-subsample-0]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-full-0]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[parallel-differentiable_loss-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-11]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-True-Size]
0.00s setup    tests/distributions/test_delta.py::test_shapes[False-0-0]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-None-batch_shape3]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape3]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[map-a,a->-]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[ab,bcdi,deij->eij-ij]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-0-1]
0.00s setup    tests/distributions/test_mask.py::test_mask[2-2-1]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(4,)-Real()]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Uniform-Size-sample_shape2]
0.00s setup    tests/distributions/test_delta.py::test_shapes[True-2-1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-0-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-1-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.3]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag[mat_size1-1-batch_shape1]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape2]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-2.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0--0.5]
0.00s setup    tests/infer/test_valid_models.py::test_dim_allocation_ok[TraceEnum_ELBO-True]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceGraph_ELBO-subsample]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-full-inf]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-subsample-0]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-30-False]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-1-0]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-()-Real()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Dirichlet-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Gamma-Size-sample_shape1]
0.00s setup    tests/distributions/test_delta.py::test_shapes[True-3-1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-1.0-0.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Gamma-True-initial_shape2-proposed_shape2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.3]
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[1-1.0]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-1.0]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-2.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8-1.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-15]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Exponential-Size-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-1.0-0.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-1.0]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-1.0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,wxi,xi->i]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8--1.0]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-0-0]
0.00s setup    tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape0-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-8]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Normal-tuple-sample_shape1]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->a-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag[mat_size0-2-batch_shape1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(3, 2)-(7,)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[sample-ab,bc->-]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-2.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7--1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01--0.5]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-full-inf]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide_2[AutoDelta-independent]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoNormal-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-2-1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-17]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Geometric-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-tuple-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-0-7]
0.00s setup    tests/optim/test_optim.py::test_clip_norm[3.0-clip_norm-Adam]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[2-1-()]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.27-0.43]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-1-0]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape2-event_shape2]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[()-()-GreaterThan(lower_bound=0.0)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-19]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-12]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-True-sample_shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-True-list]
0.00s setup    tests/distributions/test_conjugate.py::test_mean[dist3]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-100]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-7-7]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.1]
0.00s setup    tests/poutine/test_counterfactual.py::test_counterfactual_query[False-True-False]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[2-2-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8-0.5]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-full-0]
0.00s setup    tests/infer/test_jit.py::test_beta_bernoulli[JitTraceEnum_ELBO-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[1-parallel-parallel]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-flat-1]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-5]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LogNormal-list-sample_shape1]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-0-7]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoIAFNormal]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.01]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-10-False]
0.00s setup    tests/infer/test_enum.py::test_differentiable_loss[sequential-batch]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_mvn_shape[1-(4,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-12]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Multinomial-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LogNormal-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Uniform-list-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_overdispersed_sir_smoke[{'num_quant_bins': 2}-7-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.3]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[2-1-(3, 2)]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8-1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8--0.5]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_3[masked-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_3[batch-10]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal_mvn[2-()]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-list-sample_shape0]
0.00s setup    tests/contrib/oed/test_glmm.py::test_log_prob_matches[model13-nz_lm_2p_10_10_1-design3]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0--0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7-0.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_3[single-10]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide[AutoDelta-init_to_mean]
0.00s setup    tests/distributions/test_mask.py::test_mask[2-1-1]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-False-sample_shapes1]
0.00s setup    tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-2-NcvContinuous]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0--1.0--1.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-0-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.1]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-()-(6, 5)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99--1.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-True-10]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_3[masked-10]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_3[single-1]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_guide_list_x]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-19]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-False-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Stable-tuple-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Exponential-list-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0--1.0-0.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-7-7]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[map-a,a->-]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->c-2]
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[1-0.1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.00s setup    tests/ops/test_contract.py::test_ubersum[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4--0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1--1.0]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-flat-1]
0.00s setup    tests/distributions/test_rejector.py::test_standard_gamma_elbo[5.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-14]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-tuple-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-1.0-1.0]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape4-event_shape4]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[LowRankMultivariateNormal]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8--0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3-0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8-0.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3--0.5]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape0]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-2-0]
0.00s setup    tests/distributions/test_mask.py::test_mask[1-2-0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-15]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-4]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-11]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-True-sample_shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Multinomial-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Normal-Size-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Beta-tuple-sample_shape1]
0.00s setup    tests/ops/test_ssm_gp.py::test_matern_kernel[0.5-2]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[marginal-ai,ai->i-i]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-full-0]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_mvn_shape[2-()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-16]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-11]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Normal-Size-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-2-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-3-uniform]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-5-sparse]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[3-1-()]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[map-a,abi,bcij->-ij]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0-0.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8--1.0]
0.00s setup    tests/infer/test_valid_models.py::test_dim_allocation_error[Trace_ELBO-False]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceEnum_ELBO-full]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_param_ok[Trace_ELBO-full]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-2-0]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-1-2]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-2-2]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-1-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-9]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Stable-tuple-sample_shape2]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_overdispersed_seir_smoke[{'num_quant_bins': 2}-7-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.01]
0.00s setup    tests/optim/test_optim.py::test_clip_norm[5.0-clip_norm-Adam]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-()-(6, 5)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-subsample-inf]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[2-None-None]
0.00s setup    tests/distributions/test_spanning_tree.py::test_make_complete_graph[python-4-expected_grid2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-5]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-16]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Gamma-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Exponential-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[MultivariateStudentT]
0.00s setup    tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[12]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-7-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.05]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.01]
0.00s setup    tests/poutine/test_counterfactual.py::test_counterfactual_query[True-True-False]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->ac-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-w,wx,xy,yz->]
0.00s setup    tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3--0.5]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_3[batch-1]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.55-0.43]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_onehot_entropy_grad[0.5]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal_mvn[1-(4,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-17]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-3]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-6]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Geometric-tuple-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Poisson-list-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-0.0--1.0]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-7-7]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-7-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-1-7]
0.00s setup    tests/optim/test_optim.py::test_clip_norm[3.0-clip_value-Adam]
0.00s setup    tests/optim/test_optim.py::test_clip_norm[1.0-clip_norm-Adam]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-wi,wi->i]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3-1.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-2-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_normal[None-differentiable_loss]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[sequential-loss_and_grads-10]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_mask.py::test_mask[1-2-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-14]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-6]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-14]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-False-list]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_log_prob[1e-08]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-1.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.1]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8-1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_symmetric_stable[(2, 3)]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gamma_normal[True-JIT=False]
0.00s setup    tests/infer/test_jit.py::test_svi[JitTrace_ELBO-10]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-Size-sample_shape1]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape4-event_shape4]
0.00s setup    tests/poutine/test_counterfactual.py::test_counterfactual_query[True-True-True]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_reshape[3-(6,)-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1--0.5]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-2-0]
0.00s setup    tests/distributions/test_mvt.py::test_mean_var[(4,)]
0.00s setup    tests/distributions/test_mask.py::test_mask[0-2-2]
0.00s setup    tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape0-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-7]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Poisson-Size-sample_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[8-size0]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[1-1-(3, 2)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(3, 2)-(7,)]
0.00s setup    tests/infer/test_predictive.py::test_posterior_predictive_svi_manual_guide[False]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_guide_list_x]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-2-2]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-1-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-13]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-True-list]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-list-sample_shape2]
0.00s setup    tests/ops/test_welford.py::test_welford_dense[1000-1]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-3.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8-0.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8--1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0-0.5]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-mask3]
0.00s setup    tests/infer/test_valid_models.py::test_plate_ok[TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.27-0.43]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_guide_callable]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[4-(5,)]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape2-event_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-False-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-True-tuple]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.2]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Beta]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape0-value_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-3-7]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-None-batch_shape2]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4-1.0]
0.00s setup    tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[parallel-True-None]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-full-nonreparam]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[parallel-loss-1]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_guide_list_x]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_mvn_shape[3-()]
0.00s setup    tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(4,)-Real()]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-1.0]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-1-uniform]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-True-1800]
0.00s setup    tests/infer/test_jit.py::test_svi[JitTrace_ELBO-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_2[masked-10]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[sample-1]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-0-1]
0.00s setup    tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape1-event_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-12]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-18]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-False-list]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.05]
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-replay-trivial_model]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(4,)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[3-3-()]
0.00s setup    tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-2.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0-1.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[1-parallel-None]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-0-0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Bernoulli-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LogNormal-tuple-sample_shape2]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.1]
0.00s call     tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.03]
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-do-trivial_model]
0.00s setup    tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->a-2]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-ij->ji]
0.00s setup    tests/ops/test_newton.py::test_newton_step[3-None-batch_shape5]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[1-2-(4,)]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal-1e-08]
0.00s setup    tests/distributions/test_rejector.py::test_rejector[1.0-1.0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-19]
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-do-bernoulli_normal_model]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag[mat_size1-1-batch_shape0]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[sample-100]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide[AutoNormal-init_to_mean]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[3-()]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-True-tuple]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-False-Size]
0.00s setup    tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-2-NcpContinuous]
0.00s setup    tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-0-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-2-uniform]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[1-1]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[1-(4,)]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Binomial-list-sample_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-2-7]
0.00s setup    tests/ops/test_contract.py::test_ubersum[a->,a-]
0.00s setup    tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-False-None]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-10]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-2]
0.00s setup    tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-False-sample_shapes1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.3]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-Trace_ELBO-full]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-list-sample_shape1]
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-replay-normal_normal_model]
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-condition-normal_model]
0.00s setup    tests/ops/test_stats.py::test_fit_generalized_pareto[0.8-0.5]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_marginalize[1-3-()]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-aij->aj-ij]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-mask2]
0.00s setup    tests/infer/test_svgd.py::test_conjugate[IMQSteinKernel-multivariate]
0.00s setup    tests/infer/test_multi_sample_elbos.py::test_outer_inner[RenyiELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-10-False]
0.00s setup    tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape3]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal[()-()]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-True-list]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-2-random]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-i->i]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_shape[1-()-()]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[map-ai,abij->i-ij]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3--1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99-0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4--1.0]
0.00s setup    tests/infer/test_jit.py::test_svi_enum[2-parallel-parallel]
0.00s setup    tests/infer/test_enum.py::test_differentiable_loss[parallel-batch]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-dense-1]
0.00s setup    tests/distributions/test_rejector.py::test_rejector[0.25-1.0]
0.00s setup    tests/distributions/test_mask.py::test_mask[2-2-2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-13]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-9]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-6]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-3-1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-4-sparse]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixyj,iyzj->ji]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwxj,ixyj,iyzj,izwj->ij]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4-0.5]
0.00s setup    tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape1]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-dense-1]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-0-expand_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-0.0--1.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-list-sample_shape0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7--1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99-1.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1-0.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_1[single-1]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-dense-3]
0.00s setup    tests/distributions/test_rejector.py::test_rejector[1.0-0.5]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[1-(2, 3)]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.3]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoDiagonalNormal]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-5-uniform]
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-condition-trivial_model]
0.00s setup    tests/nn/test_module.py::test_decorator[1-DecoratorModel]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0--1.0]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-False]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-18]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-8]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-True-Size]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-1-uniform]
0.00s setup    tests/optim/test_optim.py::test_clip_norm[5.0-clip_value-Adam]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-()-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01-1.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-2-False]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-2-1]
0.00s setup    tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape1]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype0-event_shape2]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[1-event_shape0]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(4,)-(6, 5)]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(3, 2)-(7,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_logsumexp[1-(4,)]
0.00s setup    tests/infer/test_jit.py::test_svi_enum[2-sequential-sequential]
0.00s setup    tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape0]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-13]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-False-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-True-list]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.2]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-3-sparse]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-wx,xy,yz,zw->]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[2-None]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-,ai->a-i]
0.00s setup    tests/nn/test_module.py::test_constraints[shape7-constraint_7]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0-1.0]
0.00s setup    tests/infer/test_jit.py::test_svi[JitTraceEnum_ELBO-1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-True-Size]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Stable-tuple-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Gamma-Size-sample_shape2]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-3-random]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-2-sparse]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-wxi,xyi,yzi,zwi->i]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-wi,xi->i]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_reshape[3-(5, 6)-(5, 3, 2)]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs18-dims18-1]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01-0.0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[Trace_ELBO-full]
0.00s setup    tests/infer/test_jit.py::test_dirichlet_bernoulli[TraceEnum_ELBO-True]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoDelta]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_guide_list_x]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[3-(4,)]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-18]
0.00s setup    tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-2-expand_shape3]
0.00s setup    tests/contrib/oed/test_xexpx.py::test_xexpx[argument0-output0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{}-7-12]
0.00s call     tests/distributions/test_spanning_tree.py::test_enumerate_support[1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixj->ij]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-wi,wxi,xyi,yzi->i]
0.00s setup    tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-subsample-nonreparam]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide[AutoDelta-init_to_median]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,wxi,xyi,yzi->i]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[1-2-(3, 2)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3-1.0]
0.00s setup    tests/infer/test_jit.py::test_dirichlet_bernoulli[TraceEnum_ELBO-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-20-False]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoNormal]
0.00s setup    tests/distributions/test_rejector.py::test_rejector[0.5-1.0]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype2-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_distribution_validate_args[True-Normal-args0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-2-1-7]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_2[batch-10]
0.00s setup    tests/infer/test_enum.py::test_elbo_bern[sequential-loss_and_grads-1]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoDiagonalNormal-0.0001]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-0-2]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[3-0]
0.00s setup    tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape2]
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[2-None]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3-0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99-0.0]
0.00s setup    tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-flat-2]
0.00s setup    tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape0]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[3-3-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7--0.5]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1--0.5]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_2[single-10]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[1-sequential-parallel]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Geometric-list-sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-tuple-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-2-expand_shape3]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-4-uniform]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(3, 2)-(6, 5)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[map-a->-]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99-1.0]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=False-NUTS]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-flat-2]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-dense-3]
0.00s setup    tests/infer/test_discrete.py::test_hmm_smoke[map-100]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape0]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-3-uniform]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-i->i]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-mask2]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-dense-2]
0.00s setup    tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape1]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype0-size0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixyj,iyzj->ij]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.00s setup    tests/infer/reparam/test_studentt.py::test_moments[(4,)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4-1.0]
0.00s setup    tests/infer/reparam/test_neutra.py::test_reparam_log_joint[neals_funnel-kwargs0]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=False-HMC]
0.00s setup    tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-True-tuple]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-100]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.05]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-ij->ij]
0.00s setup    tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape1-split_gelman_rubin]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(3, 2)-()]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[a,ai,abij->bij-ij]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-flat-1]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide[AutoNormal-None]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-auto_class7]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class9]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal_mvn[3-(2, 3)]
0.00s setup    tests/distributions/test_conjugate.py::test_gamma_poisson_log_prob[shape0]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[3-0.1]
0.00s setup    tests/infer/reparam/test_studentt.py::test_moments[(2, 3)]
0.00s setup    tests/infer/test_jit.py::test_svi[JitTraceMeanField_ELBO-1]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-flat-3]
0.00s setup    tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape2]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-3]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[Uniform-Size-sample_shape0]
0.00s setup    tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[16]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,wx,xy,yz->]
0.00s setup    tests/ops/test_gaussian.py::test_marginalize[3-1-()]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[Trace_ELBO-1]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceTMC_ELBO-subsample]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoDelta]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-10]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LogNormal-list-sample_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_expand_by[LogNormal-Size-sample_shape2]
0.00s setup    tests/poutine/test_properties.py::test_commutes[trace-condition-bernoulli_normal_model]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[2-size2]
0.00s setup    tests/infer/test_jit.py::test_svi_enum[2-parallel-sequential]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_2[masked-1]
0.00s setup    tests/infer/test_enum.py::test_elbo_normal[parallel-differentiable_loss]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-1.0]
0.00s call     tests/infer/test_valid_models.py::test_plate_broadcast_error[TraceGraph_ELBO]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[2-3-(4,)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7-0.0]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op0-JIT=False]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=True-NUTS]
0.00s setup    tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-1-1]
0.00s setup    tests/distributions/test_conjugate.py::test_log_prob_support[dist0-None]
0.00s setup    tests/distributions/test_categorical.py::test_batch_log_dims[dim=2-probs0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-i,wi->i]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(3, 2)-(7,)]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0-1.0]
0.00s setup    tests/infer/test_jit.py::test_dirichlet_bernoulli[JitTraceEnum_ELBO-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_iplate[2-parallel-None]
0.00s setup    tests/infer/test_discrete.py::test_plate_smoke[2-sample-elbo]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-14]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceTMC_ELBO-subsample]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-3-2-1]
0.00s setup    tests/ops/test_welford.py::test_welford_diagonal[1000-1]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[4-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(3, 2)-()]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(4,)-(6, 5)]
0.00s setup    tests/nn/test_module.py::test_constraints[shape24-constraint_24]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-1000]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[2-size0]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[2-()]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-2-1-1]
0.00s setup    tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape0-_cummin]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_reshape[1-(5, 6)-(5, 3, 2)]
0.00s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_reparameterized_three_layers
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01--0.5]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-ai->a-i]
0.00s setup    tests/nn/test_module.py::test_constraints[shape15-constraint_15]
0.00s setup    tests/infer/test_svgd.py::test_conjugate[RBFSteinKernel-multivariate]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoMultivariateNormal]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-()-3]
0.00s setup    tests/nn/test_module.py::test_constraints[shape20-constraint_20]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0-0.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01-0.0]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0-1.0]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_guide_list_x]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[3-2]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-10.0]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s call     tests/poutine/test_poutines.py::BlockHandlerTests::test_block_expose_fn
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7-1.0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_ok[TraceTMC_ELBO-subsample]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-False-sample_shapes1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-ji,iwj->ji]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[1-10.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_1[batch-1]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_mvn_shape[2-(4,)]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype2-event_shape0]
0.00s setup    tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape0-gelman_rubin]
0.00s setup    tests/ops/test_ssm_gp.py::test_matern_kernel[1.5-1]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[3-3]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-0.1]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape2-event_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-1-1-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[3-size2]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[map-a,abi,bcij,bdik->-ijk]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[bi,ia->,i,ia,ib,iab-i]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0--0.5]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-1-1]
0.00s setup    tests/poutine/test_poutines.py::test_replay_enumerate_poutine[1--3]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[2-1-(4,)]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal-0.0001]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[3-4]
0.00s setup    tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape4]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.3]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,wxi,xi->i]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[map-ai,ai->i-i]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01--1.0]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoDelta-Trace_ELBO]
0.00s setup    tests/contrib/test_minipyro.py::test_plate_ok[minipyro-jit]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[sample-a->-]
0.00s setup    tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-dense-2]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-auto_class6]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.01--1.0]
0.00s setup    tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape0-autocorrelation]
0.00s setup    tests/ops/test_gaussian.py::test_affine_normal[1-3-(4,)]
0.00s setup    tests/infer/reparam/test_neutra.py::test_neals_funnel_smoke[True]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-1000]
0.00s setup    tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape1-autocorrelation]
0.00s setup    tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-()-()]
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_dbn[True-False]
0.00s setup    tests/infer/test_discrete.py::test_distribution_masked[sample-elbo]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[3-(2, 3)]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype0-event_shape1]
0.00s setup    tests/ops/test_stats.py::test_fit_generalized_pareto[1.3-0.5]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoDiagonalNormal]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype1-event_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_distribution_validate_args[True-Exponential-args2]
0.00s setup    tests/distributions/test_conjugate.py::test_mean[dist5]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-<lambda>]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-1-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-3-uniform]
0.00s setup    tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape1-_cummin]
0.00s setup    tests/infer/test_autoguide.py::test_predictive[AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[2-3]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape0-sample_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-1-1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-1-sparse]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1398]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixj->ji]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,xi->i]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,ixj->ij]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-ij,iwj->ij]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[ai->i-i]
0.00s setup    tests/infer/test_predictive.py::test_deterministic[event_shape0-False]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-1.5]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-1-sparse]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-3-random]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-ij->ji]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0-0.0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.01]
0.00s call     tests/nn/test_module.py::test_constraints[shape0-constraint_0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-1-7]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_distribution[1-()]
0.00s setup    tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0--1.0]
0.00s setup    tests/infer/test_jit.py::test_infer_discrete[sample-1]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[2-2]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-5-uniform]
0.00s setup    tests/infer/test_valid_models.py::test_dim_allocation_ok[Trace_ELBO-False]
0.00s setup    tests/distributions/test_rejector.py::test_rejector[1.0-2.0]
0.00s setup    tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape0-split_gelman_rubin]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_logistic_regression[True-JIT=True]
0.00s setup    tests/infer/test_valid_models.py::test_plate_ok[TraceTMC_ELBO-full]
0.00s setup    tests/infer/test_jit.py::test_svi[JitTraceEnum_ELBO-10]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[3-5]
0.00s setup    tests/distributions/test_one_hot_categorical.py::test_batch_log_dims[dim=1-probs0]
0.00s setup    tests/distributions/test_distributions.py::test_distribution_validate_args[False-Exponential-args2]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-0.1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-5-random]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-4-uniform]
0.00s setup    tests/ops/test_special.py::test_log_beta_stirling[1.0]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[2-0.1]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[map-ai,abij,acik->i-ijk]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs1-dims1-1]
0.00s setup    tests/distributions/test_distributions.py::test_distribution_validate_args[True-Gamma-args1]
0.00s call     tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[2-True]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-4-uniform]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-2-random]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-1-random]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[dct--3]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[3-1.0]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(6),:,z(5,1)]-(5, 6, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_add[2-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-1-1-1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-1-sparse]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-2-sparse]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs14-dims14-3]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_single[OneHotCategorical-prior4]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-2-1-1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-5-sparse]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_distribution[2-(3, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-bi,abi->b-i]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_guide_module_callable]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-0.5]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-1.0]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-5-uniform]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-5-sparse]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoDelta]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape0]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-10.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.01]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-5-random]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-1-uniform]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-wi,wxi,xi->i]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.00s setup    tests/infer/test_enum.py::test_elbo_normal[parallel-loss]
0.00s setup    tests/distributions/test_von_mises.py::test_log_prob_normalized[10.0]
0.00s setup    tests/distributions/test_reshape.py::test_idempotent[0-0]
0.00s setup    tests/distributions/test_ig.py::test_sample[2.5-3.3]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-4-random]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-4-sparse]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-5-sparse]
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[2-1.0]
0.00s setup    tests/ops/test_contract.py::test_adjoint_shape[map-a,abi->-i]
0.00s setup    tests/distributions/test_stable.py::test_normal[1-0]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[3-1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape6]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[2-(4,)]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-1-expand_shape2]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[i->,i-i]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_logistic_regression[False-JIT=False]
0.00s setup    tests/distributions/test_stable.py::test_normal[2-21]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[False-True]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype1-event_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_distribution_validate_args[False-Normal-args0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_distribution_validate_args[False-Gamma-args1]
0.00s call     tests/contrib/tracking/test_hashing.py::test_aps_try_add[0.1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-2-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-3-sparse]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-2-uniform]
0.00s setup    tests/infer/test_jit.py::test_bernoulli_enumerate[shape2-False]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class11]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-2.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.000001-1.0]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal_mvn[3-(4,)]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-2]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-2-uniform]
0.00s setup    tests/ops/test_contract.py::test_ubersum[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-<lambda>]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-True]
0.00s setup    tests/distributions/test_conjugate.py::test_gamma_poisson_log_prob[shape2]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-5-random]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-3-random]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,ixj->ji]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs15-dims15-3]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs7-dims7-3]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoLaplaceApproximation-Trace_ELBO]
0.00s setup    tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-2-1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-1-random]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-5-uniform]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-2-uniform]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-5-random]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixj->ij]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[False-A4]
0.00s setup    tests/ops/test_contract.py::test_ubersum[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.00s setup    tests/distributions/test_categorical.py::test_support_dims[dim=2-probs0]
0.00s setup    tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[pyro-jit]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape2-event_shape2]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-,w->]
0.00s setup    tests/infer/reparam/test_stable.py::test_stable[LatentStableReparam-()]
0.00s setup    tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape2]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-1000]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.01]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoIAFNormal]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-1-1-1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.25]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-3-1]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoLaplaceApproximation]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-3-random]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-4-random]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwxj,ixyj,iyzj,izwj->ji]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-False]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceEnum_ELBO-parallel]
0.00s setup    tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape3]
0.00s setup    tests/infer/test_enum.py::test_mixture_of_diag_normals[MixtureOfDiagNormals-scale0]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_class9]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape2]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape4-event_shape4]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-3-sparse]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-True-list]
0.00s setup    tests/poutine/test_mapdata.py::test_cond_indep_stack[plate-20]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_predictive[True-100]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-mask2]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal[(2, 3)-()]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-0.1]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-10.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-1-1-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-4-uniform]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-3-sparse]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,w->]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_logistic_regression[False-JIT=True]
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_dbn[True-True]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-1]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class12]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoDelta-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_ig.py::test_sample[3.0-4.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-2-1]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[bi,ia->,i,ia,ib,iab-i]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-EnergyDistance_noprior]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_conjugate.py::test_log_prob_support[dist2-values2]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-1000]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape1-event_shape1]
0.00s setup    tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape2-_cummin]
0.00s setup    tests/ops/test_contract.py::test_ubersum[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[TraceMeanField_ELBO-1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-0.9]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-1000]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[1.0-10.0]
0.00s setup    tests/contrib/test_util.py::test_rvv[a0-b0]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-4-random]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[7-size0]
0.00s setup    tests/infer/test_jit.py::test_categorical_enumerate[shape2-True]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class12]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-1]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape1-event_shape1]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs16-dims16-3]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs13-dims13-4]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_ok[TraceTMC_ELBO-subsample]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape4]
0.00s call     tests/ops/test_contract.py::test_ubersum[->-]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-1-7]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.00s setup    tests/nn/test_module.py::test_constraints[shape13-constraint_13]
0.00s setup    tests/infer/test_predictive.py::test_deterministic[event_shape0-True]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoDiagonalNormal]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_guide_module_callable]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype1-event_shape0]
0.00s setup    tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype0-event_shape0]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoMultivariateNormal]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-2-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,ixj->ji]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-->]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gaussian_hmm[30]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-mask3]
0.00s setup    tests/infer/test_predictive.py::test_deterministic[event_shape1-True]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape3-event_shape3]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-2-sparse]
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[2-0.1]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[ab->,a,b,ab,ba-]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[1-(2, 3)]
0.00s setup    tests/distributions/test_omt_mvn.py::test_log_prob[OMTMultivariateNormal]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[10.0-1.0]
0.00s call     tests/contrib/autoname/test_named.py::test_nested
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-1-sparse]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-1-random]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[1-size2]
0.00s setup    tests/nn/test_module.py::test_constraints[shape22-constraint_22]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape0]
0.00s setup    tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-100]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape1-event_shape1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-ji,iwj->ji]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->a-ij]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoNormal]
0.00s setup    tests/distributions/test_conjugate.py::test_gamma_poisson_log_prob[shape1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[10.0-1]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape3-event_shape3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-3-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-4-random]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[i->-i]
0.00s setup    tests/nn/test_module.py::test_constraints[shape10-constraint_10]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class10]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal_mvn[2-(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-2-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[3-size0]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-mask3]
0.00s setup    tests/infer/test_jit.py::test_infer_discrete[sample-2]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoDiagonalNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-1.0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape6]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.03]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-2-7]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.00s setup    tests/infer/test_autoguide.py::test_predictive[auto_class7]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoMultivariateNormal-TraceEnum_ELBO]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Uniform-True-initial_shape0-proposed_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-3-2-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1426]
0.00s setup    tests/nn/test_module.py::test_constraints[shape9-constraint_9]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-True]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoDiagonalNormal]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape0]
0.00s setup    tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[2-True]
0.00s setup    tests/nn/test_module.py::test_constraints[shape14-constraint_14]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceEnum_ELBO-full]
0.00s setup    tests/infer/test_svgd.py::test_conjugate[RBFSteinKernel-univariate]
0.00s setup    tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape6]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-1]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[a->,a-]
0.00s setup    tests/ops/test_contract.py::test_partition_terms[inputs5-dims5-4]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_model_error_stray_batch_dims[NUTS-kwargs1]
0.00s setup    tests/infer/test_autoguide.py::test_predictive[AutoLaplaceApproximation]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoNormal-TraceGraph_ELBO]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-0.01]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-4]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[5.0-5.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-2-7]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-wxi,xyi,yzi,zwi->i]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoDiagonalNormal]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-1.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.01]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-1.0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwxj,ixyj,iyzj,izwj->ji]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,wxi,xyi,yzi->i]
0.00s setup    tests/ops/test_ssm_gp.py::test_matern_kernel[0.5-3]
0.00s setup    tests/distributions/test_stable.py::test_normal[1-1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_categorical.py::test_sample_dims[dim=3-probs0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.03]
0.00s call     tests/ops/test_stats.py::test_split_gelman_rubin_agree_with_gelman_rubin
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-3-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwj->ji]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.00s setup    tests/nn/test_module.py::test_constraints[shape4-constraint_4]
0.00s setup    tests/infer/test_jit.py::test_bernoulli_enumerate[shape0-True]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1005]
0.00s setup    tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape0-_pi]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-TraceTMC_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-TraceGraph_ELBO]
0.00s setup    tests/infer/test_jit.py::test_bernoulli_enumerate[shape1-True]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class9]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoIAFNormal-TraceEnum_ELBO]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.1]
0.00s call     tests/poutine/test_poutines.py::ConditionHandlerTests::test_condition
0.00s call     tests/contrib/tracking/test_hashing.py::test_aps_try_add[100]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[1-size1]
0.00s setup    tests/ops/test_contract.py::test_ubersum[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.00s setup    tests/infer/test_jit.py::test_categorical_enumerate[shape2-False]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_class8]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoDelta]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-1.0]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[3.0-0.1]
0.00s setup    tests/distributions/test_ig.py::test_sample[3.0-3.3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.05]
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[3-0.1]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[,i,i->,i-i]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.00s setup    tests/nn/test_module.py::test_constraints[shape21-constraint_21]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op1-JIT=False]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.75]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[1-0]
0.00s setup    tests/distributions/test_one_hot_categorical.py::test_sample_dims[dim=2-probs0]
0.00s call     tests/params/test_module.py::test_module_nn[outer]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[1-1.0]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),0]-(4,)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_marginal[a->-]
0.00s setup    tests/infer/test_jit.py::test_svi[JitTraceGraph_ELBO-10]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoDiagonalNormal-0.1]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoMultivariateNormal-Trace_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-0.9]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.03]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-False]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[Trace_ELBO-sequential]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class10]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-0.1]
0.00s call     tests/distributions/test_mask.py::test_mask_noop[None]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-->]
0.00s setup    tests/nn/test_module.py::test_constraints[shape8-constraint_8]
0.00s setup    tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape3]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_guide_callable]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-3-7]
0.00s setup    tests/ops/test_ssm_gp.py::test_matern_kernel[2.5-2]
0.00s setup    tests/ops/test_contract.py::test_ubersum[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[-1.0--1.0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixyj,iyzj->ij]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[abi,b->,b,ai,abi-i]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[,i,i->,i-i]
0.00s setup    tests/infer/test_jit.py::test_infer_discrete[map-2]
0.00s setup    tests/infer/test_jit.py::test_svi[TraceGraph_ELBO-10]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoMultivariateNormal-0.1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-1-7]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-True]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[Trace_ELBO-subsample]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-TraceTMC_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_predictive[AutoNormal]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class9]
0.00s setup    tests/distributions/test_reshape.py::test_idempotent[2-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[5.0-5.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-3-1]
0.00s setup    tests/poutine/test_mapdata.py::test_replay[nested_iplate-20]
0.00s setup    tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape0-_hpdi]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-TraceTMC_ELBO]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-1.0]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape2-event_shape2]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-i,wi->i]
0.00s setup    tests/ops/test_tensor_utils.py::test_precision_to_scale_tril[batch_shape1-event_shape1]
0.00s setup    tests/infer/test_valid_models.py::test_set_has_rsample_ok[Trace_ELBO-True]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_beta[4.0-4.0]
0.00s setup    tests/poutine/test_poutines.py::test_replay_enumerate_poutine[0--3]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[TraceMeanField_ELBO-0.1]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[3.0-0.9]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[0-2]
0.00s setup    tests/distributions/test_mvt.py::test_log_prob_normalization[1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-w,x->]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[3-10.0]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_prior-1]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-TraceTMC_ELBO]
0.00s setup    tests/infer/test_jit.py::test_cond_indep_equality[x1-y1]
0.00s setup    tests/infer/test_discrete.py::test_distribution_3[sample]
0.00s setup    tests/distributions/test_stable.py::test_shape[batch_shape1-sample_shape2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-3-7]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-wx,xy,yz,zw->]
0.00s setup    tests/ops/test_contract.py::test_ubersum[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_predictive[False-200]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceEnum_ELBO-full]
0.00s setup    tests/infer/test_predictive.py::test_deterministic[event_shape1-False]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class11]
0.00s setup    tests/distributions/test_categorical.py::test_support_dims[dim=1-probs0]
0.00s setup    tests/poutine/test_poutines.py::test_replay_enumerate_poutine[2--1]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_reparam_stable[NUTS-kwargs1]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gamma_normal[False-JIT=False]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class9]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape4]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape6]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape2-sample_shape2]
0.00s call     tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape0-value_shape0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwj->ij]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[3.0-0.25]
0.00s setup    tests/infer/test_discrete.py::test_distribution_1[sample-elbo]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoDiagonalNormal-1e-08]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwxj,ixyj,iyzj,izwj->ij]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_distribution[3-()]
0.00s setup    tests/ops/test_gaussian.py::test_add[3-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceEnum_ELBO-None]
0.00s setup    tests/infer/test_jit.py::test_beta_bernoulli[TraceEnum_ELBO-True]
0.00s setup    tests/infer/test_autoguide.py::test_predictive[AutoMultivariateNormal]
0.00s setup    tests/distributions/test_stable.py::test_normal[1-20]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_beta[4.0-0.5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape3]
0.00s setup    tests/infer/test_jit.py::test_bernoulli_enumerate[shape0-False]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape3]
0.00s setup    tests/distributions/test_one_hot_categorical.py::test_support_dims[dim=3-probs0]
0.00s setup    tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[minipyro-py]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[idct-2]
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[3-1.0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceGraph_ELBO-subsample]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-EnergyDistance_prior]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[1.0-0.9]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[1-()]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.3]
0.00s call     tests/distributions/test_one_hot_categorical.py::test_sample_dims[dim=1-probs0]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape3-sample_shape0]
0.00s setup    tests/poutine/test_mapdata.py::test_cond_indep_stack[iplate-20]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixj->ji]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[1-event_shape2]
0.00s setup    tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape2-_hpdi]
0.00s setup    tests/ops/test_ssm_gp.py::test_matern_kernel[2.5-3]
0.00s setup    tests/ops/test_indexing.py::test_hmm_example[-4--5]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[,i,i,ia,ia->,i,ia-i]
0.00s setup    tests/ops/test_contract.py::test_ubersum[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoNormal]
0.00s setup    tests/distributions/test_conjugate_update.py::test_beta_binomial[()-(4,)]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceEnum_ELBO-subsample]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape3-event_shape3]
0.00s call     tests/contrib/tracking/test_hashing.py::test_aps_try_add[1]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-1]
0.00s setup    tests/distributions/test_stable.py::test_normal[2--1]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.8-0.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-1.0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-ij->ij]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_ok[TraceEnum_ELBO-full]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_guide_module_callable]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class12]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoDelta-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-2-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwj->ij]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoDiagonalNormal]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-auto_class7]
0.00s setup    tests/distributions/test_stable.py::test_normal[1-21]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape5]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-True]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.01]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size2-2]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[ab->,a,b,ab,ba-]
0.00s setup    tests/infer/test_autoguide.py::test_predictive[AutoDiagonalNormal]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.01]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_noprior-0.1]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[TraceMeanField_ELBO-scale2]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[None-1]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoDelta]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_class9]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-0.1]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[0-3]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal_mvn[2-(4,)]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-4]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s setup    tests/ops/test_contract.py::test_ubersum[->-]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.99-0.0]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoLowRankMultivariateNormal]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape3-event_shape3]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape3-event_shape3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-3-2-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1228]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[1-None]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[Trace_ELBO-scale2]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-EnergyDistance_prior]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.3]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceEnum_ELBO-sequential]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-2.0]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[3-()]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[2-0]
0.00s setup    tests/distributions/test_one_hot_categorical.py::test_sample_dims[dim=3-probs0]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[False-p_mask3]
0.00s setup    tests/distributions/test_conjugate_update.py::test_gamma_poisson[()-(4,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-1-1]
0.00s setup    tests/contrib/test_minipyro.py::test_local_param_ok[minipyro-py]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-1-7]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,ixj->ij]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[dct-0]
0.00s setup    tests/nn/test_module.py::test_constraints[shape0-constraint_0]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-5]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_beta[1.0-1.0]
0.00s setup    tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[pyro-jit]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,xi->i]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[54]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[10-event_shape1]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[10-event_shape0]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[2-10.0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceGraph_ELBO-full]
0.00s setup    tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceEnum_ELBO-True]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_class8]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_stable.py::test_normal[2-0]
0.00s setup    tests/poutine/test_poutines.py::test_enumerate_poutine[0--2]
0.00s setup    tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape2-gelman_rubin]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-mask4]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[sequential-1]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoNormal]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoLowRankMultivariateNormal-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoDiagonalNormal-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_stable.py::test_normal[2-1]
0.00s setup    tests/distributions/test_mvt.py::test_rsample[2-3.9]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[block-trivial_model]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5,1),:]-(5, 2, 4)]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[->-]
0.00s setup    tests/infer/test_discrete.py::test_distribution_3[sample-elbo]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-0.9]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-2.0]
0.00s setup    tests/distributions/test_stable.py::test_normal[1--1]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_beta[1.0-0.5]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal[()-(4,)]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape4-event_shape4]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gamma_normal[False-JIT=True]
0.00s setup    tests/infer/test_discrete.py::test_distribution_1[sample]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[1.0-1.0]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape2-sample_shape0]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-False]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[True-p_mask2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-1.5]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[2-size1]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[2-event_shape2]
0.00s setup    tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape1-_quantile]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_distribution[3-(3, 2)]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-10.0]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[2-(2, 3)]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape1]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal_mvn[3-()]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.1]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_predictive[True-None]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-TraceGraph_ELBO]
0.00s setup    tests/infer/test_jit.py::test_svi[TraceEnum_ELBO-10]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-auto_class7]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoDiagonalNormal-TraceGraph_ELBO]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[1.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[5.0-5]
0.00s call     tests/poutine/test_mapdata.py::test_cond_indep_stack[plate-20]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-1-1]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[block-bernoulli_normal_model]
0.00s setup    tests/ops/test_gaussian.py::test_add[1-(4,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[i->,i-i]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-EnergyDistance_prior]
0.00s setup    tests/infer/test_jit.py::test_svi[JitTraceGraph_ELBO-1]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-4]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoMultivariateNormal-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[2-2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_categorical[9]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-1.0]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Laplace]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges735]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[,i,ia->,i,ia-i]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op1-JIT=True]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-TraceEnum_ELBO]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[4-2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-1.0]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-10]
0.00s setup    tests/distributions/test_rejector.py::test_shape_augmented_beta[0.5-4.0]
0.00s setup    tests/distributions/test_lkj.py::test_log_prob_eta1[4]
0.00s call     tests/poutine/test_poutines.py::ConditionHandlerTests::test_stack_overwrite_behavior
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[1-None]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_predictive[False-None]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[1-(5,)]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[0-1]
0.00s call     tests/distributions/test_shapes.py::test_one_hot_categorical_shape
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[3-event_shape1]
0.00s setup    tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape1-autocovariance]
0.00s setup    tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape0-_quantile]
0.00s setup    tests/ops/test_special.py::test_log_binomial_stirling[0.11]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-mask2]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal_100-0.0001]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-1.1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape1]
0.00s call     tests/poutine/test_poutines.py::test_arg_kwarg_error
0.00s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_reparameterized_N_is_3
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[Trace_ELBO-parallel]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoNormal]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoDiagonalNormal-Trace_ELBO]
0.00s setup    tests/distributions/test_rejector.py::test_rejector[0.25-0.5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape4]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-0.01]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-1-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-1.5]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape0]
0.00s setup    tests/poutine/test_poutines.py::test_enumerate_poutine[2--1]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[3-None]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-()-()]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[,i,i->,i-i]
0.00s setup    tests/infer/test_jit.py::test_svi[Trace_ELBO-10]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-10.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-0.9]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape5]
0.00s setup    tests/infer/reparam/test_stable.py::test_subsample_smoke[StableReparam-subsample]
0.00s setup    tests/infer/test_discrete.py::test_distribution_masked[map]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoIAFNormal-Trace_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-1.0]
0.00s setup    tests/distributions/test_one_hot_categorical.py::test_sample_dims[dim=1-probs0]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-1.0]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoLowRankMultivariateNormal]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceTMC_ELBO-subsample]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size0-1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape3]
0.00s setup    tests/distributions/test_von_mises.py::test_log_prob_normalized[0.03]
0.00s setup    tests/distributions/test_stable.py::test_normal[0.5--1]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-p_mask2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-1-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-wx,xy,yz,zw->]
0.00s setup    tests/infer/test_valid_models.py::test_plate_ok[TraceGraph_ELBO-subsample]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide_2[AutoNormal-independent]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape0-sample_shape0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[0.0-1.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.03]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.03]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[1.0-10]
0.00s call     tests/poutine/test_poutines.py::BlockHandlerTests::test_block_hide_fn
0.00s call     tests/distributions/test_one_hot_categorical.py::test_sample_dims[dim=3-probs0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-2-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj->ji]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[idct-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[dct--1]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[2-event_shape1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_add[1-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceGraph_ELBO-subsample]
0.00s setup    tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceGraph_ELBO-True]
0.00s setup    tests/infer/test_enum.py::test_elbo_rsvi[None]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape4]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.03]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.1]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Cauchy]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[4-event_shape0]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-mask3]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-mask2]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[parallel-inf]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape0]
0.00s setup    tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[()-()]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.01]
0.00s call     tests/distributions/test_lkj.py::test_constraint[value_shape0]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[ai->-i]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-True]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-TraceEnum_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide_2[AutoNormal-dependent]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixj->ij]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwj->ij]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5),0]-(5,)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-TraceGraph_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-TraceEnum_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-TraceEnum_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class11]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoNormal-Trace_ELBO]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-False]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.01]
0.00s setup    tests/contrib/test_minipyro.py::test_local_param_ok[pyro-jit]
0.00s call     tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[2-False]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op0-JIT=True]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_guide_callable]
0.00s setup    tests/distributions/test_stable.py::test_normal[2-20]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-3]
0.00s setup    tests/distributions/test_conjugate_update.py::test_beta_binomial[(3, 2)-()]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoLaplaceApproximation]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape4-event_shape4]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[idct-0]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:,z(4)]-(4, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceGraph_ELBO-None]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-TraceEnum_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-Trace_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_plate_ok[TraceEnum_ELBO-full]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape1]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes10]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape2]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[dct-3]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceGraph_ELBO-parallel]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-TraceEnum_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceTMC_ELBO-subsample]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.25]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-1.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.3]
0.00s call     tests/poutine/test_mapdata.py::test_cond_indep_stack[plate-5]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Geometric]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-2-1-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-w->]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-scale3]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_stable.py::test_normal[0.5-0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape6]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[-1.0-0.0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.3]
0.00s call     tests/infer/test_valid_models.py::test_plate_stack_sizes[sizes1]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceEnum_ELBO-subsample]
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[3-None]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-mask4]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_ok[TraceGraph_ELBO-full]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoLowRankMultivariateNormal-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-0.1]
0.00s setup    tests/distributions/test_stable.py::test_variance[1.7-1.5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape1]
0.00s setup    tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(4,)-(3, 2)]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-1-1]
0.00s setup    tests/poutine/test_mapdata.py::test_cond_indep_stack[plate-5]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-w,w->]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-TraceGraph_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_guide_module_callable]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-0.5]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[0-1]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype1-size2]
0.00s setup    tests/contrib/oed/test_xexpx.py::test_xexpx[argument2-output2]
0.00s setup    tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[pyro-py]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Exponential]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-2-1-7]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),0,0]-(5,)]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-0.1]
0.00s setup    tests/infer/test_jit.py::test_svi[TraceGraph_ELBO-1]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_guide_callable]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoDelta]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoIAFNormal-TraceGraph_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoLowRankMultivariateNormal-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape3]
0.00s setup    tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(4,)-()]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape2-event_shape2]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_model_error_enum_dim_clash[HMC-kwargs0]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_logistic_regression[True-JIT=False]
0.00s setup    tests/infer/test_discrete.py::test_distribution_3[map]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_guide_callable]
0.00s setup    tests/distributions/test_unit.py::test_expand[batch_shape2-sample_shape1]
0.00s setup    tests/distributions/test_categorical.py::test_batch_log_dims[dim=3-probs0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-4]
0.00s setup    tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_rsample[False-relu]
0.00s setup    tests/poutine/test_poutines.py::test_enumerate_poutine[2--2]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-TraceGraph_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceTMC_ELBO-False]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_guide_list_x]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-10.0]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[2-(5,)]
0.00s setup    tests/distributions/test_stable.py::test_sample[2.0-0.5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape6]
0.00s setup    tests/distributions/test_one_hot_categorical.py::test_support_dims[dim=1-probs0]
0.00s setup    tests/distributions/test_conjugate_update.py::test_gamma_poisson[()-()]
0.00s setup    tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[()-(4,)]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoMultivariateNormal]
0.00s call     tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-0-expand_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-2-7]
0.00s setup    tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape2-effective_sample_size]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_reparam_stable[HMC-kwargs0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_ok[Trace_ELBO-subsample]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-TraceGraph_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-30]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[1-0]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[2-3]
0.00s setup    tests/distributions/test_conjugate.py::test_log_prob_support[dist1-None]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-1.5]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-1.0]
0.00s setup    tests/contrib/test_minipyro.py::test_constraints[minipyro-py]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),:,z(5)]-(5, 3)]
0.00s setup    tests/ops/test_contract.py::test_ubersum[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceTMC_ELBO-full]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_single[Categorical-prior2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-1.1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal[(4,)-(4,)]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape2-event_shape2]
0.00s setup    tests/poutine/test_poutines.py::test_replay_enumerate_poutine[1--2]
0.00s setup    tests/poutine/test_poutines.py::test_replay_enumerate_poutine[0--1]
0.00s setup    tests/poutine/test_mapdata.py::test_replay[plate-20]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-w,x->]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[2-False]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-True]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[parallel-1]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-5]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[10.0-1.0]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),0,:]-(5, 4)]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_ok[Trace_ELBO-subsample]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[3-(2, 3)]
0.00s setup    tests/distributions/test_stable.py::test_normal[0.5-20]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[2-4]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.03]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape1-event_shape1]
0.00s call     tests/contrib/tracking/test_hashing.py::test_aps_try_add[10]
0.00s setup    tests/nn/test_module.py::test_decorator[2-AttributeModel]
0.00s setup    tests/infer/reparam/test_stable.py::test_subsample_smoke[LatentStableReparam-full]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.000001-0.5]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[1-4]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape3]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-3-1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[1.0-10.0]
0.00s call     tests/contrib/randomvariable/test_random_variable.py::test_chaining
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-3-1]
0.00s setup    tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape0-autocovariance]
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[3-10.0]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-scale2]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoNormal]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class9]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape0]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[False-False]
0.00s setup    tests/distributions/test_conjugate_update.py::test_gamma_poisson[(3, 2)-()]
0.00s setup    tests/distributions/test_conjugate_update.py::test_gamma_poisson[()-(3, 2)]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-3]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-ij,iwj->ij]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),:,:]-(5, 3, 4)]
0.00s setup    tests/infer/reparam/test_stable.py::test_subsample_smoke[StableReparam-full]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_predictive[True-200]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-full-reparam]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoLaplaceApproximation-TraceGraph_ELBO]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape4]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape6]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-1-7]
0.00s setup    tests/poutine/test_poutines.py::test_replay_enumerate_poutine[2--2]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-w,w->]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-False]
0.00s setup    tests/infer/test_valid_models.py::test_vectorized_num_particles[TraceEnum_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.25]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[1-2]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape3]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-p_mask2]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal[(4,)-()]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_mvn_shape[1-()]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-1.5]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape1-sample_shape0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixyj,iyzj->ji]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-w,wx,x->]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5),:]-(5, 2, 4)]
0.00s setup    tests/ops/test_contract.py::test_ubersum[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceGraph_ELBO-full]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoDelta]
0.00s setup    tests/poutine/test_mapdata.py::test_replay[plate-5]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[8-size1]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_unit.py::test_expand[batch_shape1-sample_shape1]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_bernoulli_shapes[probs0]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[True-True]
0.00s setup    tests/distributions/test_conjugate_update.py::test_beta_binomial[(4,)-(4,)]
0.00s setup    tests/poutine/test_poutines.py::test_enumerate_poutine[1--2]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixyj,iyzj->ji]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_distribution[1-(4,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.00s setup    tests/nn/test_module.py::test_decorator[1-AttributeModel]
0.00s setup    tests/infer/test_jit.py::test_infer_discrete[sample-10]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-auto_class6]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.25]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape6]
0.00s setup    tests/poutine/test_mapdata.py::test_cond_indep_stack[iplate-5]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-0.1]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-TraceTMC_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-10.0]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-3-1]
0.00s setup    tests/poutine/test_poutines.py::test_enumerate_poutine[1--3]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[dct-2]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5,1),:,z(6)]-(5, 6, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_add[1-(3, 2)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_param_ok[Trace_ELBO-subsample]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_single[OneHotCategorical-prior5]
0.00s setup    tests/infer/test_autoguide.py::test_factor[AutoLaplaceApproximation-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape3]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape3]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal[(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_conjugate.py::test_log_prob_support[dist3-values3]
0.00s setup    tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[minipyro-py]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),0,z(5)]-(5,)]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-1.1]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.8--0.5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape4]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape3]
0.00s setup    tests/distributions/test_conjugate_update.py::test_beta_binomial[(3, 2)-(3, 2)]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,ixj->ij]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-,w->]
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[2-10.0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_add[1-(4,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[,i,ia->,i,ia-i]
0.00s setup    tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceTMC_ELBO-True]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape1]
0.00s setup    tests/distributions/test_von_mises.py::test_sample[0.01--1.5707963267948966]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape5]
0.00s setup    tests/distributions/test_conjugate_update.py::test_beta_binomial[(4,)-(3, 2)]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.03]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Chi2]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-True]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class9]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape4]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-100.0]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape4-event_shape4]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[LowRankMultivariateNormal]
0.00s setup    tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[0-False]
0.00s setup    tests/poutine/test_poutines.py::test_replay_enumerate_poutine[2--3]
0.00s setup    tests/ops/test_contract.py::test_ubersum[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-TraceGraph_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceEnum_ELBO-full]
0.00s setup    tests/infer/test_jit.py::test_svi[Trace_ELBO-1]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_single[OneHotCategorical-prior6]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-1.1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-10.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.3--0.5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape3]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape6]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-True]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-100.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-1]
0.00s setup    tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_log_prob[False-relu]
0.00s setup    tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[minipyro-jit]
0.00s setup    tests/ops/test_tensor_utils.py::test_repeated_matmul[1-size0]
0.00s setup    tests/ops/test_contract.py::test_contract_to_tensor[example7]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-scale2]
0.00s setup    tests/infer/test_discrete.py::test_distribution_2[sample-elbo]
0.00s setup    tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal_100-0.1]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal[()-(2, 3)]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Gamma]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwxj,ixyj,iyzj,izwj->ij]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,0,z(5)]-(5, 2)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2,1)]-(2, 3)]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-TraceGraph_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[0.1-0.9]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-0.5]
0.00s setup    tests/distributions/test_stable.py::test_shape[batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[1-3]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape1]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[1.0-1.0]
0.00s setup    tests/poutine/test_poutines.py::test_enumerate_poutine[1--1]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.00s setup    tests/infer/reparam/test_neutra.py::test_reparam_log_joint[dirichlet_categorical-kwargs1]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceTMC_ELBO-full]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoDelta]
0.00s setup    tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoDiagonalNormal]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class11]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape2]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes7]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[2-0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape2]
0.00s setup    tests/contrib/test_minipyro.py::test_plate_ok[pyro-jit]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape3-event_shape3]
0.00s setup    tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape1-_hpdi]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.00s setup    tests/infer/reparam/test_stable.py::test_subsample_smoke[SymmetricStableReparam-full]
0.00s setup    tests/infer/test_predictive.py::test_posterior_predictive_svi_one_hot
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-2.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-0.1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-1.1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape2]
0.00s setup    tests/distributions/test_stable.py::test_shape[batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_stable.py::test_shape[batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape4]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_conjugate_update.py::test_gamma_poisson[(4,)-()]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,wx,x->]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-,w->]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoDelta]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class10]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape2]
0.00s setup    tests/distributions/test_conjugate_update.py::test_beta_binomial[(4,)-()]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-4]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-True-initial_shape1-proposed_shape1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,wi->i]
0.00s setup    tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape1-effective_sample_size]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[abi,b->,b,ai,abi-i]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_class8]
0.00s setup    tests/distributions/test_reshape.py::test_idempotent[1-0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape3]
0.00s setup    tests/distributions/test_conjugate_update.py::test_gamma_poisson[(3, 2)-(3, 2)]
0.00s setup    tests/distributions/test_conjugate_update.py::test_gamma_poisson[(4,)-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-3-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-w,wx,x->]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[idct-3]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[False-A5]
0.00s setup    tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape2-False]
0.00s setup    tests/distributions/test_unit.py::test_expand[batch_shape0-sample_shape0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape5]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype1-size1]
0.00s setup    tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[()-(3, 2)]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-3]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-1.5]
0.00s call     tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[False-True-50-expected1]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[idct--4]
0.00s setup    tests/ops/test_packed.py::test_broadcast_all[shapes3]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(6),z(5,1)]-(5, 6, 2)]
0.00s setup    tests/nn/test_module.py::test_constraints[shape23-constraint_23]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gamma_normal[True-JIT=True]
0.00s setup    tests/infer/test_jit.py::test_traceenum_elbo[1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.1--1.0]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape6]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal[(2, 3)-(4,)]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-100.0]
0.00s setup    tests/contrib/oed/test_xexpx.py::test_xexpx[argument1-output1]
0.00s setup    tests/contrib/test_minipyro.py::test_constraints[minipyro-jit]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,x->]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[4-event_shape1]
0.00s setup    tests/ops/test_contract.py::test_ubersum[bi,ia->,i,ia,ib,iab-i]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-False]
0.00s setup    tests/infer/test_valid_models.py::test_dim_allocation_ok[TraceGraph_ELBO-False]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_ok[Trace_ELBO-full]
0.00s setup    tests/infer/test_jit.py::test_svi[TraceEnum_ELBO-1]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_guide_module_callable]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-0.1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.0]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype0-size2]
0.00s setup    tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(3, 2)-(4,)]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-ij->ji]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwj->ji]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2),0]-(2,)]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-scale3]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-mask3]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoDelta]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-2.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape3]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.4--0.5]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[2-5]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape2-sample_shape1]
0.00s setup    tests/contrib/test_minipyro.py::test_local_param_ok[pyro-py]
0.00s setup    tests/contrib/test_minipyro.py::test_plate_ok[pyro-py]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-3-7]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj->ij]
0.00s setup    tests/ops/test_welford.py::test_welford_dense[1-1]
0.00s setup    tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape2-autocovariance]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->bij-ij]
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_dbn[False-True]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-nested_auto_guide_callable]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.5]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[2-1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape3]
0.00s setup    tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[minipyro-jit]
0.00s call     tests/ops/test_contract.py::test_ubersum_collide_implemented[ubersum-False]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[2-event_shape0]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[False-A1]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[True-A4]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[ai->ai-i]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[1-False]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[sequential-inf]
0.00s setup    tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-auto_class6]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.75]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape2]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape1-event_shape1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwj->ji]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5,1),z(6)]-(5, 6)]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[0.1-0.1]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_guide_module_callable]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoDiagonalNormal]
0.00s setup    tests/distributions/test_von_mises.py::test_von_mises_3d[0.1]
0.00s setup    tests/distributions/test_categorical.py::test_batch_log_dims[dim=1-probs0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[0.0-0.0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-1-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,ixj->ji]
0.00s setup    tests/ops/test_welford.py::test_welford_diagonal[1000-7]
0.00s setup    tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-TraceGraph_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-nested_auto_guide_callable]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class10]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-0.1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-2.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape0]
0.00s setup    tests/distributions/test_stable.py::test_variance[2.0-0.5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape3]
0.00s setup    tests/contrib/test_minipyro.py::test_constraints[pyro-jit]
0.00s setup    tests/ops/test_newton.py::test_newton_step_converges[2-1.0]
0.00s setup    tests/ops/test_indexing.py::test_index[1-1]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...]-(2, 3, 4)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[->-]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[4-5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-1.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape1]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.3--1.0]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal[(4,)-(2, 3)]
0.00s setup    tests/distributions/test_categorical.py::test_sample_dims[dim=1-probs0]
0.00s setup    tests/distributions/test_categorical.py::TestCategorical::test_log_prob_sum
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.1]
0.00s call     tests/pyroapi/test_pyroapi.py::test_generate_data[minipyro]
0.00s setup    tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape0-effective_sample_size]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4)]-(4, 3)]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enumeration_in_tree[TraceEinsumEvaluator]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceGraph_ELBO-full]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_single[Categorical-prior3]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_guide_module_callable]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape4]
0.00s setup    tests/poutine/test_poutines.py::test_enumerate_poutine[0--1]
0.00s setup    tests/ops/test_indexing.py::test_index[0-0]
0.00s setup    tests/infer/reparam/test_transform.py::test_log_normal[(4,)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[Trace_ELBO-None]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoDiagonalNormal]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoDiagonalNormal]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.7--0.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-10]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape3]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape2]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-2-7]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5,1),0]-(5, 2)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),:]-(4, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_distribution[3-(4,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->cij-ij]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_dirichlet_categorical[JIT=False]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_class9]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_guide_callable]
0.00s setup    tests/distributions/test_unit.py::test_shapes[batch_shape0]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[True-p_mask3]
0.00s setup    tests/distributions/test_conjugate_update.py::test_beta_binomial[()-(3, 2)]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[1.0--1.0]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[0.0--1.0]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1429]
0.00s setup    tests/poutine/test_poutines.py::test_replay_enumerate_poutine[1--1]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-False]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_guide_module_callable]
0.00s setup    tests/distributions/test_unit.py::test_expand[batch_shape0-sample_shape2]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.4--1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.0--1.0]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs2]
0.00s setup    tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[pyro-py]
0.00s setup    tests/poutine/test_poutines.py::test_enumerate_poutine[2--3]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(5,1),z(4)]-(5, 4)]
0.00s setup    tests/nn/test_module.py::test_constraints[shape16-constraint_16]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_noprior-1]
0.00s setup    tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape1]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_single[Bernoulli-0.2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.5]
0.00s setup    tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(3, 2)-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Uniform-True-initial_shape1-proposed_shape1]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[,i,i,ia,ia->,i,ia-i]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-scale2]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_prior-scale2]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_guide_list_x]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoDelta]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.7-0.5]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape6]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[False-p_mask2]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape2-event_shape2]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[abi,b->,b,ai,abi-i]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[None-inf]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceGraph_ELBO-sequential]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-TraceEnum_ELBO]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[4-2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.75]
0.00s setup    tests/distributions/test_unit.py::test_expand[batch_shape2-sample_shape2]
0.00s setup    tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape4]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[-1.0-1.0]
0.00s call     tests/poutine/test_poutines.py::TraceHandlerTests::test_trace_param_only
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Uniform-True-initial_shape2-proposed_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[dct--4]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-mask3]
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_dbn[False-False]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[1.0-0.75]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoDelta]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.75]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape5]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-iwxj,ixyj,iyzj,izwj->ji]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-True]
0.00s setup    tests/infer/test_valid_models.py::test_set_has_rsample_ok[Trace_ELBO-False]
0.00s setup    tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-TraceEnum_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[0.1-0.75]
0.00s setup    tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(3, 2)-()]
0.00s setup    tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-3]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape2-event_shape2]
0.00s setup    tests/ops/test_contract.py::test_ubersum[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-mask2]
0.00s setup    tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoDiagonalNormal]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-nested_auto_guide_callable]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs3]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape2]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape4]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape1-sample_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges572]
0.00s setup    tests/ops/test_ssm_gp.py::test_matern_kernel[2.5-1]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[,ai,abij->aij-ij]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class12]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class10]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-10.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.999999-0.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.0-1.0]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-1]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-100]
0.00s setup    tests/distributions/test_reshape.py::test_reshape_reshape[1-2]
0.00s setup    tests/ops/test_tensor_utils.py::test_idct[shape1]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-scale3]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[Trace_ELBO-subsample]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class12]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoIAFNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.5]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.75]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.1-0.0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape4]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(5,1),z(4)]-(5, 4)]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.00s setup    tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_guide_module_callable]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-30]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1274]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-i,wi->i]
0.00s setup    tests/ops/test_newton.py::test_newton_step_trust[1-10.0]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_logsumexp[1-()]
0.00s setup    tests/nn/test_module.py::test_constraints[shape1-constraint_1]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_guide_list_x]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.3-1.0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype0-size1]
0.00s call     tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-0-expand_shape1]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_ok[TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-EnergyDistance_noprior]
0.00s setup    tests/infer/test_autoguide.py::test_subsample_guide_2[AutoDelta-dependent]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class10]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.7--1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.99-0.0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape1]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape3]
0.00s setup    tests/contrib/test_util.py::test_rmv[A0-b0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[wx,xy,yz,zw->]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-0.1]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.4-1.0]
0.00s setup    tests/distributions/test_categorical.py::test_support_dims[dim=3-probs0]
0.00s setup    tests/poutine/test_poutines.py::test_replay_enumerate_poutine[0--2]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-ij->ji]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[idct--2]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[1-event_shape1]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[True-A5]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[,i->-i]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[i->i-i]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[Trace_ELBO-0.1]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceGraph_ELBO-subsample]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.99-0.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1336]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[False-A3]
0.00s setup    tests/nn/test_module.py::test_decorator[2-DecoratorModel]
0.00s setup    tests/nn/test_module.py::test_constraints[shape19-constraint_19]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_ok[TraceTMC_ELBO-full]
0.00s setup    tests/infer/test_jit.py::test_bernoulli_enumerate[shape2-True]
0.00s setup    tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[4-(2, 3)]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-100]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1367]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[trace-normal_normal_model]
0.00s setup    tests/ops/test_contract.py::test_adjoint_marginal[ai,abij,acik->i-ijk]
0.00s setup    tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-TraceTMC_ELBO]
0.00s setup    tests/infer/test_jit.py::test_svi[TraceMeanField_ELBO-10]
0.00s setup    tests/infer/test_enum.py::test_elbo_normal[None-loss]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-1.0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape5]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[trace-bernoulli_normal_model]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj->ij]
0.00s setup    tests/ops/test_ssm_gp.py::test_matern_kernel[1.5-3]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[ab->,a,b,ab,ba-]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.75]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.3-0.5]
0.00s setup    tests/distributions/test_one_hot_categorical.py::test_support_dims[dim=2-probs0]
0.00s setup    tests/distributions/test_conjugate.py::test_variance[dist5]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape1-event_shape1]
0.00s setup    tests/ops/test_contract.py::test_ubersum[,i->,i-i]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[Trace_ELBO-full]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_vector[False-dense]
0.00s setup    tests/infer/test_discrete.py::test_distribution_2[map]
0.00s setup    tests/distributions/test_stable.py::test_sample[2.0--1.0]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1305]
0.00s setup    tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[1-False]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[True-A0]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-TraceTMC_ELBO]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[3]
0.00s setup    tests/infer/test_jit.py::test_svi_irregular_batch_size[Trace_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[0.1-0.25]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class12]
0.00s setup    tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoDelta]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoIAFNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.75]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.3--1.0]
0.00s setup    tests/distributions/test_conjugate.py::test_variance[dist0]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-True-initial_shape2-proposed_shape2]
0.00s setup    tests/poutine/test_poutines.py::test_enumerate_poutine[0--3]
0.00s setup    tests/infer/test_jit.py::test_svi[TraceMeanField_ELBO-1]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.0--0.5]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size0-2]
0.00s setup    tests/ops/test_contract.py::test_adjoint_marginal[ab,bc->-]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_noprior-scale2]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-True]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-mask3]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_jit.py::test_beta_bernoulli[TraceEnum_ELBO-False]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[0-0]
0.00s setup    tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape4]
0.00s setup    tests/distributions/test_one_hot_categorical.py::test_batch_log_dims[dim=3-probs0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-3-7]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Normal]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape1-event_shape1]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[NaiveDirichlet]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-Trace_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceTMC_ELBO-subsample]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceEnum_ELBO-subsample]
0.00s setup    tests/infer/test_enum.py::test_elbo_normal[parallel-loss_and_grads]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.25]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.000001-0.0]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-10]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceEnum_ELBO-full]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[True-A2]
0.00s setup    tests/nn/test_module.py::test_constraints[shape5-constraint_5]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-1]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-0.1]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoLaplaceApproximation]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoMultivariateNormal]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.8--1.0]
0.00s setup    tests/distributions/test_kl.py::test_kl_independent_normal_mvn[1-(2, 3)]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[MultivariateNormal]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj->ji]
0.00s setup    tests/ops/test_tensor_utils.py::test_precision_to_scale_tril[batch_shape0-event_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size2-1]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_add[2-(4,)]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[,i,i,ia,ia->,i,ia-i]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-scale2]
0.00s setup    tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceEnum_ELBO-False]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-TraceEnum_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-EnergyDistance_prior]
0.00s setup    tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape0-False]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class11]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.01-0.0]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-10]
0.00s setup    tests/distributions/test_reshape.py::test_idempotent[2-0]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[True-False]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[idct--3]
0.00s setup    tests/ops/test_contract.py::test_ubersum_collide_implemented[naive_ubersum-True]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-1]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_prior-0.1]
0.00s setup    tests/infer/test_discrete.py::test_distribution_2[sample]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[3-2]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class11]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-False]
0.00s setup    tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[1-True]
0.00s setup    tests/poutine/test_mapdata.py::test_replay[iplate-5]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...,z(3)]-(3,)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.00s setup    tests/nn/test_module.py::test_constraints[shape12-constraint_12]
0.00s setup    tests/nn/test_module.py::test_constraints[shape6-constraint_6]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gamma_poisson[False]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gamma_beta[JIT=True]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-0.9]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[2.0-0.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.1-0.5]
0.00s setup    tests/contrib/test_minipyro.py::test_local_param_ok[minipyro-jit]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-True-initial_shape0-proposed_shape0]
0.00s setup    tests/ops/test_contract.py::test_ubersum[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-mask4]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-mask3]
0.00s setup    tests/infer/test_valid_models.py::test_plate_ok[Trace_ELBO-full]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-Trace_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoDelta]
0.00s call     tests/distributions/test_one_hot_categorical.py::test_sample_dims[dim=2-probs0]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape2-event_shape2]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[a,ai,bij->bij-ij]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-Trace_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-1.1]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.1-0.0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-w,wx,xy,yz->]
0.00s setup    tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape2-split_gelman_rubin]
0.00s setup    tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape2-_quantile]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[0.1-0.5]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_vector[False-flat]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.000001--1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.99--1.0]
0.00s setup    tests/distributions/test_conjugate_update.py::test_gamma_poisson[(4,)-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-False-sample_shapes0]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[abcd]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[False-A0]
0.00s setup    tests/ops/test_contract.py::test_ubersum[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-Trace_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_ok[TraceGraph_ELBO-subsample]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[1.0-0.5]
0.00s setup    tests/infer/test_enum.py::test_elbo_normal[None-loss_and_grads]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-2]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.7-0.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.1-1.0]
0.00s setup    tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype1-size0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-w->]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-EnergyDistance_noprior]
0.00s setup    tests/infer/test_jit.py::test_bernoulli_enumerate[shape1-False]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.01-1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.01-0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.999999--1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.8-0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.1-0.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-100]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[FoldedNormal]
0.00s setup    tests/distributions/test_conjugate_update.py::test_gamma_poisson[(3, 2)-(4,)]
0.00s setup    tests/contrib/test_minipyro.py::test_constraints[pyro-py]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-list-sample_shape2]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1369]
0.00s setup    tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape2-_pi]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[2.0--1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.7-1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[2.0-0.0]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[10-event_shape2]
0.00s setup    tests/ops/test_gaussian.py::test_add[2-(4,)]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoNormal]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[3-event_shape0]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5),0]-(5, 2)]
0.00s setup    tests/ops/test_gaussian.py::test_logsumexp[1-()]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[,i->i-i]
0.00s setup    tests/ops/test_contract.py::test_ubersum[,i,i,ia,ia->,i,ia-i]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-mask2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.25]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.1--0.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-1-7]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-w->]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[4-event_shape2]
0.00s setup    tests/ops/test_indexing.py::test_index[args5-expected5]
0.00s setup    tests/ops/test_indexing.py::test_hmm_example[-3--4]
0.00s setup    tests/ops/test_contract.py::test_adjoint_marginal[a,abi,bcij->-ij]
0.00s setup    tests/infer/test_jit.py::test_categorical_enumerate[shape0-False]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[3-2]
0.00s setup    tests/infer/test_autoguide.py::test_guide_list[AutoDiagonalNormal]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.8-0.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.4-1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.4-0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.3-0.5]
0.00s setup    tests/distributions/test_reshape.py::test_idempotent[2-2]
0.00s setup    tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(4,)-(4,)]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceGraph_ELBO-subsample]
0.00s setup    tests/ops/test_stats.py::test_crps_empirical[3-event_shape2]
0.00s setup    tests/nn/test_module.py::test_constraints[shape18-constraint_18]
0.00s setup    tests/nn/test_module.py::test_constraints[shape11-constraint_11]
0.00s setup    tests/infer/test_enum.py::test_mixture_of_diag_normals[MixtureOfDiagNormalsSharedCovariance-scale1]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[4-5]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.1--0.5]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-p_mask3]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-2]
0.00s call     tests/distributions/test_spanning_tree.py::test_partition_function[6]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct[shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size1-3]
0.00s setup    tests/nn/test_module.py::test_constraints[shape17-constraint_17]
0.00s setup    tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceTMC_ELBO-full]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[31]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.25]
0.00s setup    tests/distributions/test_reshape.py::test_idempotent[0-1]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceTMC_ELBO-full]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-True-initial_shape1-proposed_shape1]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size0-3]
0.00s setup    tests/ops/test_contract.py::test_contract_to_tensor[example3]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[1-True]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-TraceTMC_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-TraceGraph_ELBO]
0.00s setup    tests/infer/test_jit.py::test_categorical_enumerate[shape0-True]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[1.0-0.25]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[2.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.7-1.0]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-100]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[dct--2]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size2-3]
0.00s setup    tests/ops/test_special.py::test_log_binomial_stirling[0.02]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:,0]-(2,)]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-TraceTMC_ELBO]
0.00s setup    tests/distributions/test_unit.py::test_expand[batch_shape1-sample_shape2]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.7--1.0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-i->i]
0.00s setup    tests/ops/test_indexing.py::test_hmm_example[-5--3]
0.00s setup    tests/ops/test_contract.py::test_adjoint_marginal[ai,ai->i-i]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-TraceTMC_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_zip[1.0-0.1]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.4-0.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.4-0.0]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-p_mask3]
0.00s setup    tests/perf/test_benchmark.py::test_benchmark[BernoulliBeta::NUTS]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Stable-Size-sample_shape2]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[4]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[2.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.4--0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.1-1.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-False-tuple]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-wi->i]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[sample-->]
0.00s setup    tests/infer/test_valid_models.py::test_dim_allocation_error[TraceEnum_ELBO-False]
0.00s setup    tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-TraceTMC_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-TraceEnum_ELBO]
0.00s setup    tests/infer/test_jit.py::test_categorical_enumerate[shape1-False]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.000001--0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.999999-0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.99-1.0]
0.00s setup    tests/ops/test_special.py::test_log_beta_stirling[0.2]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.8-1.0]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-30]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-3]
0.00s setup    tests/distributions/test_conjugate_update.py::test_beta_binomial[(3, 2)-(4,)]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_subsample_smoke[None]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwxj,ixyj,iyzj,izwj->ij]
0.00s setup    tests/ops/test_special.py::test_log_beta_stirling[0.10]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[i->,i-i]
0.00s setup    tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-Trace_ELBO]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_single[Categorical-prior1]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.999999-1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.8--1.0]
0.00s setup    tests/distributions/test_reshape.py::test_idempotent[0-2]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-ji,iwj->ji]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-TraceGraph_ELBO]
0.00s setup    tests/infer/test_jit.py::test_categorical_enumerate[shape1-True]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.4-0.5]
0.00s call     tests/ops/test_gaussian.py::test_add[2-()]
0.00s call     tests/distributions/test_gaussian_mixtures.py::test_gsm_log_prob
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[LowRankMultivariateNormal]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi->i]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size1-1]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[True-A1]
0.00s setup    tests/ops/test_indexing.py::test_index[args6-expected6]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[,i,ia->,i,ia-i]
0.00s setup    tests/infer/test_inference.py::test_energy_distance_univariate[0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[2.0-1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.99--0.5]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-3]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[idct--1]
0.00s setup    tests/ops/test_gaussian.py::test_add[3-(4,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_add[1-()]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.00s setup    tests/ops/test_contract.py::test_ubersum[,i,i->,i-i]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[,i->,i-i]
0.00s setup    tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceGraph_ELBO-False]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-30]
0.00s setup    tests/distributions/test_one_hot_categorical.py::test_batch_log_dims[dim=2-probs0]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-True-initial_shape2-proposed_shape2]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[marginal-ij,iwj->ij]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[,i->,i-i]
0.00s setup    tests/ops/test_contract.py::test_ubersum[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.00s setup    tests/infer/test_autoguide.py::test_predictive[auto_class6]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.8-0.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.8--0.5]
0.00s call     tests/distributions/test_stable.py::test_variance[1.7-1.5]
0.00s setup    tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape1-_pi]
0.00s setup    tests/ops/test_gaussian.py::test_add[1-()]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.0-0.0]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-0.01]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-2]
0.00s call     tests/distributions/test_conjugate.py::test_log_prob_support[dist3-values3]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges4]
0.00s setup    tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[0-True]
0.00s setup    tests/ops/test_welford.py::test_welford_dense[1000-7]
0.00s setup    tests/infer/test_valid_models.py::test_dim_allocation_ok[TraceEnum_ELBO-False]
0.00s setup    tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-EnergyDistance_noprior]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_vector[True-dense]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_vector[True-flat]
0.00s setup    tests/distributions/test_unit.py::test_expand[batch_shape1-sample_shape0]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.99--1.0]
0.00s setup    tests/distributions/test_categorical.py::test_sample_dims[dim=2-probs0]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges474]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-wi->i]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_model_error_enum_dim_clash[NUTS-kwargs1]
0.00s setup    tests/distributions/test_unit.py::test_expand[batch_shape2-sample_shape0]
0.00s setup    tests/distributions/test_stable.py::test_shape[batch_shape0-sample_shape1]
0.00s call     tests/poutine/test_mapdata.py::test_custom_subsample[iplate]
0.00s call     tests/ops/test_gaussian.py::test_add[3-()]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[replay-trivial_model]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[0.999999--0.5]
0.00s setup    tests/contrib/gp/test_models.py::test_mean_function_VSGP
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Cauchy]
0.00s setup    tests/ops/test_special.py::test_log_binomial_stirling[1e-08]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_beta_bernoulli[JIT=True]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.01--0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.01--0.5]
0.00s setup    tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[2-False]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-0.9]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.7-0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.99-1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.99--0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.4--1.0]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[9]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges211]
0.00s setup    tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size1-2]
0.00s setup    tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-TraceEnum_ELBO]
0.00s setup    tests/distributions/test_stable.py::test_shape[batch_shape1-sample_shape0]
0.00s setup    tests/ops/test_welford.py::test_welford_diagonal[1-1]
0.00s setup    tests/infer/test_valid_models.py::test_dim_allocation_error[TraceGraph_ELBO-False]
0.00s setup    tests/infer/test_autoguide.py::test_discrete_parallel[AutoNormal]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape1-event_shape1]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[replay-normal_normal_model]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[block-normal_normal_model]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct_dim[dct-1]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gaussian_hmm[3]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[3-5]
0.00s setup    tests/distributions/test_reshape.py::test_idempotent[1-2]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-2]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_subsample_smoke[init_to_median]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1240]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[trace-trivial_model]
0.00s setup    tests/infer/test_valid_models.py::test_plate_ok[TraceGraph_ELBO-full]
0.00s setup    tests/poutine/test_mapdata.py::test_replay[iplate-20]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_add[3-()]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.00s setup    tests/ops/test_contract.py::test_ubersum[ab->,a,b,ab,ba-]
0.00s setup    tests/infer/test_discrete.py::test_distribution_1[map]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[3-5]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.3-0.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.3--0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.1--1.0]
0.00s setup    tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape0-sample_shape1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-3-7]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[True-A3]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),:,0]-(5, 3)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_reentrant_history[1]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.3-0.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.01-0.5]
0.00s setup    tests/distributions/test_lkj.py::test_log_prob_d2[0.1]
0.00s setup    tests/poutine/test_mapdata.py::test_replay[nested_iplate-5]
0.00s setup    tests/poutine/test_mapdata.py::test_cond_indep_stack[nested_iplate-5]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[->-]
0.00s setup    tests/distributions/test_unit.py::test_expand[batch_shape0-sample_shape1]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.7-0.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.8-1.0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-3-7]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-Trace_ELBO-subsample]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_shape[map-wxi,xyi,yzi,zwi->i]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_add[3-(4,)]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceEnum_ELBO-full]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-list-sample_shape1]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[block-normal_model]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:,:]-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_distribution[1-(3, 2)]
0.00s setup    tests/nn/test_module.py::test_constraints[shape2-constraint_2]
0.00s setup    tests/infer/test_jit.py::test_infer_discrete[map-10]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.3-1.0]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Dirichlet]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...]-(2, 3)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[,i->,i-i]
0.00s setup    tests/distributions/test_stable.py::test_sample[2.0-1.0]
0.00s setup    tests/distributions/test_reshape.py::test_reshape[1-3]
0.00s setup    tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-TraceGraph_ELBO]
0.00s setup    tests/infer/test_discrete.py::test_prob[grad]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.01-1.0]
0.00s setup    tests/distributions/test_haar.py::test_haar_ortho[1]
0.00s setup    tests/ops/test_gaussian.py::test_add[2-()]
0.00s setup    tests/infer/test_valid_models.py::test_dim_allocation_error[TraceEnum_ELBO-True]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-Trace_ELBO-subsample]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[bi,ia->,i,ia,ib,iab-i]
0.00s setup    tests/distributions/test_stable.py::test_sample[2.0--0.5]
0.00s setup    tests/distributions/test_stable.py::test_sample[0.99-0.5]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[trace-normal_model]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(2)]-(2,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_add[3-(3, 2)]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[2]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-False]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-tuple-sample_shape1]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges89]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0,z(4)]-(4,)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2))[:]-(2,)]
0.00s setup    tests/distributions/test_stable.py::test_sample_2[1.01-0.0]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_normal_shape[()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-3-7]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.00s setup    tests/ops/test_contract.py::test_einsum_linear[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.00s setup    tests/distributions/test_stable.py::test_shape[batch_shape2-sample_shape2]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoMultivariateNormal]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape0-sample_shape0]
0.00s setup    tests/poutine/test_mapdata.py::test_cond_indep_stack[nested_iplate-20]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),:]-(4, 3)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),z(4)]-(4,)]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.01--1.0]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.0-0.5]
0.00s setup    tests/distributions/test_stable.py::test_shape[batch_shape1-sample_shape1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-1-7]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),z(5,1)]-(5, 4)]
0.00s setup    tests/distributions/test_stable.py::test_shape[batch_shape2-sample_shape1]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[:,0]-(2,)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[0,0]-()]
0.00s setup    tests/infer/test_autoguide.py::test_callable[AutoLowRankMultivariateNormal]
0.00s setup    tests/contrib/oed/test_ewma.py::test_ewma[0.5]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...]-(2,)]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_add[2-(3, 2)]
0.00s setup    tests/nn/test_module.py::test_constraints[shape3-constraint_3]
0.00s setup    tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-list-sample_shape1]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[replay-bernoulli_normal_model]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.00s setup    tests/ops/test_contract.py::test_contract_to_tensor[example1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-1-1]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[False-p_mask3]
0.00s call     tests/contrib/epidemiology/test_util.py::test_cat2_scalar[(2, 3, 4)]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_bernoulli_latent_model[JIT=False]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-3]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_init[-1.0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-3-1]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[10]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[acbd]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4,1)]-(4, 2)]
0.00s setup    tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->bi-ij]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs4]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.00s setup    tests/ops/test_contract.py::test_ubersum[i->,i-i]
0.00s setup    tests/ops/test_contract.py::test_contract_tensor_tree[example4]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-True-HMC-_empty_model]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-list-sample_shape2]
0.00s setup    tests/ops/test_indexing.py::test_index[args8-expected8]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),0,0]-(5,)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[:,z(4)]-(4, 2)]
0.00s setup    tests/ops/test_contract.py::test_naive_ubersum[a->,a-]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[28]
0.00s setup    tests/distributions/test_stable.py::test_sample[1.7--0.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges9]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,:,z(5)]-(5, 2, 3)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),0]-(4,)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2))[z(3)]-(3,)]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_all_discrete_sites_log_prob[TraceEinsumEvaluator]
0.00s setup    tests/distributions/test_reshape.py::test_idempotent[1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges366]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[cba]
0.00s setup    tests/infer/test_autoguide.py::test_guide_list[AutoIAFNormal]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[FoldedNormal]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoDiagonalNormal]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2))[0]-()]
0.00s setup    tests/infer/test_autoguide.py::test_guide_list[AutoNormal]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_delta_smoke[init_to_mean]
0.00s setup    tests/ops/test_contract.py::test_ubersum_collide_ok_3[naive_ubersum]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-list-sample_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-2-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges39]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0,0]-()]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...,:]-(2,)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...,0]-()]
0.00s setup    tests/ops/test_gamma_gaussian.py::test_add[2-()]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-4]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-tuple-sample_shape0]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[0,z(4)]-(4,)]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-3]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[LowRankMultivariateNormal]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-3-1]
0.00s call     tests/ops/test_contract.py::test_adjoint_shape[map-a->-]
0.00s setup    tests/ops/test_gaussian.py::test_add[3-()]
0.00s setup    tests/infer/reparam/test_studentt.py::test_moments[()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-1-7]
0.00s setup    tests/ops/test_contract.py::test_adjoint_marginal[a,abi->-i]
0.00s setup    tests/ops/test_contract.py::test_ubersum_jit[a->,a-]
0.00s setup    tests/ops/test_contract.py::test_ubersum[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges68]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[bca]
0.00s setup    tests/ops/test_indexing.py::test_index[args4-expected4]
0.00s setup    tests/ops/test_contract.py::test_contract_tensor_tree[example7]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-True-NUTS-_empty_model]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-2]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[MultivariateNormal]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges559]
0.00s setup    tests/ops/test_contract.py::test_ubersum_collide_ok_2[ubersum]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_nonreparameterized_both_baselines
0.00s setup    tests/infer/test_autoguide.py::test_callable_return_dict[AutoNormal]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-tuple-sample_shape2]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[->]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0,:]-(3,)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0]-(2,)]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[30]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-Size-sample_shape1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1109]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1359]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,0,z(5)]-(5,)]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[29]
0.00s setup    tests/infer/test_autoguide.py::test_discrete_parallel[AutoLaplaceApproximation]
0.00s setup    tests/infer/test_autoguide.py::test_discrete_parallel[AutoIAFNormal]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-3-7]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),z(4)]-(4,)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:]-(2, 3)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_distribution[2-(4,)]
0.00s setup    tests/ops/test_contract.py::test_adjoint_marginal[ai,abij->i-ij]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-False-HMC-_empty_model]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[27]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-1]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Poisson]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-1-1]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_beta_bernoulli[JIT=False]
0.00s setup    tests/infer/test_autoguide.py::test_callable_return_dict[AutoLaplaceApproximation]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-3-7]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape1-event_shape1]
0.00s setup    tests/ops/test_contract.py::test_adjoint_marginal[a,a->-]
0.00s setup    tests/ops/test_contract.py::test_contract_tensor_tree[example0]
0.00s setup    tests/infer/test_valid_models.py::test_no_log_prob_ok[EnergyDistance_noprior]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-1-7]
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-1-(2, 3, 5)]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges937]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(6),0,z(5,1)]-(5, 6)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-1-1]
0.00s call     tests/poutine/test_poutines.py::ConditionHandlerTests::test_stack_success
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5,1,1)]-(5, 2, 3)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-3-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-True-list]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1079]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[75]
0.00s setup    tests/ops/test_linalg.py::test_sym_rinverse[False-A2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-3-1]
0.00s setup    tests/infer/test_valid_models.py::test_plate_no_size_ok[TraceTMC_ELBO]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-()-1-1]
0.00s setup    tests/poutine/test_properties.py::test_idempotent[replay-normal_model]
0.00s setup    tests/ops/test_indexing.py::test_index[None-None]
0.00s setup    tests/infer/test_autoguide.py::test_callable[AutoLaplaceApproximation]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-3-7]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Categorical]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[72]
0.00s setup    tests/ops/test_stats.py::test_resample[True]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5,1),z(6)]-(5, 6, 2)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5),:]-(5, 4)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),:,:]-(5, 3, 4)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),0,:]-(5, 4)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),z(5,1)]-(5, 4)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-3-7]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_delta_smoke[init_to_median]
0.00s call     tests/distributions/test_lkj.py::test_constraint[value_shape1]
0.00s setup    tests/optim/test_multi.py::test_optimizers[<lambda>1]
0.00s setup    tests/infer/test_autoguide.py::test_callable_return_dict[AutoDiagonalNormal]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1019]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges159]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5,1),0,z(6)]-(5, 6)]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5),z(5)]-(5,)]
0.00s setup    tests/ops/test_gaussian.py::test_rsample_distribution[2-()]
0.00s setup    tests/distributions/test_lkj.py::test_log_prob_eta1[10]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges675]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2),:]-(2, 4)]
0.00s setup    tests/infer/test_enum.py::test_vectorized_importance[100000]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-1-7]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.1]
0.00s call     tests/distributions/test_conjugate.py::test_log_prob_support[dist2-values2]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges69]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1299]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(6),z(5,1)]-(5, 6)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1285]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),:,0]-(5, 3)]
0.00s setup    tests/ops/test_contract.py::test_ubersum[abi,b->,b,ai,abi-i]
0.00s setup    tests/contrib/tracking/test_em.py::test_em[False]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-1-7]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Categorical]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[0,:]-(3,)]
0.00s setup    tests/infer/test_autoguide.py::test_callable[AutoDiagonalNormal]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-Size-sample_shape2]
0.00s setup    tests/infer/test_valid_models.py::test_vectorized_num_particles[TraceGraph_ELBO]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-3-7]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-Trace_ELBO]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5),z(5)]-(5, 2)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-1-7]
0.00s call     tests/distributions/test_spanning_tree.py::test_enumerate_support[6]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges903]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,:,z(5)]-(5, 3)]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[25]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-()-3-1]
0.00s call     tests/distributions/test_spanning_tree.py::test_enumerate_support[2]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_all_discrete_sites_log_prob[TraceTreeEvaluator]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-3-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-Size-sample_shape0]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_non_enumerated_plate_ok[None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1389]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2,3)]-(2, 3)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-list-sample_shape0]
0.00s setup    tests/ops/test_contract.py::test_ubersum[,i,ia->,i,ia-i]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[FoldedNormal]
0.00s setup    tests/contrib/tracking/test_em.py::test_em_nested_in_svi[False]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-3-1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[ij->ji]
0.00s setup    tests/distributions/test_von_mises.py::test_von_mises_3d[1.1]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[w,x->]
0.00s setup    tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(3, 2)-3-7]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_subsample_smoke[init_to_mean]
0.00s setup    tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(3)]-(2, 3)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges292]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1149]
0.00s setup    tests/ops/test_indexing.py::test_index[args7-expected7]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[21]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[NaiveBeta]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[VonMises]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_enumerate_support
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1419]
0.00s setup    tests/ops/test_special.py::test_log_binomial_stirling[1.0]
0.00s setup    tests/ops/test_special.py::test_log_binomial_stirling[0.10]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(4,)-1-1]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-TraceGraph_ELBO]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[False-True]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-False-tuple]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-Size-sample_shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[69]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-(4,)-1-1]
0.00s setup    tests/ops/test_indexing.py::test_index[args9-expected9]
0.00s setup    tests/distributions/test_lkj.py::test_constraint[value_shape1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-1-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[58]
0.00s setup    tests/infer/test_enum.py::test_vectorized_importance[10000]
0.00s setup    tests/infer/test_autoguide.py::test_callable[AutoMultivariateNormal]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[1.0]
0.00s setup    tests/infer/mcmc/test_adaptation.py::test_arrowhead_mass_matrix[True]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[11]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-1-(4,)-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-True-Size]
0.00s setup    tests/ops/test_indexing.py::test_index[args3-expected3]
0.00s setup    tests/infer/test_autoguide.py::test_init_loc_fn[AutoDelta]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-3-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-tuple-sample_shape0]
0.00s setup    tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape2]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[True-False]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_potential_fn_pickling[True]
0.00s setup    tests/distributions/test_lkj.py::test_log_prob_d2[0.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-1-1]
0.00s setup    tests/ops/test_special.py::test_log_binomial_stirling[1e-06]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[10]
0.00s setup    tests/distributions/test_von_mises.py::test_log_prob_normalized[3.0]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[DirichletMultinomial]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[29]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-0-(3, 2)-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1090]
0.00s setup    tests/ops/test_stats.py::test_resample[False]
0.00s setup    tests/infer/test_valid_models.py::test_markov_history[0]
0.00s setup    tests/infer/test_valid_models.py::test_nonnested_plate_plate_ok[Trace_ELBO]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-()-1-7]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_categorical[1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-3-7]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-False]
0.00s setup    tests/ops/test_contract.py::test_contract_to_tensor[example6]
0.00s setup    tests/infer/test_autoguide.py::test_discrete_parallel[AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_von_mises.py::test_von_mises_3d[0.9]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs0]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[18]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_6[sequential]
0.00s setup    tests/distributions/test_von_mises.py::test_von_mises_3d[2.0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-1-1]
0.00s setup    tests/ops/test_contract.py::test_contract_to_tensor[example5]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[20]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-Size-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-list-sample_shape0]
0.00s setup    tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[TraceTMC_ELBO]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[23]
0.00s setup    tests/infer/test_autoguide.py::test_callable_return_dict[AutoDelta]
0.00s setup    tests/infer/test_autoguide.py::test_callable[AutoDelta]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_onehot_entropy_grad[0.3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-1-7]
0.00s call     tests/poutine/test_poutines.py::InferConfigHandlerTests::test_infer_config_sample
0.00s setup    tests/distributions/test_spanning_tree.py::test_log_prob[2]
0.00s setup    tests/distributions/test_spanning_tree.py::test_partition_function[6]
0.00s setup    tests/distributions/test_lkj.py::test_log_prob_d2[2.0]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[LKJCorrCholesky]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[LowRankMultivariateNormal]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-tuple-sample_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-3-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[13]
0.00s setup    tests/ops/test_special.py::test_log_binomial_stirling[0.01]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[24]
0.00s setup    tests/infer/test_discrete.py::test_prob[value]
0.00s setup    tests/distributions/test_von_mises.py::test_log_prob_normalized[100.0]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[NaiveBeta]
0.00s setup    tests/distributions/test_lkj.py::test_log_prob_d2[5.0]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[DirichletMultinomial]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges395]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(3, 2)-3-1]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[False-p_mask2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-Size-sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Poisson]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-1-7]
0.00s call     tests/poutine/test_poutines.py::LiftHandlerTests::test_memoize
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-list-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-tuple-sample_shape2]
0.00s setup    tests/infer/test_autoguide.py::test_callable[AutoNormal]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-(3, 2)-1-1]
0.00s call     tests/ops/test_gaussian.py::test_add[1-()]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceGraph_ELBO-full]
0.00s setup    tests/ops/test_special.py::test_log_binomial_stirling[0.0001]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[2-False-NUTS-_empty_model]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-2-7]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[BetaBinomial]
0.00s setup    tests/ops/test_contract.py::test_ubersum_5[ubersum]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_bernoulli_latent_model[JIT=True]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[11]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_hmm[4]
0.00s setup    tests/distributions/test_spanning_tree.py::test_log_prob[5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-3-7]
0.00s setup    tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoLaplaceApproximation]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-tuple-sample_shape1]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[13]
0.00s setup    tests/distributions/test_lkj.py::test_log_prob_eta1[2]
0.00s setup    tests/contrib/gp/test_models.py::test_mean_function_SGPR
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Dirichlet-True-initial_shape0-proposed_shape0]
0.00s setup    tests/distributions/test_mvt.py::test_mean_var[(3, 2)]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[LowRankMultivariateNormal]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-1-1]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[26]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Poisson]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[DirichletMultinomial]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_amortized_smoke[init_to_mean]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-Trace_ELBO]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges564]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plates_dependency_ok[sequential]
0.00s setup    tests/distributions/test_lkj.py::test_constraint[value_shape2]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[FoldedNormal]
0.00s setup    tests/optim/test_optim.py::test_dynamic_lr[scheduler0]
0.00s setup    tests/ops/test_special.py::test_log_binomial_stirling[0.2]
0.00s setup    tests/infer/test_inference.py::test_energy_distance_multivariate[0]
0.00s setup    tests/distributions/test_von_mises.py::test_von_mises_3d[1.0]
0.00s setup    tests/distributions/test_spanning_tree.py::test_partition_function[3]
0.00s setup    tests/contrib/tracking/test_em.py::test_em[True]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges907]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_hmm[1]
0.00s setup    tests/distributions/test_von_mises.py::test_von_mises_3d[0.5]
0.00s setup    tests/distributions/test_spanning_tree.py::test_log_prob[6]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[DirichletMultinomial]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-()-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-Size-sample_shape2]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges430]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixj->ij]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gaussian_hmm[2]
0.00s setup    tests/infer/test_enum.py::test_svi_step_guide_uses_grad[parallel]
0.00s setup    tests/distributions/test_von_mises.py::test_von_mises_3d[10.0]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Stable]
0.00s setup    tests/contrib/test_minipyro.py::test_generate_data_plate[minipyro]
0.00s setup    tests/ops/test_contract.py::test_ubersum_4[ubersum]
0.00s setup    tests/distributions/test_spanning_tree.py::test_log_prob[4]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_categorical[3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-3-7]
0.00s call     tests/poutine/test_mapdata.py::test_custom_subsample[plate]
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-0-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-Size-sample_shape1]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges546]
0.00s setup    tests/distributions/test_spanning_tree.py::test_partition_function[2]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[6]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct[shape0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_sizes[sizes1]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_2[10]
0.00s setup    tests/distributions/test_von_mises.py::test_log_prob_normalized[1.0]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[LowRankMultivariateNormal]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Geometric]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_categorical[7]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_amortized_smoke[None]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[62]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[LKJCorrCholesky]
0.00s setup    tests/contrib/gp/test_models.py::test_mean_function_VGP_whiten
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[61]
0.00s teardown tests/infer/test_abstract_infer.py::test_information_criterion
0.00s setup    tests/distributions/test_spanning_tree.py::test_partition_function[4]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[NaiveDirichlet]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-Trace_ELBO-full]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1223]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1125]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[73]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[RejectionExponential]
0.00s call     tests/ops/test_gaussian.py::test_marginalize_shape[1-1-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-Size-sample_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[17]
0.00s setup    tests/ops/test_special.py::test_log_beta_stirling[0.11]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_hmm[10]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[FoldedNormal]
0.00s setup    tests/test_examples.py::test_cpu[rsa/semantic_parsing.py --num-samples=10]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[20]
0.00s call     tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceGraph_ELBO-full]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-2-7]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[21]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_median at 0x7f6d86b7e840>]
0.00s setup    tests/infer/test_jit.py::test_svi_irregular_batch_size[TraceMeanField_ELBO]
0.00s setup    tests/distributions/test_spanning_tree.py::test_log_prob[3]
0.00s setup    tests/distributions/test_spanning_tree.py::test_partition_function[5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-6-(3, 2)-3-1]
0.00s setup    tests/test_examples.py::test_cpu[contrib/timeseries/gp_models.py -m lcmgp --test --num-steps=2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-list-sample_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[91]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[acdb]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_non_enumerated_plate_ok[sequential]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_ok[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_lkj.py::test_log_prob_d2[1.0]
0.00s setup    tests/optim/test_optim.py::test_clippedadam_pass[5.0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[wi,wxi,xi->i]
0.00s setup    tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTraceGraph_ELBO]
0.00s setup    tests/distributions/test_lkj.py::test_constraint[value_shape0]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[48]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[17]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Exponential]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[41]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_missing_config_warning[False]
0.00s setup    tests/ops/test_special.py::test_log_binomial_stirling[0.05]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gamma_beta[JIT=False]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Binomial]
0.00s setup    tests/contrib/test_minipyro.py::test_generate_data[pyro]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_dirichlet_categorical[JIT=True]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[3]
0.00s setup    tests/distributions/test_von_mises.py::test_log_prob_normalized[0.3]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Poisson]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[ba]
0.00s setup    tests/ops/test_contract.py::test_ubersum_collide_ok_2[naive_ubersum]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_subsample_param_ok[TraceEnum_ELBO]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[LogNormal]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-tuple-sample_shape2]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges757]
0.00s setup    tests/optim/test_multi.py::test_optimizers[<lambda>2]
0.00s setup    tests/ops/test_contract.py::test_ubersum_2[naive_ubersum]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[ShapeAugmentedBeta]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_dependent_matern_gp[1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-3-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-False-Size]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges484]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges768]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges464]
0.00s call     tests/distributions/test_spanning_tree.py::test_log_prob[6]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[87]
0.00s setup    tests/infer/test_valid_models.py::test_not_has_rsample_ok[TraceEnum_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[2]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[LowRankMultivariateNormal]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Multinomial]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[67]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_7[10]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-3-7]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[16]
0.00s setup    tests/distributions/test_haar.py::test_haar_ortho[8]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[MultivariateStudentT]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[RejectionExponential]
0.00s call     tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-TraceGraph_ELBO]
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-1-(2, 3, 5)]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwj->ij]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[7]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[a]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[HalfCauchy]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[22]
0.00s setup    tests/infer/test_autoguide.py::test_callable_return_dict[AutoMultivariateNormal]
0.00s setup    tests/distributions/test_haar.py::test_haar_ortho[7]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[OneHotCategorical]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-2-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges147]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gamma_poisson[True]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-False-list]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Multinomial]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges137]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[dbca]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-3-7]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ShapeAugmentedDirichlet]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -k=1]
0.00s call     tests/distributions/test_categorical.py::test_sample_dims[dim=3-probs0]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges272]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges73]
0.00s setup    tests/optim/test_optim.py::test_dynamic_lr[scheduler2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[19]
0.00s setup    tests/infer/reparam/test_transform.py::test_log_normal[()]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gaussian_mixture_model[JIT=False]
0.00s setup    tests/infer/test_enum.py::test_avoid_nan[sequential]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Uniform]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Uniform]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[ab]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes0]
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-1-(4, 5)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-True-tuple]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges282]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[38]
0.00s setup    tests/ops/test_integrator.py::test_trajectory[QuarticOscillator]
0.00s setup    tests/ops/test_contract.py::test_contract_tensor_tree[example2]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Poisson]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_haar_transform
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[LogNormal]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.05]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges876]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges362]
0.00s setup    tests/infer/test_inference.py::test_energy_distance_univariate[0.0001]
0.00s setup    tests/infer/test_enum.py::test_multi_dependence_enumeration
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_svi_custom_smoke[True]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1168]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1298]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1156]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[97]
0.00s setup    tests/ops/test_arrowhead.py::test_utilities[5]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_feasible at 0x7f6d86b7e730>0]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1307]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges521]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[57]
0.00s setup    tests/ops/test_packed.py::test_broadcast_all[shapes0]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-()]
0.00s setup    tests/ops/test_contract.py::test_ubersum_4[naive_ubersum]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_bernoulli_latent_model[JIT=True]
0.00s setup    tests/distributions/test_reshape.py::test_sample_shape_order
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[23]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[2-event_shape0]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[adcb]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[abdc]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_dirichlet_categorical[JIT=False]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-2-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges196]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges132]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1392]
0.00s setup    tests/optim/test_optim.py::test_autowrap[Adam]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape1-0.5-0.6-0.4]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges802]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[wxi,xyi,yzi,zwi->i]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[66]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[acb]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-False-2200]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_ok[TraceEnum_ELBO]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape1-0.5-0.6-1.1]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[ExtendedBinomial]
0.00s call     tests/distributions/test_lkj.py::test_constraint[value_shape2]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Multinomial]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges828]
0.00s call     tests/distributions/test_mask.py::test_kl_divergence_type[False-False]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges727]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixj->ji]
0.00s setup    tests/infer/test_autoguide.py::test_nested_autoguide[TraceEnum_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_nested_autoguide[Trace_ELBO]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_shared_cov_log_prob[3]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Gamma]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Cauchy]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixyj,iyzj->ij]
0.00s setup    tests/ops/test_contract.py::test_contract_tensor_tree[example1]
0.00s setup    tests/infer/test_valid_models.py::test_enum_in_model_diamond_error[False]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-subsample-reparam]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[3]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-False-Size]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1078]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges41]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges942]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixyj,iyzj->ji]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[bacd]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enumeration_in_dag[TraceTreeEvaluator]
0.00s setup    tests/infer/test_valid_models.py::test_markov_history[1]
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_reentrant_history[2]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_no_size_ok[TraceEnum_ELBO]
0.00s call     tests/distributions/test_shapes.py::test_categorical_log_prob_shape
0.00s setup    tests/optim/test_optim.py::test_dynamic_lr[scheduler1]
0.00s setup    tests/infer/test_csis.py::test_csis_parameter_update
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Pareto]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Categorical]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[BetaBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Gamma]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[4-event_shape0]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1233]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[bac]
0.00s setup    tests/test_examples.py::test_cpu[contrib/autoname/mixture.py --num-epochs=1]
0.00s call     tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-1-7]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[dacb]
0.00s setup    tests/distributions/test_polya_gamma.py::test_polya_gamma[batch_shape2]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges95]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[34]
0.00s setup    tests/ops/test_contract.py::test_ubersum_3[naive_ubersum]
0.00s setup    tests/infer/test_valid_models.py::test_no_log_prob_ok[EnergyDistance_prior]
0.00s setup    tests/infer/test_autoguide.py::test_guide_list[AutoLaplaceApproximation]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Categorical]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[28]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[ij->ij]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[2]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[TraceTMC_ELBO]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-3-1]
0.00s teardown tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=5 --jit]
0.00s call     tests/poutine/test_poutines.py::test_method_decorator_interface_condition
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges398]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges287]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges53]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[wi->i]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[abc]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[functools.partial(<function init_to_value at 0x7f6d86b7e9d8>, values={'x': tensor(3.)})]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_mean at 0x7f6d86b7e8c8>1]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[11]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Cauchy]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[ExtendedBetaBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Beta]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_cat2_scalar[()]
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-1-(5,)]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1267]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[33]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_subsample_param_ok[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_categorical[2]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[NaiveDirichlet]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Cauchy]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Uniform]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-2-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges710]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[16]
0.00s setup    tests/infer/test_autoguide.py::test_guide_list[AutoMultivariateNormal]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Delta]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-2-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges974]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_normal_shape[(4,)]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Cauchy]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Geometric]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Bernoulli]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1100]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[cab]
0.00s setup    tests/ops/test_contract.py::test_ubersum_1[ubersum]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_beta_binomial[True]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[7]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[RejectionExponential]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges690]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges485]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-2]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_1_gate[1.0]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[SpanningTree]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Beta]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[10-event_shape0]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1272]
0.00s setup    tests/poutine/test_poutines.py::test_pickling[<lambda>1]
0.00s setup    tests/infer/test_valid_models.py::test_variable_clash_in_guide_error[Trace_ELBO]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[NegativeBinomial]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges451]
0.00s setup    tests/infer/test_autoguide.py::test_predictive[AutoDelta]
0.00s setup    tests/infer/test_autoguide.py::test_nested_autoguide[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_polya_gamma.py::test_polya_gamma[batch_shape1]
0.00s setup    tests/distributions/test_lkj.py::test_log_prob_eta1[3]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[LKJCorrCholesky]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges98]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges88]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges864]
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_grid[False]
0.00s setup    tests/infer/test_valid_models.py::test_vectorized_num_particles[Trace_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_enum_in_model_diamond_error[True]
0.00s setup    tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[TraceTMC_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_0_gate[0.5]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_categorical[4]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_log_prob[1]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Normal]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_amortized_smoke[init_to_median]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges448]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges163]
0.00s setup    tests/optim/test_optim.py::test_clippedadam_clip[1.0]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[ji,iwj->ji]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[dcab]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[19]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[14]
0.00s setup    tests/infer/test_inference.py::test_energy_distance_multivariate[1]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[RelaxedOneHotCategorical]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[OneHotCategorical]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[GammaPoisson]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[ShapeAugmentedDirichlet]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_aps_hash[10]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[3-event_shape0]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges763]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges560]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[55]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<pyro.infer.autoguide.initialization._InitToGenerated object at 0x7f6d757450f0>]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[22]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[12]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwj,ixj->ij]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[15]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_4[Trace_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_plate_ok[TraceGraph_ELBO]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[30]
0.00s setup    tests/infer/test_autoguide.py::test_callable_return_dict[AutoLowRankMultivariateNormal]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-False-list]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges127]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[12]
0.00s setup    tests/ops/test_contract.py::test_contract_tensor_tree[example3]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-None-()]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_sizes[sizes2]
0.00s setup    tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data0]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges605]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_gaussian_mixture_model[JIT=True]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_mean at 0x7f6d86b7e8c8>0]
0.00s setup    tests/distributions/test_mask.py::test_mask_noop[(3, 2)]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[LKJCorrCholesky]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Normal]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-3-7]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[15]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[LogNormal]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[MultivariateNormal]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -ss=2 -n=4 -d=20 -p=1000 -f 2 --svi]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[84]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_5[10]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[4]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges712]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges155]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1242]
0.00s setup    tests/ops/test_tensor_utils.py::test_idct[shape2]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_bernoulli_latent_model[JIT=False]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[TraceEnum_ELBO]
0.00s setup    tests/infer/test_predictive.py::test_shapes[True]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_3[1]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges120]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[60]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[functools.partial(<function init_to_uniform at 0x7f6d86b7e950>, radius=0.1)]
0.00s setup    tests/distributions/test_mask.py::test_mask_noop[None]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Exponential]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1095]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges667]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges210]
0.00s setup    tests/optim/test_optim.py::test_clippedadam_lrd[5.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[50]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[9]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_ok[Trace_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_0_gate[0.9]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[NaiveDirichlet]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1384]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1261]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_missing_config_warning[True]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_hmm[20]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_hmm[2]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Weibull]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Categorical]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_persistent_smoke[bp]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.05]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges825]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges730]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1086]
0.00s setup    tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[Trace_ELBO]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[StudentT]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Dirichlet]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1007]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges587]
0.00s setup    tests/infer/test_valid_models.py::test_mean_field_mask_ok[False]
0.00s setup    tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape3]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1031]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges984]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges245]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges160]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[81]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[78]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[20]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[AVFMultivariateNormal]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_categorical[6]
0.00s setup    tests/distributions/test_conjugate.py::test_mean[dist0]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[1.0]
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-1-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-2-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1292]
0.00s setup    tests/ops/test_contract.py::test_contract_to_tensor[example2]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_uniform at 0x7f6d86b7e950>]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_2[Trace_ELBO]
0.00s setup    tests/infer/test_jit.py::test_svi_irregular_batch_size[TraceEnum_ELBO]
0.00s setup    tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data3]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Categorical]
0.00s call     tests/distributions/test_categorical.py::test_view_reshape_bug
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1029]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges443]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges101]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges28]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1384]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[35]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[4]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_dirichlet_categorical[JIT=True]
0.00s setup    tests/infer/mcmc/test_adaptation.py::test_arrowhead_mass_matrix[False]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Geometric]
0.00s setup    tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape1]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[1]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[LogNormal]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[VonMises]
0.00s teardown tests/test_examples.py::test_jit[contrib/gp/sv-dkl.py --epochs=1 --num-inducing=4 --jit]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-2-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges218]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges493]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[18]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_sample at 0x7f6d86b7e7b8>0]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[2]
0.00s setup    tests/infer/test_valid_models.py::test_nonnested_plate_plate_ok[TraceEnum_ELBO]
0.00s setup    tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTraceEnum_ELBO]
0.00s setup    tests/distributions/test_polya_gamma.py::test_polya_gamma[batch_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ExtendedBetaBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[BetaBinomial]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Dirichlet-True-initial_shape1-proposed_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-1-7]
0.00s setup    tests/infer/test_autoguide.py::test_scores[AutoIAFNormal]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges60]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1400]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1164]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges867]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges152]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[63]
0.00s setup    tests/ops/test_arrowhead.py::test_utilities[2]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[ShapeAugmentedBeta]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1421]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges934]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges892]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges397]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges340]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges55]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1065]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges704]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges623]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[i->i]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_3[TraceGraph_ELBO]
0.00s setup    tests/infer/test_jit.py::test_traceenum_elbo[10]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_7[1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_1_gate[10.0]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_0_gate[1.1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_0_gate[0.1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1432]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges610]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1302]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges467]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges292]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[w,wx,x->]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_4[10]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_3[10]
0.00s setup    tests/distributions/test_shapes.py::test_one_hot_categorical_log_prob_shape
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_categorical[8]
0.00s setup    tests/distributions/test_extended.py::test_extended_binomial[0.1]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[Multinomial]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1144]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1055]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges42]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges25]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1127]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges934]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges695]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges615]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges436]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges14]
0.00s setup    tests/optim/test_optim.py::test_clippedadam_pass[1.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[46]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[dcba]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[dbac]
0.00s setup    tests/infer/test_valid_models.py::test_mean_field_mask_ok[True]
0.00s setup    tests/infer/test_valid_models.py::test_plate_iplate_ok[TraceTMC_ELBO]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Binomial]
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-1-(4, 5)]
0.00s call     tests/distributions/test_spanning_tree.py::test_make_complete_graph[cpp-3-expected_grid1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1375]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1027]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1021]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges296]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[77]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_2[1]
0.00s setup    tests/infer/test_autoguide.py::test_guide_list[AutoLowRankMultivariateNormal]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1175]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges925]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges175]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1106]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges784]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges196]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[36]
0.00s setup    tests/infer/test_valid_models.py::test_plate_enum_discrete_no_discrete_vars_warning[True]
0.00s setup    tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data1]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[4]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1207]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges205]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges162]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1420]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges908]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges775]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges459]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[,w->]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[45]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[3]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_0_gate[10.0]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Poisson]
0.00s setup    tests/distributions/test_extended.py::test_extended_binomial[0.02]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Multinomial]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges909]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1435]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1293]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1165]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges710]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges442]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges10]
0.00s setup    tests/optim/test_multi.py::test_optimizers[<lambda>3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[39]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[6]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_value at 0x7f6d86b7e9d8>]
0.00s setup    tests/infer/test_valid_models.py::test_plate_no_size_ok[TraceGraph_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_not_has_rsample_ok[Trace_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_6[1]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[VonMises3D]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[MixtureOfDiagNormalsSharedCovariance]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Gamma]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[5]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1373]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1217]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1038]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges515]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges122]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1039]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges758]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges651]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges356]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[28]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[5]
0.00s setup    tests/ops/test_contract.py::test_ubersum_2[ubersum]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_sample at 0x7f6d86b7e7b8>1]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_ok[Trace_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[0]
0.00s setup    tests/infer/test_autoguide.py::test_discrete_parallel[AutoDelta]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_log_prob[3]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Beta]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1408]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges861]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges482]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges225]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges224]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges738]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges225]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges35]
0.00s setup    tests/params/test_module.py::test_module_sequential[nn_module0]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[8]
0.00s setup    tests/infer/test_valid_models.py::test_plate_enum_discrete_no_discrete_vars_warning[False]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_4[1]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-2-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1150]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges995]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges741]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges116]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1152]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1019]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges933]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwj->ji]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[82]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[80]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Uniform]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1419]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges985]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges532]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges339]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges164]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges44]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges27]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1385]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1260]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges593]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges313]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges260]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges242]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[76]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[32]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_variable_clash_error[Trace_ELBO]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1319]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1253]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges143]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1362]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1112]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1007]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges888]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges597]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges465]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges19]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[wi,xi->i]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[93]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[59]
0.00s setup    tests/infer/test_autoguide.py::test_init_loc_fn[AutoMultivariateNormal]
0.00s setup    tests/infer/test_autoguide.py::test_scores[AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[BetaBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Cauchy]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Cauchy]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[LKJCorrCholesky]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[6]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1172]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges801]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges412]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges232]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1100]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges520]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges331]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges39]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[badc]
0.00s setup    tests/ops/test_integrator.py::test_energy_conservation[QuarticOscillator]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_feasible at 0x7f6d86b7e730>1]
0.00s setup    tests/infer/test_autoguide.py::test_discrete_parallel[AutoDiagonalNormal]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Uniform]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1180]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges982]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges887]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges750]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges463]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges271]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges178]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges773]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges729]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges194]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[i,wi->i]
0.00s setup    tests/distributions/test_spanning_tree.py::test_enumerate_support[1]
0.00s setup    tests/distributions/test_distributions.py::test_expand_enumerate_support
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Binomial]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[ShapeAugmentedGamma]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_add[0.1]
0.00s setup    tests/contrib/oed/test_ewma.py::test_ewma[0.9]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.2]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.2]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges373]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges333]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges322]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges249]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges204]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1434]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1269]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges200]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges84]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_model_with_subsample[5]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_hmm[30]
0.00s setup    tests/infer/test_autoguide.py::test_scores[AutoMultivariateNormal]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[NaiveDirichlet]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Geometric]
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1355]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1242]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1220]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges962]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges871]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges721]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges633]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges620]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges277]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges7]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[ij,iwj->ij]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[70]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_haar.py::test_haar_ortho[4]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Normal]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[ShapeAugmentedBeta]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[16]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges973]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges949]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1396]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1062]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges591]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges298]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges197]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges54]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges30]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges12]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges0]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[cdab]
0.00s setup    tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data2]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_6[10]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model[3]
0.00s setup    tests/infer/test_autoguide.py::test_scores[AutoNormal]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs1]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[27]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-2-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1410]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1081]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges935]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges321]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges230]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1244]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1209]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1141]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1040]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges564]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges364]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges42]
0.00s setup    tests/optim/test_multi.py::test_optimizers[<lambda>4]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[64]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[25]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[inf]
0.00s setup    tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTrace_ELBO]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[MaskedDistribution]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Normal]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-2-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges575]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges521]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1306]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1140]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1093]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges175]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[90]
0.00s setup    tests/distributions/test_independent.py::test_to_event[normal]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1383]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges870]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges311]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges983]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges373]
0.00s setup    tests/optim/test_optim.py::test_clippedadam_lrd[3.0]
0.00s setup    tests/optim/test_optim.py::test_clippedadam_lrd[1.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[43]
0.00s setup    tests/infer/test_enum.py::test_unsafe_bernoulli[sample_shape0]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[ShapeAugmentedGamma]
0.00s setup    tests/distributions/test_delta.py::test_expand[batch_shape0]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges637]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges491]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges30]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1132]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1089]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges783]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges596]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges311]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges150]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[99]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[11]
0.00s setup    tests/infer/test_enum.py::test_compute_marginals_hmm[3]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[OneHotCategorical]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Gamma]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1358]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1337]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1304]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges606]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges142]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges941]
0.00s setup    tests/infer/test_valid_models.py::test_variable_clash_in_guide_error[TraceEnum_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_1[10]
0.00s setup    tests/infer/test_enum.py::test_avoid_nan[None]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[MixtureSameFamily]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[8]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[Dirichlet-True-initial_shape2-proposed_shape2]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges993]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[LogNormal]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Beta]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[ShapeAugmentedDirichlet]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges342]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges263]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges786]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges766]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges678]
0.00s setup    tests/optim/test_optim.py::test_clippedadam_clip[5.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[88]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[52]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[42]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[ShapeAugmentedDirichlet]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[MultivariateNormal]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[HalfCauchy]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges694]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges73]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1191]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1172]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1073]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges940]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges821]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges696]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[79]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[53]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[51]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[49]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[44]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[bdac]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[2-False]
0.00s setup    tests/infer/test_valid_models.py::test_not_has_rsample_ok[TraceTMC_ELBO]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model[2]
0.00s setup    tests/distributions/test_mask.py::test_mask_type[True]
0.00s setup    tests/distributions/test_kl.py::test_kl_delta_normal_shape[(2, 3)]
0.00s setup    tests/distributions/test_hmm.py::test_discrete_hmm_categorical[5]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Bernoulli]
0.00s teardown tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=7 --jit]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges952]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges309]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges239]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1123]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges246]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges131]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[RelaxedOneHotCategoricalStraightThrough]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[VonMises]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[12]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges260]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges79]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges234]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[96]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[56]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[27]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[2]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Binomial]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[Bernoulli]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-ClippedAdam-optim_kwargs1-jit]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1153]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges588]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges241]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges26]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1226]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1163]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges849]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges750]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges154]
0.00s setup    tests/ops/test_tensor_utils.py::test_idct[shape0]
0.00s setup    tests/ops/test_integrator.py::test_energy_conservation[CircularPlanetaryMotion]
0.00s setup    tests/infer/test_valid_models.py::test_plate_reuse_ok[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Beta]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges294]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges254]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges236]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges851]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges583]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges50]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[40]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_ok[EnergyDistance_prior]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_0_gate[2.0]
0.00s setup    tests/distributions/test_mask.py::test_mask_type[mask3]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1274]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1250]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges336]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges193]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwxj,ixyj,iyzj,izwj->ji]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[98]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[85]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[9]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_conditional_householder
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ExtendedBetaBinomial]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges129]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[w,w->]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[31]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_plate_ok[Trace_ELBO]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[LowRankMultivariateNormal]
0.00s setup    tests/distributions/test_mask.py::test_mask_noop[()]
0.00s setup    tests/distributions/test_haar.py::test_haar_ortho[3]
0.00s setup    tests/distributions/test_extended.py::test_extended_beta_binomial[0.05]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[MultivariateNormal]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob_shape[RejectionExponential]
0.00s setup    tests/distributions/test_delta.py::test_expand[batch_shape4]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[19]
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-0-(4, 5)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1139]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1343]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges898]
0.00s setup    tests/ops/test_integrator.py::test_trajectory[HarmonicOscillator]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plates_dependency_ok[None]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_plate_ok[TraceEnum_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_1_gate[1.1]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_0_gate[1.0]
0.00s setup    tests/distributions/test_extended.py::test_extended_binomial[0.0]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Bernoulli]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[ShapeAugmentedGamma]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges594]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[30]
0.00s setup    tests/ops/test_contract.py::test_contract_tensor_tree[example6]
0.00s setup    tests/infer/test_valid_models.py::test_nonnested_plate_plate_ok[TraceGraph_ELBO]
0.00s setup    tests/infer/test_enum.py::test_svi_step_guide_uses_grad[sequential]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Exponential]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_aps_hash[0.1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges722]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges609]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwj,ixj->ji]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[95]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[65]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Gamma]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-()-3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges727]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges304]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges55]
0.00s setup    tests/optim/test_optim.py::test_clippedadam_clip[3.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[29]
0.00s setup    tests/ops/test_packed.py::test_broadcast_all[shapes2]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_onehot_svi_usage
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Categorical]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-False-list]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges365]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges259]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges822]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges133]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[68]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[37]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_3[TraceEnum_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_1_gate[2.0]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[BetaBinomial]
0.00s setup    tests/distributions/test_categorical.py::TestCategorical::test_support
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1406]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges866]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges759]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges240]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1078]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges90]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[74]
0.00s setup    tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[TraceEnum_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_1_gate[0.9]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[RejectionExponential]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges879]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges840]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges877]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges29]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[86]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[bcad]
0.00s setup    tests/infer/test_valid_models.py::test_plate_broadcast_error[Trace_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_1[10]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ShapeAugmentedDirichlet]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[ExtendedBetaBinomial]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges186]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges484]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges156]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[89]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_non_enumerated_plate_ok[parallel]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[1]
0.00s setup    tests/infer/test_sampling.py::ImportanceTest::test_importance_prior
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[5]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_in_iplate[9]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[ExtendedBetaBinomial]
0.00s teardown tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=3 --tmc --tmc-num-samples=2 --jit]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1302]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges722]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[24]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[2-True]
0.00s setup    tests/infer/test_autoguide.py::test_guide_list[AutoDelta]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_1_gate[0.1]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Cauchy]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_aps_try_add[10]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1218]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[23]
0.00s setup    tests/ops/test_integrator.py::test_trajectory[CircularPlanetaryMotion]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[2]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model[4]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Uniform]
0.00s setup    tests/distributions/test_delta.py::test_expand[batch_shape5]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_aps_hash[100]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-False-Size]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1322]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges403]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges547]
0.00s setup    tests/infer/test_enum.py::test_unsafe_bernoulli[sample_shape1]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Normal]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[HalfCauchy]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1317]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1169]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges250]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges479]
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[functools.partial(<function init_to_median at 0x7f6d86b7e840>, num_samples=4)]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[TruncatedPolyaGamma]
0.00s setup    tests/distributions/test_haar.py::test_haar_ortho[9]
0.00s setup    tests/distributions/test_extended.py::test_extended_binomial[0.05]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[NaiveBeta]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[13]
0.00s setup    tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 4 --tmc --tmc-num-samples=2]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_init[scale2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges529]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges245]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges906]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges841]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges770]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges16]
0.00s setup    tests/infer/test_valid_models.py::test_variable_clash_in_model_error[Trace_ELBO]
0.00s setup    tests/contrib/oed/test_ewma.py::test_ewma[0.99]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Poisson]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1416]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1236]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges586]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1010]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges709]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges370]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[83]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[14]
0.00s setup    tests/ops/test_integrator.py::test_energy_conservation[HarmonicOscillator]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[inf]
0.00s setup    tests/infer/test_jit.py::test_svi_irregular_batch_size[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[MultivariateStudentT]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges418]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges394]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges890]
0.00s setup    tests/optim/test_optim.py::test_clippedadam_pass[3.0]
0.00s setup    tests/ops/test_tensor_utils.py::test_dct[shape2]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[92]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[47]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[26]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[cabd]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_plates_dependency_ok[parallel]
0.00s setup    tests/infer/test_valid_models.py::test_plate_reuse_ok[Trace_ELBO]
0.00s setup    tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTraceMeanField_ELBO]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes5]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_bernoulli_shapes[probs1]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Multinomial]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[ExtendedBetaBinomial]
0.00s teardown tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential --jit]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1070]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges854]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[w->]
0.00s setup    tests/infer/test_valid_models.py::test_markov_history[3]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[5]
0.00s setup    tests/test_examples.py::test_cpu[mixed_hmm/experiment.py --timesteps=1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1018]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges714]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges177]
0.00s setup    tests/params/test_module.py::test_param_no_grad[outer]
0.00s setup    tests/params/test_module.py::test_param_no_grad[outest]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_4[TraceEnum_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_plate_stack_sizes[sizes0]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[HalfCauchy]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[15]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-True-Size]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1258]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges820]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges781]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges302]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges151]
0.00s setup    tests/infer/test_valid_models.py::test_not_has_rsample_ok[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[MultivariateStudentT]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges381]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges190]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_subsample_param_ok[Trace_ELBO]
0.00s setup    tests/distributions/test_zero_inflated.py::test_zip_1_gate[0.5]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes15]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Gumbel]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[FisherSnedecor]
0.00s setup    tests/distributions/test_mask.py::test_mask_type[mask2]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[MultivariateStudentT]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[LKJCorrCholesky]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Bernoulli]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[ExtendedBinomial]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[10]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-False-tuple]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-2-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1349]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges942]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges534]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges334]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges105]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges586]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges349]
0.00s setup    tests/optim/test_optim.py::test_autowrap[ClippedAdam]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[bcda]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enumeration_in_dag[TraceEinsumEvaluator]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_enumeration_in_tree[TraceTreeEvaluator]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_ok[TraceTMC_ELBO]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[10]
0.00s setup    tests/infer/test_multi_sample_elbos.py::test_inner_outer[RenyiELBO]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes7]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[20]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges100]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1195]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges633]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges3]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[71]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_2[1]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-None-None]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[3]
0.00s setup    tests/infer/test_autoguide.py::test_discrete_parallel[AutoMultivariateNormal]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[RelaxedBernoulliStraightThrough]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[CoalescentTimesWithRate]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Stable]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[MultivariateStudentT]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -a]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1411]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1312]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1285]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1063]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges774]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges769]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[dabc]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_plate_ok[TraceTMC_ELBO]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[DiscreteHMM]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Gamma]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1162]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges868]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1203]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges375]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[Trace_ELBO]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model[5]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[MixtureOfDiagNormals]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[8]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-2-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges910]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges732]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges480]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges181]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1405]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1371]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges882]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges646]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges94]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[w,wx,xy,yz->]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[cbad]
0.00s setup    tests/infer/test_valid_models.py::test_plate_reuse_ok[TraceTMC_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[Trace_ELBO]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[MultivariateNormal]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[ExtendedBetaBinomial]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[26]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges856]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges376]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges319]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges90]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1332]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1182]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges357]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges99]
0.00s setup    tests/ops/test_special.py::test_log_beta_stirling[1e-08]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[GammaPoisson]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[ExtendedBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Binomial]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_aps_try_add[1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-2-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1386]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1354]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1194]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1152]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1115]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges799]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges628]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges386]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1303]
0.00s setup    tests/ops/test_tensor_utils.py::test_periodic_features[94]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[cbda]
0.00s setup    tests/ops/test_integrator.py::test_time_reversibility[CircularPlanetaryMotion]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_5[1]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_7[1]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model[30]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.55-0.33]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Beta]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Unit]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-2-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges989]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges764]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges413]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges408]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges36]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1324]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges394]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges289]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[1]
0.00s setup    tests/infer/test_valid_models.py::test_plate_iplate_ok[Trace_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[TraceEnum_ELBO]
0.00s setup    tests/distributions/test_spanning_tree.py::test_log_prob[1]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Exponential]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-False-tuple]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1347]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1142]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges247]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges63]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges31]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1366]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1259]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges164]
0.00s setup    tests/ops/test_contract.py::test_ubersum_collide_ok_3[ubersum]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_4[TraceGraph_ELBO]
0.00s setup    tests/infer/test_jit.py::test_traceenum_elbo[2]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[ZeroInflatedPoisson]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[BetaBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[NaiveDirichlet]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Multinomial]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Bernoulli]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1359]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges685]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges329]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges314]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges85]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges21]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1439]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1353]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges662]
0.00s setup    tests/poutine/test_poutines.py::test_pickling[<lambda>0]
0.00s setup    tests/infer/test_valid_models.py::test_plate_reuse_ok[TraceEnum_ELBO]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-2]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[HalfCauchy]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Dirichlet]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[21]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1036]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1026]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges784]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges534]
0.00s setup    tests/infer/test_valid_models.py::test_plate_broadcast_error[TraceGraph_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_7[10]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_3[10]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes13]
0.00s setup    tests/distributions/test_extended.py::test_extended_beta_binomial[0.02]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Dirichlet]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[GammaPoisson]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Stable]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-True-Size]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1438]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1179]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1110]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges957]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges789]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges470]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges465]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1229]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1027]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges701]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[wi,wi->i]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_2[TraceEnum_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_plate_broadcast_error[TraceEnum_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_variable_clash_error[TraceGraph_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_1[1]
0.00s setup    tests/distributions/test_relaxed_straight_through.py::test_bernoulli_entropy_grad[0.5]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ExtendedBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Normal]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_aps_try_add[100]
0.00s call     tests/contrib/tracking/test_hashing.py::test_aps_init[scale2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges719]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges335]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges43]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1169]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges794]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges726]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges81]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[bdca]
0.00s setup    tests/ops/test_integrator.py::test_time_reversibility[HarmonicOscillator]
0.00s setup    tests/infer/test_valid_models.py::test_plate_no_size_ok[Trace_ELBO]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-3]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_6[parallel]
0.00s setup    tests/infer/test_autoguide.py::test_init_loc_fn[AutoLowRankMultivariateNormal]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Categorical]
0.00s setup    tests/contrib/test_minipyro.py::test_generate_data[minipyro]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1255]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1189]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges695]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges638]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges423]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1345]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges869]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges846]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges748]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges581]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges417]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges268]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges221]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges57]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_variable_clash_error[TraceEnum_ELBO]
0.00s setup    tests/infer/test_autoguide.py::test_init_loc_fn[AutoDiagonalNormal]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Stable]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[CoalescentTimes]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Bernoulli]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ExtendedBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ShapeAugmentedBeta]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1237]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1186]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1049]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges599]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges409]
0.00s setup    tests/integration_tests/test_tracegraph_elbo.py::NormalNormalTests::test_elbo_nonreparameterized
0.00s setup    tests/infer/test_valid_models.py::test_plate_iplate_ok[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[ZeroInflatedDistribution]
0.00s setup    tests/distributions/test_extended.py::test_extended_beta_binomial[0.0]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Gamma]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[ExtendedBetaBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[GammaPoisson]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[18]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges905]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges692]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1023]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges492]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges179]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges146]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[adbc]
0.00s setup    tests/infer/test_valid_models.py::test_plate_iplate_ok[TraceEnum_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_iplate_variable_clash_error[TraceTMC_ELBO]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plates_2[10]
0.00s setup    tests/distributions/test_spanning_tree.py::test_partition_function[1]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[IndependentHMM]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[GaussianMRF]
0.00s setup    tests/distributions/test_independent.py::test_to_event[exponential]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Categorical]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Stable]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Binomial]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[OneHotCategorical]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1270]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges979]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges895]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges794]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1216]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1130]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1096]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges984]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges315]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_1[1]
0.00s setup    tests/infer/test_enum.py::test_unsafe_bernoulli[sample_shape2]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes2]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[HalfCauchy]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Dirichlet]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[BetaBinomial]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_add[100]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1401]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges890]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges46]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1117]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1042]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges666]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges276]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges255]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges4]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[wi,wxi,xyi,yzi->i]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[cadb]
0.00s setup    tests/infer/test_valid_models.py::test_variable_clash_in_model_error[TraceTMC_ELBO]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Empirical]
0.00s setup    tests/distributions/test_delta.py::test_expand[batch_shape1]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[24]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[0.1]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[LKJCorrCholesky]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1275]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1174]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1157]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges915]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges779]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges455]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1379]
0.00s setup    tests/params/test_module.py::test_random_module[outest]
0.00s setup    tests/params/test_module.py::test_module_nn[outer]
0.00s setup    tests/ops/test_packed.py::test_unpack_pack[cdba]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_potential_fn_pickling[False]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_ok[TraceMeanField_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_grid[True]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[ImproperUniform]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[Poisson]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[OneHotCategorical]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[17]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1265]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1105]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges960]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges574]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges554]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges989]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges107]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_3[Trace_ELBO]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[7]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[2]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[TransformedDistribution]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[25]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[22]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-2-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges900]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges772]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges460]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges130]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1058]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges924]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges916]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges859]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges539]
0.00s setup    tests/infer/test_enum.py::test_avoid_nan[parallel]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ShapeAugmentedGamma]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Geometric]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Dirichlet]
0.00s setup    tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --aux-loss --jit]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=3]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1132]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges128]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges58]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1431]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges997]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges970]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges526]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges102]
0.00s setup    tests/infer/test_valid_models.py::test_variable_clash_in_guide_error[TraceGraph_ELBO]
0.00s setup    tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Unit]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Laplace]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[HalfNormal]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[GammaGaussianHMM]
0.00s setup    tests/distributions/test_omt_mvn.py::test_log_prob[AVFMultivariateNormal]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Stable]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_cat2_scalar[(2, 3, 4)]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges616]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges277]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges80]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1104]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges612]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges425]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges203]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges58]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges11]
0.00s setup    tests/infer/test_enum.py::test_svi_step_guide_uses_grad[None]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[ExtendedBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Binomial]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[14]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[9]
0.00s setup    tests/contrib/forecast/test_forecaster.py::test_svi_custom_smoke[False]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1396]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1147]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges937]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges648]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges623]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges601]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges176]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges971]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges807]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges654]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges607]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges178]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges162]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges72]
0.00s setup    tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_2[TraceGraph_ELBO]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[5]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Normal]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1184]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1068]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges846]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges102]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges998]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges620]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges522]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges487]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges466]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges412]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges281]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges247]
0.00s setup    tests/infer/test_valid_models.py::test_markov_history[2]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Beta]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[RejectionExponential]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1280]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1049]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges539]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges371]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges269]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges462]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges66]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_error[shapes0]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[VonMises]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ShapeAugmentedBeta]
0.00s setup    tests/distributions/test_distributions.py::test_batch_log_prob[Multinomial]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_aps_try_add[0.1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges947]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges930]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges863]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges324]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges246]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges49]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1413]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1255]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1246]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1143]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges718]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges610]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges322]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges26]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_in_iplate[2]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Exponential]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[VonMises]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1297]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1024]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges990]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges836]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges569]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges475]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges349]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges106]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1044]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges929]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges872]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges573]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges399]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges115]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges97]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges82]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_dirichlet_categorical_grad_adapt
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_scalar[dense]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes2]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[LogNormal]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_add[1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-True-list]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1439]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges831]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges82]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1222]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1122]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges514]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges106]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges86]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges75]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges45]
0.00s setup    tests/params/test_module.py::test_random_module[outer]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-2-3]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[8]
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_3[1]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Multinomial]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[DirichletMultinomial]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Delta]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[MultivariateStudentT]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Dirichlet]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1260]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1073]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges851]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges686]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges643]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges34]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges688]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges443]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges229]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges85]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges43]
0.00s setup    tests/ops/test_packed.py::test_broadcast_all[shapes4]
0.00s setup    tests/ops/test_contract.py::test_contract_tensor_tree[example5]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[ShapeAugmentedDirichlet]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Stable]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_aps_init[-1.0]
0.00s teardown tests/test_examples.py::test_jit[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -p=10000 --sequential --jit]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges961]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges543]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges450]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges191]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1418]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1091]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1060]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges839]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges682]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges358]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges323]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges105]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges70]
0.00s setup    tests/optim/test_optim.py::test_autowrap[SGD]
0.00s setup    tests/infer/test_valid_models.py::test_variable_clash_in_model_error[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes0]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[VonMises]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Rejector]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[LogisticNormal]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[LinearHMM]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Bernoulli]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Multinomial]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[BetaBinomial]
0.00s call     tests/infer/test_valid_models.py::test_plate_stack_sizes[sizes0]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1435]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges975]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges885]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges723]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges549]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges544]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges514]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges400]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1311]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1138]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1036]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges833]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges582]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges378]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges64]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges9]
0.00s setup    tests/infer/test_valid_models.py::test_reparam_ok[EnergyDistance_noprior]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[FoldedDistribution]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Poisson]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[ExtendedBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[LowRankMultivariateNormal]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[GammaPoisson]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[GammaPoisson]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_aps_hash[1]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_add[10]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_init[0.0]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[LKJCorrCholesky]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1206]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges858]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges842]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges752]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges622]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges581]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges524]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges19]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1162]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1142]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges122]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges98]
0.00s setup    tests/ops/test_contract.py::test_contract_to_tensor[example4]
0.00s setup    tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[TraceGraph_ELBO]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes14]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[GammaPoisson]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[ContinuousBernoulli]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[VonMises]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[MultivariateNormal]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-3-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-Size-sample_shape0]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1362]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1042]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges786]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges655]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges417]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges114]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges52]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1254]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1252]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges966]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges815]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges565]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges137]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges126]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges92]
0.00s setup    tests/optim/test_optim.py::test_autowrap[DCTAdam]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwj->ji]
0.00s setup    tests/ops/einsum/test_adjoint.py::test_marginal[iwj->ij]
0.00s setup    tests/infer/test_valid_models.py::test_variable_clash_in_model_error[TraceEnum_ELBO]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes0]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Uniform]
0.00s setup    tests/distributions/test_mask.py::test_mask_type[False]
0.00s setup    tests/distributions/test_extended.py::test_extended_beta_binomial[0.1]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[DirichletMultinomial]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Multinomial]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[OneHotCategorical]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape0-event_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-True-list]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1216]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1122]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1009]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges402]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges145]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges75]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges72]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges64]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges62]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1312]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1202]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1198]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1083]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges848]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges740]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges383]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges237]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges88]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges79]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes1]
0.00s setup    tests/distributions/test_spanning_tree.py::test_enumerate_support[5]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[NaiveBeta]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[MultivariateNormal]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-True-tuple]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1330]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1290]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1170]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1146]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges797]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges725]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges700]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges675]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges597]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges306]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges275]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges151]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1047]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges865]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges641]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges592]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges500]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges480]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges438]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges396]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges208]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges95]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges6]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Independent]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[ExtendedBetaBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[NaiveDirichlet]
0.00s setup    tests/contrib/timeseries/test_gp.py::test_dependent_matern_gp[3]
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-0-(2, 3, 5)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1338]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1182]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges978]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges865]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges810]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges762]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges660]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges618]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges592]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges565]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges505]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges115]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges77]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges70]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges10]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1339]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1155]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1057]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges963]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges962]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges946]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges725]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges680]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges328]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges215]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges145]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges142]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges127]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges27]
0.00s setup    tests/optim/test_optim.py::test_dynamic_lr[scheduler3]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes12]
0.00s setup    tests/distributions/test_spanning_tree.py::test_enumerate_support[6]
0.00s setup    tests/distributions/test_mask.py::test_mask_noop[(4,)]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1420]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1364]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges970]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges673]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges591]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges516]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges478]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges223]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges219]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges84]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges29]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges16]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1314]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1281]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges813]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges505]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges498]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges306]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges111]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges69]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges59]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges52]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges36]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[OneHotCategorical]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[ExtendedBinomial]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[OneHotCategorical]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[100]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1380]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1229]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges894]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges802]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges778]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges682]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges612]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges596]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges552]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges17]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1114]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1099]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges862]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges810]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges619]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges282]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges62]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model[20]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[MaskedMixture]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[Bernoulli]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Exponential]
0.00s setup    tests/distributions/test_distributions.py::test_batch_entropy_shape[Geometric]
0.00s call     tests/distributions/test_pickle.py::test_pickle[TransformedDistribution]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1409]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1352]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1332]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1232]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1021]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1002]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges965]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges946]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges841]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges829]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges822]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges757]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges462]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges270]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges94]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges93]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges61]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1275]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1178]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1034]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges912]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges769]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges579]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges515]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges402]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges372]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges332]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges305]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges123]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges32]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges24]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges20]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges15]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges5]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges2]
0.00s setup    tests/infer/test_enum.py::test_iter_discrete_traces_scalar[flat]
0.00s setup    tests/distributions/test_unit.py::test_shapes[batch_shape2]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[LogNormal]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Geometric]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[FoldedNormal]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1342]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1202]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1164]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1138]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1050]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1034]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges971]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges966]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges754]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges662]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges658]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges556]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges522]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges472]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges449]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges434]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges430]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges362]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges120]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges78]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1425]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1422]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1146]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges905]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges618]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges552]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges485]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges454]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges429]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges386]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges295]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges272]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges252]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges192]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges129]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges124]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges49]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges22]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges21]
0.00s setup    tests/distributions/test_unit.py::test_shapes[batch_shape1]
0.00s setup    tests/distributions/test_spanning_tree.py::test_enumerate_support[2]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Exponential]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_cat2_scalar[(2,)]
0.00s call     tests/distributions/test_categorical.py::test_sample_dims[dim=1-probs0]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[LKJCorrCholesky]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1374]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1199]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1082]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1080]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1062]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges898]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges862]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges855]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges826]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges742]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges530]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges222]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges187]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges83]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges66]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges65]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges48]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1432]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1329]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1258]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1135]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges606]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges568]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges529]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges420]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges395]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges250]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges132]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges114]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges78]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges37]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes3]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[BetaBinomial]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_aps_init[0.0]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1426]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1422]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1262]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1245]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges959]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges943]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges876]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges782]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges781]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges705]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges502]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges242]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges229]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges165]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges118]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges92]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges89]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges15]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1338]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1187]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1177]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1074]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1072]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1006]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges950]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges850]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges840]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges746]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges602]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges232]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges213]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges149]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges138]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges65]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges61]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges25]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges13]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes8]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[Chi2]
0.00s setup    tests/distributions/test_delta.py::test_expand[batch_shape2]
0.00s setup    tests/contrib/epidemiology/test_util.py::test_cat2_scalar[(2, 3)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-Size-sample_shape2]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1407]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1399]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1394]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1321]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1266]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1135]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1119]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1101]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1011]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges994]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges912]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges903]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges869]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges452]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges442]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges419]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges354]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges312]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges169]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges59]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges45]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges12]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1408]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1268]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1194]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1014]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges858]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges694]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges664]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges559]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges502]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges337]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges165]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges134]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges121]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges34]
0.00s setup    tests/ops/test_contract.py::test_ubersum_collide_ok_1[ubersum]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_error[shapes1]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[Geometric]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Exponential]
0.00s call     tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[1-True]
0.00s call     tests/distributions/test_categorical.py::test_sample_dims[dim=2-probs0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-2-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1437]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1227]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1209]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1154]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1060]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges972]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges923]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges914]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges582]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges474]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges426]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges391]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges286]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges125]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges57]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1410]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1330]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1317]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1294]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1225]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1145]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1029]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges853]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges762]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges683]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges489]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges482]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges450]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges326]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges182]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges174]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges119]
0.00s setup    tests/infer/test_smcfilter.py::test_systematic_sample[6]
0.00s setup    tests/distributions/test_delta.py::test_expand[batch_shape3]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape3]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1425]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1222]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1155]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1149]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1145]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1102]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1069]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges906]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges899]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges893]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges839]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges809]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges669]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges654]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges594]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges517]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges507]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges255]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges192]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges188]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges97]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges51]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges22]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges6]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1415]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1296]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1262]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1185]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1050]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1020]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1009]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1002]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges919]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges809]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges532]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges527]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges379]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges110]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges93]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges53]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges46]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges41]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes2]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1402]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1301]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1159]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges939]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges921]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges880]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges777]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges650]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges646]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges370]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges346]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges221]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges124]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1402]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1276]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges975]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges954]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges732]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges712]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges673]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges575]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges241]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges87]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges74]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges18]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes7]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes11]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[MultivariateNormal]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[ShapeAugmentedGamma]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1329]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1298]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1214]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1204]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1186]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1087]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1006]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges944]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges907]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges817]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges800]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges790]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges735]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges724]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges714]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges666]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges642]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges605]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges602]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges510]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges466]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges461]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges352]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges214]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges157]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges156]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1399]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1375]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1334]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1321]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1282]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1245]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1206]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1118]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1087]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1055]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges992]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges893]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges834]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges826]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges798]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges797]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges772]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges754]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges642]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges637]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges609]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges538]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges371]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges286]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges235]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges209]
0.00s setup    tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_in_iplate[5]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[GammaPoisson]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[Categorical]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[NaiveBeta]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[10]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1391]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1379]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1294]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1288]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1263]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1106]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1059]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges954]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges902]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges808]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges804]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges768]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges717]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges702]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges678]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges652]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges561]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges493]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges429]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges363]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges332]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges167]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges139]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges138]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges18]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1218]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1161]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1092]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges952]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges918]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges904]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges777]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges739]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges652]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges548]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges530]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges516]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges408]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges385]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges222]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges202]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges172]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges155]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges147]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges139]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges76]
0.00s setup    tests/poutine/test_mapdata.py::test_custom_subsample[plate]
0.00s setup    tests/infer/test_enum.py::test_hmm_enumerate_model[10]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes1]
0.00s setup    tests/distributions/test_spanning_tree.py::test_enumerate_support[3]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[LKJCorrCholesky]
0.00s setup    tests/distributions/test_independent.py::test_to_event[mvn]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support[Binomial]
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-aij->ai-ij]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1434]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1344]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1299]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1234]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1195]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1108]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1092]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1075]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1015]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1005]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges998]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges875]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges850]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges744]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges737]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges689]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges647]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges570]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges428]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges379]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges201]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges179]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges172]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges149]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges2]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges0]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1347]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1319]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1292]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1289]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1271]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1265]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1231]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1214]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1190]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1189]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1082]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges930]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges921]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges795]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges779]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges759]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges736]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges698]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges690]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges589]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges542]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges525]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges513]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges458]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges449]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges382]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges354]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges266]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges259]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges258]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges219]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges206]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges169]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges47]
0.00s setup    tests/poutine/test_poutines.py::test_pickling[<lambda>3]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes8]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes3]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes10]
0.00s setup    tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ShapeAugmentedGamma]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[NaiveBeta]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1414]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1360]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1282]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1281]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1248]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1074]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges996]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges933]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges919]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges795]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges749]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges738]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges706]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges680]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges626]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges604]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges525]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges330]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges300]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges293]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges280]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges258]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges213]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges152]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges136]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges109]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges74]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges20]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges14]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1355]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1280]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1235]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1219]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1211]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1208]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1128]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1119]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1051]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges994]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges969]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges939]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges902]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges852]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges702]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges665]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges578]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges494]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges488]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges342]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges341]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges340]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges293]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges265]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges263]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges148]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges141]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges100]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges63]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges60]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges8]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes5]
0.00s setup    tests/distributions/test_coalescent.py::test_sample_is_valid[4]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1284]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1198]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1096]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1072]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1047]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1046]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1014]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges918]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges676]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges671]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges656]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges619]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges579]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges538]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges512]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges174]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges35]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges3]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1224]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1160]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1137]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges830]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges761]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges650]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges566]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges507]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges410]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges254]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges170]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges135]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges118]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges31]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes12]
0.00s setup    tests/distributions/test_spanning_tree.py::test_enumerate_support[4]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[RelaxedBernoulli]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_aps_init[scale2]
0.00s call     tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[1-False]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1412]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1389]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1371]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1366]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1213]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1114]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1056]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1010]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges992]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges938]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges699]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges641]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges595]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges498]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges492]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges385]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges361]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges233]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges199]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges177]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges126]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges110]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges50]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges37]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1394]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1386]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1357]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1335]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1309]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1290]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1250]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1234]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1210]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1134]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1124]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1008]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges995]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges979]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges842]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges789]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges778]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges672]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges635]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges631]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges614]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges585]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges426]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges422]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges335]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges309]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges274]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges185]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges77]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges51]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[Uniform]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1382]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1295]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1278]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1235]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1178]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1165]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1032]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1012]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges948]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges884]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges860]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges812]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges715]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges687]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges608]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges566]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges546]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges519]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges503]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges432]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges387]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges358]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges317]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges153]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges133]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1406]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1387]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1368]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1168]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges982]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges911]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges909]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges804]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges752]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges715]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges616]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges588]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges574]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges570]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges557]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges556]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges497]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges473]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges472]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges457]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges404]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges390]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges330]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges273]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges223]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges189]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges130]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges128]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges116]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges91]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges48]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges38]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges33]
0.00s setup    tests/poutine/test_mapdata.py::test_custom_subsample[iplate]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes4]
0.00s setup    tests/distributions/test_distributions.py::test_enumerate_support_shape[ShapeAugmentedBeta]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-1.0-0.5-1.5]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-tuple-sample_shape0]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1385]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1378]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1327]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1315]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1314]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1259]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1249]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1243]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1212]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1211]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1205]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1173]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1035]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1000]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges969]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges955]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges882]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges838]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges835]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges746]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges718]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges649]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges629]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges625]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges621]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges542]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges535]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges444]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges438]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges421]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges372]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges359]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges345]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges318]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges257]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges161]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges146]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges135]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges107]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges54]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges5]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1360]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1354]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1213]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1212]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1205]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1052]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1030]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges965]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges938]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges914]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges818]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges753]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges679]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges624]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges551]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges540]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges440]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges384]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges345]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges318]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges267]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges205]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges186]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges168]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges157]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges71]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges67]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges23]
0.00s setup    tests/ops/test_integrator.py::test_time_reversibility[QuarticOscillator]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes5]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes9]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes13]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[InverseGamma]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_init[scale2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-False-initial_shape0-proposed_shape0]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1415]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1348]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1267]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1240]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1219]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1193]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1185]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1137]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1117]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1099]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1051]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges976]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges929]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges859]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges845]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges813]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges792]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges785]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges771]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges736]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges713]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges698]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges672]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges667]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges665]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges632]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges457]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges435]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges415]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges399]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges336]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges302]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges264]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges262]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges209]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges208]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges170]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges159]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges150]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges134]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges111]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges99]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges96]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges47]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges24]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1411]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1322]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1315]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1266]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1041]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1032]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges968]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges959]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges935]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges896]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges885]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges875]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges829]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges705]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges659]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges658]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges630]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges628]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges511]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges475]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges463]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges452]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges366]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges290]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges199]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges117]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges83]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes6]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes6]
0.00s setup    tests/contrib/test_minipyro.py::test_generate_data_plate[pyro]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-0.5--0.5-0.5]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape5]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1423]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1339]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1325]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1309]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1279]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1254]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1238]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1225]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1191]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1130]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1088]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1070]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1037]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1033]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges964]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges916]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges886]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges874]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges849]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges803]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges758]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges709]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges674]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges634]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges537]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges458]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges453]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges439]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges406]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges392]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges388]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges355]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges298]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges268]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges244]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges231]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges166]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges154]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges104]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges40]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges32]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges23]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1397]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1372]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1325]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1288]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1166]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1139]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1109]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1101]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges988]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges956]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges847]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges835]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges792]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges691]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges509]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges453]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges355]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges352]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges350]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges264]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges201]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges153]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges44]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1427]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1306]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1272]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1123]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1098]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1097]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1065]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1054]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1028]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1022]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges977]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges926]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges834]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges818]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges798]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges703]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges635]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges553]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges425]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges424]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges389]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges325]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges304]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges265]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges195]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges173]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges103]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges71]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1428]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1349]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1313]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1301]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1263]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1249]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1223]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1204]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1150]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1120]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1113]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1088]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1085]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1054]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1015]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1012]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1011]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges990]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges961]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges889]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges874]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges817]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges782]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges685]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges632]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges625]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges365]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges344]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges300]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges253]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges68]
0.00s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_three_layers
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes4]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes8]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[ZeroInflatedNegativeBinomial]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[NaiveDirichlet]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1353]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1231]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1230]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1210]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1134]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges986]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges922]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges913]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges867]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges805]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges730]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges696]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges640]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges593]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges562]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges520]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges479]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges422]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges367]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges331]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges308]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges299]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges220]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges206]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges189]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges182]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges112]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges76]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges56]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1374]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1369]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1270]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1176]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1066]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges987]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges985]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges925]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges922]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges910]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges880]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges844]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges803]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges790]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges785]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges774]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges719]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges699]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges634]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges622]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges595]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges577]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges550]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges543]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges483]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges470]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges414]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges397]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges389]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges301]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges238]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges191]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges109]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges103]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes6]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape0-event_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1393]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1370]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1331]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1300]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1276]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1128]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1091]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1067]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1066]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1045]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1025]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges888]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges881]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges830]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges770]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges765]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges763]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges761]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges739]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges583]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges578]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges550]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges489]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges451]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges382]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges281]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges228]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges226]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1433]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1412]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1407]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1352]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1278]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1200]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1193]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1171]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges978]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges945]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges894]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges879]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges871]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges687]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges660]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges537]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges535]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges439]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges421]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges407]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges334]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges327]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges324]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges285]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges262]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges214]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges195]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges187]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges173]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges158]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1308]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1252]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1203]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1093]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1089]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1085]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1058]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1052]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges889]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges883]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges828]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges814]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges755]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges653]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges645]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges639]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges378]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges377]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges350]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges316]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges273]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges237]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges160]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges67]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1370]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1307]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1297]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1173]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1090]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1046]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1026]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges993]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges788]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges708]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges692]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges649]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges645]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges553]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges434]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges368]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges339]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges316]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges310]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges278]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges269]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges257]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges104]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-2-3]
0.00s setup    tests/distributions/test_pickle.py::test_pickle[GaussianHMM]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape0]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1388]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1335]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1311]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1289]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1192]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1167]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1112]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1039]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges988]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges854]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges820]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges819]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges816]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges807]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges773]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges766]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges751]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges726]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges545]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges486]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges436]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges348]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges338]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges296]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges291]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges252]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges251]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges210]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges198]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges141]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges119]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges113]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges11]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1401]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1326]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1170]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1151]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1126]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1110]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1069]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges967]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges955]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges899]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges897]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges866]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges749]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges629]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges576]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges294]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges270]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges183]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges125]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges101]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges96]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges56]
0.00s setup    tests/poutine/test_poutines.py::test_pickling[<lambda>2]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1429]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1418]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1367]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1365]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1343]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1286]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1271]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1269]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1226]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1196]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1120]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1083]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges999]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges980]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges963]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges920]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges825]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges791]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges767]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges745]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges683]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges651]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges509]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges344]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges285]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges235]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges131]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges91]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges86]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1378]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1361]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1286]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1159]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1154]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1129]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1111]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1077]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1075]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1038]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges986]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges845]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges800]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges767]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges745]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges689]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges684]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges643]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges590]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges569]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges562]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges545]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges491]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges448]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges446]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges432]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges411]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges367]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges363]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges347]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges226]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes3]
0.00s setup    tests/distributions/test_one_hot_categorical.py::TestOneHotCategorical::test_support
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-list-sample_shape0]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1372]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1357]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1326]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1324]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1318]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1133]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1129]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1061]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges981]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges940]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges852]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges821]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges806]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges697]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges636]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges573]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges548]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges540]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges533]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges499]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges469]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges440]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges433]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges404]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges396]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges374]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges328]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges326]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges305]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges274]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges193]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges171]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1438]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1423]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1390]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1380]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1351]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1295]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1257]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1232]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1045]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges957]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges932]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges843]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges742]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges733]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges674]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges587]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges558]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges555]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges506]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges415]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges403]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges380]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges314]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges291]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges239]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges166]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges40]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges28]
0.00s teardown tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --aux-loss --jit]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape0-event_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1436]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1334]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1323]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1121]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1030]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1020]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges928]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges908]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges873]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges832]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges815]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges720]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges691]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges603]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges555]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges511]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges445]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges368]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges303]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges295]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges289]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges163]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges117]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1409]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1376]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1279]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1228]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1105]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1071]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1033]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1024]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges974]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges972]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges900]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges895]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges878]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges861]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges823]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges737]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges700]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges670]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges655]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges533]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges490]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges424]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges419]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges376]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges359]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges319]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges243]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges218]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges167]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges144]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges108]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges17]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1431]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1392]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1346]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1303]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1200]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1158]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1136]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1086]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges991]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges615]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges613]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges611]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges506]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges487]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges471]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges414]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges410]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges401]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges383]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges337]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges183]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges123]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges81]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges38]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges13]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges8]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1424]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1414]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1382]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1381]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1358]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1348]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1227]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1199]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1102]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1031]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1025]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges973]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges948]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges926]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges917]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges870]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges836]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges831]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges787]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges669]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges519]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges393]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges288]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges228]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges161]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges80]
0.00s setup    tests/ops/test_contract.py::test_ubersum_collide_ok_1[naive_ubersum]
0.00s setup    tests/ops/test_contract.py::test_contract_to_tensor[example0]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes11]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Uniform]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1398]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1310]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1239]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges917]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges897]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges853]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges776]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges708]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges627]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges624]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges577]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges568]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges547]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges446]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges409]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges390]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges341]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges297]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges288]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges284]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges266]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges238]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges216]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges202]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges168]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges87]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1365]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1344]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1333]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1115]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1103]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1098]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1022]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1016]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges943]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges838]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges755]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges638]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges626]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges486]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges217]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges198]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges188]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges181]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges171]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1395]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1283]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1268]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1181]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1126]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1125]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1004]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges936]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges927]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges872]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges760]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges747]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges661]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges630]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges567]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges560]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges536]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges504]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges490]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges459]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges313]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges310]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges207]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges194]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges190]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges184]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges148]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges144]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1421]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1346]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1340]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1300]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1287]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1175]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1153]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1004]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges915]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges892]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges827]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges567]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges512]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges501]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges478]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges471]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges469]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges455]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges317]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges231]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges204]
0.00s setup    tests/optim/test_optim.py::test_autowrap[RMSprop]
0.00s setup    tests/contrib/randomvariable/test_random_variable.py::test_pow
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1277]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1163]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1161]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1118]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1041]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1040]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges997]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges951]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges823]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges681]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges679]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges644]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges576]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges531]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges407]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges351]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges290]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges283]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges180]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1430]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1416]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1373]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1342]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1241]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1236]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1201]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1095]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1056]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges808]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges799]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges703]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges510]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges435]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges406]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges398]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges381]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges329]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges312]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges112]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes4]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-2-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1336]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1291]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1221]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1187]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1084]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1071]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1018]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges857]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges780]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges701]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges598]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges571]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges364]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges353]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges279]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges278]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges267]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges33]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1393]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1133]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1059]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges980]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges887]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges819]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges812]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges760]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges741]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges636]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges524]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges499]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges433]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges299]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges280]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges143]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges140]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges136]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-False-scaled]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1428]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1405]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1350]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1320]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1293]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1141]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1057]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges967]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges950]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges844]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges827]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges787]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges748]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges734]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges729]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges599]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges518]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges307]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges217]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges185]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1437]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1395]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1310]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1308]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1230]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1013]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges855]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges832]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges805]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges765]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges647]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges639]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges517]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges504]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges468]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges431]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges418]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges271]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges184]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes9]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape4]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1397]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1351]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1305]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1287]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1244]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1208]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1176]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1160]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1131]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1013]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges837]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges793]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges668]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges589]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges497]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges437]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges416]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges384]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges315]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges276]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges261]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges197]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1327]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1251]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges883]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges837]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges776]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges743]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges686]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges677]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges676]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges541]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges495]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges461]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges456]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges447]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges416]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges361]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges348]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges307]
0.00s call     tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-True-200-expected3]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape0-event_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-3-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1390]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1340]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1261]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1257]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1190]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges740]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges614]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges558]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges557]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges375]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges227]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1427]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1388]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1221]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1183]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1158]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1136]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1097]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges991]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges977]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges958]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges949]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges856]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges711]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges663]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges601]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges563]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges554]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges445]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges427]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges400]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges391]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges346]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges207]
0.00s setup    tests/ops/test_contract.py::test_ubersum_5[naive_ubersum]
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_nonreparameterized_vectorized
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1430]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1403]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1356]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1201]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1183]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1116]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges756]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges707]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges677]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges600]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges523]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges481]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1417]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1356]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1350]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1215]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1061]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges953]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges873]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges796]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges764]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges723]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges713]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges693]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges657]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges644]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges503]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges437]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges283]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges249]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges240]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges216]
0.00s setup    tests/integration_tests/test_tracegraph_elbo.py::RaoBlackwellizationTests::test_nested_iplate_in_elbo
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-2-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1316]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1151]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1103]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1048]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges958]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges924]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges878]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges775]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges607]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges585]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges572]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges477]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges468]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges431]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges405]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges253]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges243]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges234]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges211]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1243]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1053]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1037]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges928]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges913]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges811]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges724]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges561]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges508]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges392]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges333]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges303]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges275]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges256]
0.00s setup    tests/ops/test_packed.py::test_broadcast_all[shapes1]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape_strict[shapes14]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1251]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1177]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1053]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1003]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges901]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges877]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges848]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges753]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges733]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges728]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges659]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges617]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges495]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges447]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges427]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges369]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges347]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges203]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1337]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1121]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1067]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1048]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges960]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges944]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges927]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges923]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges891]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges886]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges744]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges584]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges496]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges444]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges284]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges279]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges212]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges113]
0.00s setup    tests/distributions/test_util.py::test_broadcast_shape[shapes1]
0.00s teardown tests/test_examples.py::test_cpu[sparse_regression.py --num-steps=2 --num-data=50 --num-dimensions 20]
0.00s call     tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-True-280-expected4]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1433]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1224]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges833]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges684]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges663]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges580]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges420]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges411]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges256]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges215]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1363]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1247]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1233]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1197]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1147]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1063]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1000]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges936]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges863]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges816]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges653]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges648]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges287]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges180]
0.00s setup    tests/infer/test_util.py::test_multi_frame_tensor
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoDelta]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1333]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1296]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1264]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1247]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1148]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1140]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1124]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1044]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1023]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1008]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges953]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges891]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges716]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges711]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges693]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges527]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges500]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges467]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges360]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges320]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges301]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1320]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1273]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1107]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1081]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1079]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1035]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges964]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges857]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges771]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges706]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges668]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges476]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges343]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges338]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges320]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges236]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges220]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-1-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1143]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1107]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges956]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges904]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges896]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges664]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges494]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges393]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges380]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges121]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges108]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1283]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1196]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1084]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1068]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges999]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges780]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges717]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges671]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges580]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges518]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1417]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1413]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1376]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1363]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1345]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1313]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1127]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1111]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1064]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges941]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges811]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges783]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges731]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges688]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges670]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges584]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges551]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges528]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges513]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges476]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1264]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1192]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1131]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges981]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges881]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges791]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges751]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges716]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges697]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges617]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges608]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges460]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges413]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges401]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges369]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges360]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges297]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges248]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges176]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1368]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1273]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1246]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1215]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1077]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1043]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1001]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges847]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges456]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges441]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1253]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1217]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1167]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1116]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1076]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1043]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges747]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges728]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges661]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges621]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges604]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges477]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges388]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges233]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges227]
0.00s setup    tests/infer/test_enum.py::test_backwardsample_posterior_3
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-2-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-False-initial_shape0-proposed_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-False-Size]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1404]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1381]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1361]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1341]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1328]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges704]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges508]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges496]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges483]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges343]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges327]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges323]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges248]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges212]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges158]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges140]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1377]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1323]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1304]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1256]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1248]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1181]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1174]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1157]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1003]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges901]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges734]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges627]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges531]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges523]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges441]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges387]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_affine_autoregressive
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-0-(2, 3, 5)]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1241]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1197]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1166]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1104]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges932]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges843]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges824]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges631]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges501]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges464]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges200]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1436]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1277]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1028]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges931]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges824]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges656]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges613]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges603]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges481]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges353]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges321]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges230]
0.00s setup    tests/ops/test_contract.py::test_ubersum_1[naive_ubersum]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1171]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges796]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1316]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1291]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1207]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1180]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1144]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1017]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1001]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges868]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges801]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges731]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges549]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges423]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges351]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges261]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges244]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1256]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1188]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1156]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1113]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1094]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges968]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges945]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges911]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges563]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1403]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1364]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1188]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges951]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges598]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges571]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges544]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges536]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges528]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges428]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges377]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges224]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Independent]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-2-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1237]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1016]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges987]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges931]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges864]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges657]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges488]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges357]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges356]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1404]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1383]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1284]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1148]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1094]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1080]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges976]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges947]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges611]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges374]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges325]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape2]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1424]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1017]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges983]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges590]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1331]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1064]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges814]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges793]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges720]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges681]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges640]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges600]
0.00s setup    tests/ops/test_contract.py::test_ubersum_3[ubersum]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-0.5-0.0-0.5]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges788]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges473]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1328]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1318]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1238]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1179]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1108]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges251]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-1.0-0.0-0.5]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-0.5-0.0-1.5]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1387]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1377]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges526]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges454]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1391]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges721]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1400]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1184]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges996]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges707]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-1-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1076]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges541]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges884]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges405]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges308]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-False-tuple]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1220]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-2-7]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges920]
0.00s call     tests/distributions/test_distributions.py::test_distribution_validate_args[True-Normal-args0]
0.00s setup    tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges743]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges806]
0.00s call     tests/ops/test_packed.py::test_broadcast_all[shapes2]
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-0-(5,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-3-1]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges1341]
0.00s setup    tests/poutine/test_trace_struct.py::test_topological_sort[edges860]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-full-reparam]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Binomial]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-3-1]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-1.0--0.5-1.5]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-1.0-0.5-0.5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-3-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-False-list]
0.00s setup    tests/test_generic.py::test_mcmc_interface[pyro-logistic_regression]
0.00s setup    tests/test_examples.py::test_jit[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -m=1 --enum --jit]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-2-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-True-tuple]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-False-initial_shape1-proposed_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-tuple-sample_shape1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape4]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Binomial]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-None-None]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-1.0-0.0-1.5]
0.00s call     tests/ops/test_contract.py::test_naive_ubersum[->-]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-()-GreaterThan(lower_bound=0.0)]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape0-event_shape0]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-1-1]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_structured_mass
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-3-7]
0.00s teardown tests/test_examples.py::test_jit[dmm/dmm.py --num-epochs=1 --num-iafs=1 --jit]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-False-initial_shape2-proposed_shape2]
0.00s setup    tests/contrib/randomvariable/test_random_variable.py::test_add
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-True-initial_shape0-proposed_shape0]
0.00s teardown tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=4 --jit]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-list-sample_shape1]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -a]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-2-1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-4]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-False-sample_shapes0]
0.00s setup    tests/contrib/gp/test_parameterized.py::test_parameterized
0.00s setup    tests/infer/test_inference.py::NormalNormalTests::test_renyi_reparameterized
0.00s teardown tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=3 --jit]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-False-sample_shapes1]
0.00s setup    tests/contrib/gp/test_models.py::test_mean_function_VSGP_whiten
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-list-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-Size-sample_shape1]
0.00s setup    tests/contrib/gp/test_models.py::test_mean_function_SGPR_DTC
0.00s call     tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-False-150-expected2]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Bernoulli]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-True-initial_shape2-proposed_shape2]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape0]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape3]
0.00s call     tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-0-(4, 5)]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape2]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape2]
0.00s setup    tests/infer/test_inference.py::NormalNormalTests::test_rws_reparameterized
0.00s teardown tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=1 --jit]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-False-unscaled]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_normal_shape[()]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-1-1]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_radial
0.00s teardown tests/test_examples.py::test_jit[dmm/dmm.py --num-epochs=1 --jit]
0.00s setup    tests/test_examples.py::test_cpu[contrib/oed/ab_test.py --num-vi-steps=10 --num-bo-steps=2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-nonreparam-unscaled]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-2-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-tuple-sample_shape2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-2-7]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_serialize
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[sample-->]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-5-sparse]
0.00s call     tests/nn/test_module.py::test_delete
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape5]
0.00s teardown tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=1 --tmc --tmc-num-samples=2 --jit]
0.00s teardown tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=2 --jit]
0.00s teardown tests/test_examples.py::test_cpu[contrib/cevae/synthetic.py --num-epochs=1]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-0.5-0.5-1.5]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape0]
0.00s call     tests/poutine/test_poutines.py::EscapeHandlerTests::test_trace_compose
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape0]
0.00s setup    tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=1 --jit]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ExtendedBinomial]
0.00s setup    tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_nonreparameterized_both_baselines
0.00s setup    tests/distributions/test_lowrank_mvn.py::test_scale_tril
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape4]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=4 --tmc --tmc-num-samples=2]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-0.5-0.9-0.5]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape2]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[()-()-GreaterThan(lower_bound=0.0)]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-parallel-300]
0.00s call     tests/distributions/test_stable.py::test_mean[S-0.5--0.5-0.5]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[marginal-->]
0.00s setup    tests/contrib/autoguide/test_inference.py::AutoGaussianChain::test_multivariatate_normal_auto
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-4]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape1]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape0]
0.00s setup    tests/test_examples.py::test_coverage
0.00s call     tests/distributions/test_stable.py::test_mean[S0-1.0--0.5-0.5]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(2, 3)-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-Size-sample_shape1]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-sequential-sequential-None]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-False-initial_shape1-proposed_shape1]
0.00s teardown tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel --jit]
0.00s teardown tests/test_examples.py::test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide easy]
0.00s teardown tests/test_examples.py::test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide auto]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape1]
0.00s setup    tests/infer/test_sampling.py::ImportanceTest::test_importance_guide
0.00s call     tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-False-initial_shape2-proposed_shape2]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-parallel-300]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-1.0-0.9-0.5]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape4]
0.00s teardown tests/test_examples.py::test_jit[vae/vae.py --num-epochs=1 --jit]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-4]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_whiten_vsgp
0.00s teardown tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{}-7-4]
0.00s call     tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape3]
0.00s teardown tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --num-iafs=1]
0.00s call     tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape0]
0.00s setup    tests/test_examples.py::test_cpu[einsum.py]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-0.5--0.5-1.5]
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_reparameterized_vectorized
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-3-1]
0.00s teardown tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=2 --tmc --tmc-num-samples=2 --jit]
0.00s setup    tests/contrib/gp/test_models.py::test_mean_function_GPR
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=2 --tmc --tmc-num-samples=2]
0.00s call     tests/distributions/test_shapes.py::test_zip_shape
0.00s setup    tests/infer/test_inference.py::test_sequential_plating_sum
0.00s setup    tests/ops/test_stats.py::test_effective_sample_size
0.00s teardown tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=4 --tmc --tmc-num-samples=2 --jit]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-0.5-0.5-0.5]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape0-event_shape0]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[LogNormal]
0.00s teardown tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=21]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-1.0-0.9-1.5]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape0-event_shape0]
0.00s teardown tests/test_examples.py::test_jit[vae/vae_comparison.py --num-epochs=1 --jit]
0.00s teardown tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --jit]
0.00s setup    tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_reparameterized
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_generalized_channel_permute
0.00s call     tests/ops/test_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-()]
0.00s teardown tests/infer/test_util.py::test_psis_diagnostic[5-0.95-krange1-500000]
0.00s call     tests/distributions/test_categorical.py::test_support_dims[dim=1-probs0]
0.00s teardown tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --tmc --tmc-num-samples=2]
0.00s teardown tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1]
0.00s teardown tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-None-None]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_normal_shape[(4,)]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-()-GreaterThan(lower_bound=0.0)]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[ExtendedBinomial]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-nonreparam-unscaled]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(2, 3)-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(5,)-3]
0.00s call     tests/contrib/randomvariable/test_random_variable.py::test_subtract
0.00s setup    tests/contrib/gp/test_models.py::test_mean_function_SGPR_FITC
0.00s setup    tests/integration_tests/test_tracegraph_elbo.py::BernoulliBetaTests::test_elbo_nonreparameterized
0.00s teardown tests/test_examples.py::test_cpu[vae/vae_comparison.py --num-epochs=1]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Dirichlet]
0.00s setup    tests/poutine/test_poutines.py::ReplayHandlerTests::test_replay_full_repeat
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=0 --tmc --tmc-num-samples=2]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-True-0-1000-False-2]
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_interleave
0.00s teardown tests/test_examples.py::test_cpu[vae/vae.py --num-epochs=1]
0.00s call     tests/distributions/test_stable.py::test_mean[S0-0.5-0.9-1.5]
0.00s setup    tests/nn/test_module.py::test_to_pyro_module_
0.00s setup    tests/infer/test_inference.py::SafetyTests::test_duplicate_names
0.00s teardown tests/test_examples.py::test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide custom]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=3 --tmc --tmc-num-samples=2]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-True-initial_shape0-proposed_shape0]
0.00s teardown tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[NaiveBeta]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-2-7]
0.00s setup    tests/integration_tests/test_tracegraph_elbo.py::ExponentialGammaTests::test_elbo_nonreparameterized
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-None-None]
0.00s teardown tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential]
0.00s setup    tests/poutine/test_poutines.py::QueueHandlerMixedTest::test_queue_enumerate
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-nonreparam-scaled]
0.00s setup    tests/test_examples.py::test_cpu[baseball.py --num-samples=200 --warmup-steps=100 --num-chains=2]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_conditional_spline
0.00s setup    tests/infer/test_inference.py::PoissonGammaTests::test_elbo_reparameterized
0.00s setup    tests/infer/test_inference.py::PoissonGammaTests::test_elbo_nonreparameterized
0.00s setup    tests/test_examples.py::test_cpu[contrib/gp/sv-dkl.py --binary --epochs=1 --num-inducing=4 --batch-size=1000]
0.00s setup    tests/test_examples.py::test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide custom]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(5,)-2]
0.00s setup    tests/test_examples.py::test_cpu[lda.py --num-steps=2 --num-words=100 --num-docs=100 --num-words-per-doc=8]
0.00s call     tests/contrib/randomvariable/test_random_variable.py::test_multiply_divide
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6 --raftery-parameterization]
0.00s call     tests/distributions/test_kl.py::test_kl_delta_normal_shape[(2, 3)]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-7-3]
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape0-event_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-True-initial_shape1-proposed_shape1]
0.00s setup    tests/infer/test_conjugate_gradients.py::ConjugateChainGradientTests::test_gradients
0.00s teardown tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1 --aux-loss]
0.00s setup    tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 3 --tmc --tmc-num-samples=2]
0.00s setup    tests/contrib/tracking/test_dynamic_models.py::test_NcpContinuous
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[VonMises]
0.00s call     tests/contrib/randomvariable/test_random_variable.py::test_abs
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6]
0.00s setup    tests/poutine/test_poutines.py::InferConfigHandlerTests::test_infer_config_sample
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_conditional_affine_autoregressive
0.00s setup    tests/test_examples.py::test_jit[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -p=10000 --sequential --jit]
0.00s call     tests/distributions/test_pickle.py::test_pickle[VonMises3D]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-False-scaled]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=7]
0.00s call     tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[False-False-100-expected0]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[MultivariateStudentT]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-True-initial_shape2-proposed_shape2]
0.00s teardown tests/test_examples.py::test_cpu[air/main.py --num-steps=1 --no-baseline]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(5,)-3]
0.00s setup    tests/test_examples.py::test_cpu[rsa/hyperbole.py --price=10000]
0.00s call     tests/distributions/test_shapes.py::test_bernoulli_log_prob_shape
0.00s setup    tests/infer/test_inference.py::NormalNormalTests::test_mmd_vectorized
0.00s call     tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape0-event_shape0]
0.00s setup    tests/contrib/forecast/test_evaluate.py::test_custom_warm_start
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob_shape[Stable]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(2, 3)-3]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[DirichletMultinomial]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-(3,)-()]
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-,ai->a-i]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_gamma_normal
0.00s teardown tests/test_generic.py::test_trace_handler[minipyro-beta_binomial]
0.00s setup    tests/distributions/test_hmm.py::test_gaussian_hmm_high_obs_dim
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape0]
0.00s setup    tests/params/test_param.py::ParamStoreDictTests::test_save_and_load
0.00s setup    tests/infer/mcmc/test_nuts.py::test_arrowhead_mass
0.00s setup    tests/infer/test_valid_models.py::test_plate_shape_broadcasting
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_rws_nonreparameterized
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-tuple-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-Size-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[DirichletMultinomial]
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-False-Size]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-(5,)-()-()-()-()-()]
0.00s setup    tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=2 --tmc --tmc-num-samples=2 --jit]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=1 --tmc --tmc-num-samples=2]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[MultivariateStudentT]
0.00s setup    tests/test_examples.py::test_jit[vae/vae.py --num-epochs=1 --jit]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(2, 3)-2]
0.00s setup    tests/infer/test_inference.py::TestFixedModelGuide::test_guide_and_model_both_fixed
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-3-7]
0.00s setup    tests/perf/test_benchmark.py::test_benchmark[BernoulliBeta::HMC]
0.00s call     tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-True-18-expected5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-1-1]
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape3]
0.00s setup    tests/infer/test_valid_models.py::test_tail_adaptive_error
0.00s call     tests/contrib/epidemiology/test_util.py::test_cat2_scalar[(2, 3)]
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_rws_reparameterized_vectorized
0.00s setup    tests/distributions/test_shapes.py::test_normal_log_prob_shape
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_chain
0.00s setup    tests/perf/test_benchmark.py::test_benchmark[VSGP::MultiClass_whiten=True]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-list-sample_shape0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-7-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-1-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-()-(3,)]
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_reparameterized
0.00s setup    tests/infer/test_autoguide.py::test_empty_model_error
0.00s setup    tests/contrib/tracking/test_assignment.py::test_sparse_grid_smoke
0.00s setup    tests/perf/test_benchmark.py::test_benchmark[PoissonGamma::reparam=True_Trace]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-()-()]
0.00s setup    tests/distributions/test_delta.py::TestDelta::test_batch_log_prob
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(5,)-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-2-1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-4]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-False-unscaled]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[DirichletMultinomial]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_overlapping_plates_error
0.00s setup    tests/perf/test_benchmark.py::test_benchmark[VSGP::MultiClass_whiten=False]
0.00s setup    tests/infer/test_inference.py::TestFixedModelGuide::test_guide_fixed
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-7-3]
0.00s setup    tests/infer/test_inference.py::NormalNormalTests::test_elbo_analytic_kl
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-(3,)-()]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(2, 3)-3]
0.00s setup    tests/poutine/test_poutines.py::BlockHandlerTests::test_block_tutorial_case
0.00s teardown tests/test_examples.py::test_jit[contrib/epidemiology/regional.py --jit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-0-3]
0.00s setup    tests/perf/test_benchmark.py::test_benchmark[PoissonGamma::reparam=False_TraceGraph]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-()-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-False-Size]
0.00s setup    tests/integration_tests/test_tracegraph_elbo.py::NormalNormalTests::test_elbo_reparameterized
0.00s setup    tests/distributions/test_lowrank_mvn.py::test_log_prob
0.00s setup    tests/contrib/tracking/test_dynamic_models.py::test_NcpDiscrete
0.00s call     tests/distributions/test_delta.py::TestDelta::test_batch_log_prob_shape
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_rws_reparameterized
0.00s setup    tests/contrib/tracking/test_dynamic_models.py::test_NcvDiscrete
0.00s call     tests/contrib/randomvariable/test_random_variable.py::test_pow
0.00s setup    tests/poutine/test_poutines.py::TraceHandlerTests::test_trace_full
0.00s setup    tests/infer/test_valid_models.py::test_reparam_stable
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_reparameterized_vectorized
0.00s setup    tests/distributions/test_categorical.py::TestCategorical::test_support_non_vectorized
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-3-7]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_iplate_single_ok
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_single_single_ok
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-2-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-Size-sample_shape2]
0.00s call     tests/distributions/test_spanning_tree.py::test_make_complete_graph[cpp-4-expected_grid2]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-(2, 1)-(3,)]
0.00s setup    tests/infer/test_initialization.py::test_init_to_generated
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-list-sample_shape1]
0.00s call     tests/contrib/autoname/test_scoping.py::test_no_param
0.00s setup    tests/infer/test_inference.py::PoissonGammaTests::test_rws_reparameterized
0.00s setup    tests/infer/test_inference.py::TestFixedModelGuide::test_model_fixed
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_beta_binomial
0.00s setup    tests/poutine/test_poutines.py::EscapeHandlerTests::test_all_escape
0.00s setup    tests/contrib/gp/test_models.py::test_inference_vsgp
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-10]
0.00s setup    tests/nn/test_module.py::test_clear
0.00s setup    tests/infer/reparam/test_conjugate.py::test_beta_binomial_elbo
0.00s call     tests/distributions/test_delta.py::test_shapes[False-3-2]
0.00s setup    tests/poutine/test_poutines.py::BlockHandlerTests::test_block_partial_hide
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_nonreparameterized_vectorized
0.00s setup    tests/infer/test_inference.py::PoissonGammaTests::test_renyi_reparameterized
0.00s setup    tests/infer/test_abstract_infer.py::test_information_criterion
0.00s setup    tests/distributions/test_shapes.py::test_categorical_shape
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-3-1]
0.00s setup    tests/contrib/easyguide/test_easyguide.py::test_overlapping_plates_ok
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-tuple-sample_shape2]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-sequential-None]
0.00s setup    tests/perf/test_benchmark.py::test_benchmark[PoissonGamma::reparam=True_TraceGraph]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/distributions/test_gaussian_mixtures.py::test_gsm_log_prob
0.00s setup    tests/contrib/gp/test_parameterized.py::test_nested_parameterized
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-0-7]
0.00s setup    tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential --jit]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_affine_coupling
0.00s setup    tests/contrib/tracking/test_dynamic_models.py::test_NcvContinuous
0.00s setup    tests/optim/test_optim.py::OptimTests::test_per_param_optim
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_reparameterized
0.00s setup    tests/distributions/test_lowrank_mvn.py::test_variance
0.00s setup    tests/contrib/test_util.py::test_lexpand
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-1-1]
0.00s setup    tests/contrib/tracking/test_measurements.py::test_PositionMeasurement
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-tuple-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Stable-tuple-sample_shape1]
0.00s setup    tests/infer/test_valid_models.py::test_no_plate_enum_discrete_batch_error
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_permute
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-3-1]
0.00s setup    tests/poutine/test_poutines.py::UnconditionHandlerTests::test_uncondition
0.00s setup    tests/ops/test_jit.py::test_varying_len_args
0.00s setup    tests/nn/test_module.py::test_delete
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-False-list]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-list-sample_shape2]
0.00s setup    tests/poutine/test_poutines.py::BlockHandlerTests::test_block_expose_fn
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-True-1800]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=5 --tmc --tmc-num-samples=2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Binomial]
0.00s setup    tests/nn/test_autoregressive.py::AutoRegressiveNNTests::test_masks
0.00s setup    tests/infer/mcmc/test_mcmc_util.py::test_init_to_value
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Stable-Size-sample_shape1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-()-(3,)]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-True-None]
0.00s setup    tests/infer/test_enum.py::test_elbo_hmm_growth
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 -nb=4]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[()-()-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Stable-Size-sample_shape0]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-2-3-(3, 2)]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_conditional_planar
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-True-tuple]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Stable-list-sample_shape0]
0.00s setup    tests/test_examples.py::test_jit[lkj.py --n=50 --num-chains=1 --warmup-steps=100 --num-samples=200 --jit]
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape2]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[()-()-Real()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-3-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-Size-sample_shape1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-(3,)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-3-1]
0.00s setup    tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_reparameterized_N_is_8
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-subsample-nonreparam]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-2-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[MultivariateStudentT]
0.00s setup    tests/poutine/test_poutines.py::test_method_decorator_interface_condition
0.00s setup    tests/poutine/test_poutines.py::test_decorator_interface_queue
0.00s call     tests/poutine/test_poutines.py::EscapeHandlerTests::test_discrete_escape
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-ai->a-i]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-tuple-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Stable-tuple-sample_shape2]
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape3]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-()-3]
0.00s setup    tests/poutine/test_poutines.py::IndirectLambdaHandlerTests::test_graph_structure
0.00s setup    tests/ops/test_tensor_utils.py::test_next_fast_len
0.00s setup    tests/infer/test_valid_models.py::test_mean_field_warn
0.00s setup    tests/contrib/gp/test_models.py::test_mean_function_VGP
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(3, 2)]
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-tuple-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-Size-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Stable-list-sample_shape2]
0.00s setup    tests/nn/test_module.py::test_pyro_serialize
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_nested
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_nonreparameterized
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-()]
0.00s setup    tests/infer/test_inference.py::PoissonGammaTests::test_mmd_vectorized
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Stable]
0.00s setup    tests/infer/test_valid_models.py::test_enum_discrete_single_ok
0.00s setup    tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full_hide_expose
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape1]
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_mmd_vectorized
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(5,)-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-True-Size]
0.00s setup    tests/poutine/test_poutines.py::LiftHandlerTests::test_splice
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_mutual_recursion
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-2-7]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-2]
0.00s setup    tests/infer/test_jit.py::test_grad_expand
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_conditional_affine_coupling
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-0-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-list-sample_shape2]
0.00s setup    tests/infer/test_inference.py::NormalNormalTests::test_elbo_reparameterized
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-1-7]
0.00s setup    tests/optim/test_multi.py::test_multi_optimizer_disjoint_ok
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_reentrant
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_sylvester
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(5,)-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-False-list]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Stable-tuple-sample_shape0]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-False-1800]
0.00s setup    tests/test_examples.py::test_jit[vae/vae_comparison.py --num-epochs=1 --jit]
0.00s setup    tests/test_examples.py::test_jit[eight_schools/svi.py --num-epochs=1 --jit]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-5]
0.00s call     tests/distributions/test_delta.py::test_shapes[True-3-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_by[Stable-list-sample_shape1]
0.00s setup    tests/contrib/oed/test_ewma.py::test_ewma_log
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-2]
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape0]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-(5,)-()-()-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(4,)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-1-7]
0.00s setup    tests/nn/test_module.py::test_module_cache
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-False-None]
0.00s setup    tests/infer/test_enum.py::test_backwardsample_posterior_2
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[2-True-2200]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-nonreparam-scaled]
0.00s setup    tests/poutine/test_poutines.py::test_trace_log_prob_sum_err_msg
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-1]
0.00s setup    tests/contrib/autoname/test_scoping.py::test_multi_nested
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-2]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-True-list]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Stable]
0.00s setup    tests/poutine/test_nesting.py::test_nested_reset
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-()-(3,)]
0.00s setup    tests/infer/test_valid_models.py::test_mean_field_ok
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_spline_coupling
0.00s setup    tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=4 --jit]
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape3]
0.00s setup    tests/infer/test_valid_models.py::test_three_indep_plate_at_different_depths_ok
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-3-1]
0.00s call     tests/distributions/test_stable.py::test_mean[S-1.0--0.5-0.5]
0.00s setup    tests/infer/test_inference.py::PoissonGammaTests::test_renyi_nonreparameterized
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-parallel-300]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-3]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-False-tuple]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-Size-sample_shape0]
0.00s setup    tests/ops/test_stats.py::test_cummin
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-(2, 1)-(3,)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-10]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-()-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-2-3-(3, 2)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-subsample-reparam]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_chains': 2, 'mp_context': 'spawn'}-0-7]
0.00s teardown tests/test_examples.py::test_jit[lda.py --num-steps=2 --num-words=100 --num-docs=100 --num-words-per-doc=8 --jit]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-6]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-()-3]
0.00s call     tests/distributions/test_stable.py::test_mean[S-1.0--0.5-1.5]
0.00s setup    tests/nn/test_module.py::test_svi_smoke
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_rws_nonreparameterized_vectorized
0.00s setup    tests/infer/test_inference.py::NormalNormalTests::test_renyi_nonreparameterized
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_block_autoregressive_jacobians
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-6]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_inference.py::NormalNormalTests::test_rws_nonreparameterized
0.00s setup    tests/contrib/epidemiology/test_distributions.py::test_relaxed_binomial
0.00s setup    tests/poutine/test_poutines.py::TraceHandlerTests::test_trace_param_only
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_polynomial
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_leaky_relu
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-6]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(6),z(5,1)]-(5, 6, 2)]
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-aij->aj-ij]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-True-tuple]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[RejectionExponential]
0.00s setup    tests/ops/test_stats.py::test_weighted_waic
0.00s setup    tests/contrib/tracking/test_ekf.py::test_EKFState_with_NcvContinuous
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-4]
0.00s setup    tests/test_examples.py::test_jit[contrib/cevae/synthetic.py --num-epochs=1 --jit]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-3]
0.00s call     tests/ops/test_stats.py::test_crps_empirical[1-event_shape0]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-tuple-sample_shape0]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-1.0-()]
0.00s setup    tests/infer/test_inference.py::PoissonGammaTests::test_rws_nonreparameterized
0.00s setup    tests/test_examples.py::test_cpu[rsa/schelling_false.py --num-samples=10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-5]
0.00s call     tests/distributions/test_stable.py::test_mean[S-0.5--0.5-1.5]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-True-Size]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-(2, 1)-(3,)]
0.00s setup    tests/infer/test_inference.py::test_reparam_stable
0.00s setup    tests/contrib/test_util.py::test_get_indices_sizes
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-5]
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape0]
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape1]
0.00s call     tests/distributions/test_stable.py::test_mean[S-0.5-0.0-1.5]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-(2, 3)-(2, 3)]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-glgssm-3-1]
0.00s setup    tests/distributions/test_shapes.py::test_one_hot_categorical_shape
0.00s setup    tests/test_examples.py::test_jit[contrib/epidemiology/sir.py --jit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-4]
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape2]
0.00s call     tests/distributions/test_stable.py::test_mean[S-1.0-0.5-1.5]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.6-()]
0.00s setup    tests/contrib/tracking/test_assignment.py::test_sparse_smoke
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-3-1]
0.00s call     tests/distributions/test_stable.py::test_mean[S-1.0-0.5-0.5]
0.00s setup    tests/poutine/test_poutines.py::ReplayHandlerTests::test_replay_full
0.00s setup    tests/distributions/test_delta.py::TestDelta::test_batch_log_prob_shape
0.00s setup    tests/test_examples.py::test_jit[contrib/epidemiology/regional.py --jit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2]
0.00s call     tests/distributions/test_stable.py::test_mean[S-1.0-0.0-0.5]
0.00s setup    tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_nonreparameterized
0.00s setup    tests/infer/test_inference.py::NormalNormalTests::test_elbo_tail_adaptive
0.00s setup    tests/distributions/test_util.py::test_weakmethod
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_neural_autoregressive
0.00s setup    tests/contrib/randomvariable/test_random_variable.py::test_neg
0.00s setup    tests/test_examples.py::test_jit[minipyro.py --jit]
0.00s setup    tests/test_examples.py::test_jit[minipyro.py --backend=pyro --jit]
0.00s setup    tests/test_examples.py::test_cpu[minipyro.py]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-2]
0.00s setup    tests/ops/test_stats.py::test_gelman_rubin
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-(3,)-()]
0.00s setup    tests/infer/test_jit.py::test_jit_arange_workaround
0.00s setup    tests/infer/test_inference.py::NormalNormalTests::test_mmd_nonvectorized
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-True-unscaled]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-scaled]
0.00s setup    tests/contrib/tracking/test_ekf.py::test_EKFState_with_NcpContinuous
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-ssmgp-3-1.5]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-0-3]
0.00s setup    tests/test_examples.py::test_jit[contrib/epidemiology/sir.py --jit -np=128 -ss=2 -n=4 -d=20 -p=1000 -f 2 --svi]
0.00s call     tests/distributions/test_stable.py::test_mean[S-1.0-0.9-0.5]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-()-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-True-0-1000-False-2]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_planar
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-10]
0.00s setup    tests/poutine/test_poutines.py::LiftHandlerTests::test_unlifted_param
0.00s setup    tests/contrib/gp/test_kernels.py::test_active_dims_disjoint_ok
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-3-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_shapes.py::test_categorical_log_prob_shape
0.00s setup    tests/distributions/test_categorical.py::test_view_reshape_bug
0.00s setup    tests/contrib/test_hessian.py::test_hessian_multi_variables
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-(5,)-()-()-(3, 2)]
0.00s call     tests/distributions/test_stable.py::test_mean[S-0.5-0.5-0.5]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-()-(3,)]
0.00s setup    tests/distributions/test_delta.py::TestDelta::test_mean_and_var
0.00s call     tests/distributions/test_stable.py::test_mean[S-0.5-0.0-0.5]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-(2, 1)-(3,)]
0.00s setup    tests/nn/test_module.py::test_cache
0.00s call     tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape1]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_beta_binomial_dependent_sample
0.00s setup    tests/infer/test_smcfilter.py::test_likelihood_ratio
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-0-3]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0-1.0]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-(2, 1)-(3,)]
0.00s setup    tests/nn/test_module.py::test_mixin_factory
0.00s setup    tests/infer/test_valid_models.py::test_enum_recycling_plate
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-scaled]
0.00s setup    tests/test_examples.py::test_cpu[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -m=1 --enum]
0.00s call     tests/distributions/test_stable.py::test_mean[S-1.0-0.0-1.5]
0.00s setup    tests/poutine/test_poutines.py::ConditionHandlerTests::test_stack_success
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-(2, 1)-(3,)]
0.00s setup    tests/nn/test_module.py::test_torch_serialize_decorators
0.00s setup    tests/nn/test_module.py::test_sample
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-ssmgp-3-1.5]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Binary_1D]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-(5,)-()-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-(2, 1)-(3,)]
0.00s setup    tests/nn/test_module.py::test_names
0.00s setup    tests/contrib/gp/test_kernels.py::test_transforming
0.00s setup    tests/test_examples.py::test_cpu[rsa/schelling.py --num-samples=10]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-5]
0.00s setup    tests/contrib/autoname/test_named.py::test_named_object
0.00s setup    tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=2 --jit]
0.00s setup    tests/poutine/test_poutines.py::EscapeHandlerTests::test_trace_compose
0.00s setup    tests/ops/test_special.py::test_safe_log
0.00s setup    tests/infer/test_inference.py::NormalNormalTests::test_elbo_nonreparameterized
0.00s setup    tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -hfm=3]
0.00s setup    tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 4]
0.00s call     tests/distributions/test_stable.py::test_mean[S-0.5-0.9-1.5]
0.00s call     tests/distributions/test_stable.py::test_mean[S-0.5-0.9-0.5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-2-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-()-(3,)]
0.00s setup    tests/test_examples.py::test_cpu[contrib/forecast/bart.py --num-steps=2 --stride=99999]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2]
0.00s call     tests/distributions/test_stable.py::test_mean[S-1.0-0.9-1.5]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(2, 3)-2]
0.00s setup    tests/optim/test_optim.py::test_dctadam_param_subsample
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-(2, 1)-(3,)]
0.00s setup    tests/test_examples.py::test_cpu[minipyro.py --backend=pyro]
0.00s setup    tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 1 --tmc --tmc-num-samples=2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(2, 3)-3]
0.00s setup    tests/ops/test_stats.py::test_quantile
0.00s setup    tests/infer/test_valid_models.py::test_plate_wrong_size_error
0.00s teardown tests/distributions/test_spanning_tree.py::test_make_complete_graph[cpp-2-expected_grid0]
0.00s setup    tests/test_examples.py::test_cpu[eight_schools/svi.py --num-epochs=1]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ExtendedBinomial]
0.00s setup    tests/poutine/test_poutines.py::QueueHandlerDiscreteTest::test_queue_enumerate
0.00s setup    tests/infer/test_valid_models.py::test_tail_adaptive_ok
0.00s setup    tests/infer/test_jit.py::test_simple
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-parallel-300]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 -hfm=3]
0.00s setup    tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-10]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-(5,)-()-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_csis.py::test_csis_validation_batch
0.00s setup    tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1]
0.00s setup    tests/params/test_param.py::test_dict_interface
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-(2, 1)-(3,)]
0.00s setup    tests/poutine/test_poutines.py::QueueHandlerMixedTest::test_queue_single
0.00s setup    tests/infer/test_enum.py::test_elbo_dbn_growth
0.00s setup    tests/contrib/oed/test_ewma.py::test_ewma_log_with_s
0.00s setup    tests/test_examples.py::test_cpu[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -p=10000 --sequential]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-(5,)-()-()-()]
0.00s setup    tests/poutine/test_poutines.py::LiftHandlerTests::test_memoize
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_valid_models.py::test_tail_adaptive_warning
0.00s setup    tests/distributions/test_empirical.py::test_mean_var_non_nan
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-Size-sample_shape0]
0.00s call     tests/distributions/test_stable.py::test_mean[S-0.5-0.5-1.5]
0.00s call     tests/distributions/test_shapes.py::test_normal_log_prob_shape
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-(3,)-()]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -ss=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --svi]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(5,)-2]
0.00s setup    tests/poutine/test_poutines.py::ConditionHandlerTests::test_condition
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-(3,)-()]
0.00s setup    tests/distributions/test_shapes.py::test_normal_shape
0.00s setup    tests/distributions/test_one_hot_categorical.py::TestOneHotCategorical::test_support_non_vectorized
0.00s setup    tests/test_examples.py::test_cpu[contrib/autoname/tree_data.py --num-epochs=1]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape0-event_shape1]
0.00s setup    tests/distributions/test_util.py::test_sum_leftmost
0.00s setup    tests/distributions/test_shapes.py::test_zip_shape
0.00s setup    tests/distributions/test_shapes.py::test_dirichlet_shape
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-True-0-1000-False-2]
0.00s setup    tests/distributions/test_shapes.py::test_diag_normal_log_prob_shape
0.00s setup    tests/test_examples.py::test_cpu[smcfilter.py --num-timesteps=3 --num-particles=10]
0.00s setup    tests/test_examples.py::test_cpu[sir_hmc.py -t=2 -w=2 -n=4 -d=100 -p=10000 -f 2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-5]
0.00s setup    tests/test_examples.py::test_jit[contrib/epidemiology/regional.py --jit -ss=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --svi]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -e=2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-10]
0.00s call     tests/ops/test_contract.py::test_ubersum_size_error[naive_ubersum-ab,bc->-shapes0-]
0.00s setup    tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full_hide
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_mask.py::test_kl_divergence
0.00s setup    tests/contrib/randomvariable/test_random_variable.py::test_subtract
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-1-1]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=0]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-10]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(2, 3)-2]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-(2, 1)-(3,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-full-nonreparam]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-nonreparam-scaled]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_spline_autoregressive
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(5,)-2]
0.00s setup    tests/poutine/test_mapdata.py::test_nested_iplate
0.00s setup    tests/infer/test_valid_models.py::test_plate_enum_discrete_batch_ok
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-1-1]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -c=2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-2]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Beta]
0.00s call     tests/distributions/test_delta.py::test_shapes[False-2-2]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-(2, 1)-(3,)]
0.00s setup    tests/contrib/autoname/test_named.py::test_named_list
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-(5,)-()-()-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-(5,)-()-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-3-7]
0.00s setup    tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_in_model_plate_ok
0.00s setup    tests/infer/test_enum.py::test_elbo_enumerate_plate_5
0.00s setup    tests/contrib/randomvariable/test_random_variable.py::test_chaining
0.00s setup    tests/poutine/test_poutines.py::LiftHandlerTests::test_random_module
0.00s call     tests/distributions/test_delta.py::test_shapes[False-3-1]
0.00s call     tests/contrib/randomvariable/test_random_variable.py::test_neg
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-True-0-1000-False-2]
0.00s setup    tests/infer/test_inference.py::TestFixedModelGuide::test_guide_and_model_free
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_lower_cholesky_affine
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_elu
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --haar]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-4]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-()-2]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-(2, 1)-(3,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-subsample-reparam]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-10]
0.00s setup    tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full_expose
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_overwrite
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-()]
0.00s setup    tests/poutine/test_poutines.py::TraceHandlerTests::test_trace_return
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-()-(3,)]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_householder
0.00s setup    tests/distributions/test_shapes.py::test_bernoulli_log_prob_shape
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-1-1]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 --haar]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-2]
0.00s call     tests/distributions/test_unit.py::test_expand[batch_shape0-sample_shape0]
0.00s setup    tests/ops/test_stats.py::test_autocovariance
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-(2, 1)-(3,)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_sequential_in_model_error
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_spline
0.00s setup    tests/test_examples.py::test_jit[sir_hmc.py -t=2 -w=2 -n=4 -p=10000 --jit]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-10]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-True-0-1000-False-2]
0.00s setup    tests/contrib/test_util.py::test_rtril
0.00s teardown tests/test_examples.py::test_cpu[neutra.py -n 10 --num-warmup 10 --num-samples 10]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_merge_points_small
0.00s setup    tests/contrib/randomvariable/test_random_variable.py::test_multiply_divide
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -nb=4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(4,)]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape0-event_shape0]
0.00s setup    tests/infer/test_valid_models.py::test_enum_in_model_multi_scale_error
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-3-1]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape1-event_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-1-1]
0.00s setup    tests/ops/test_stats.py::test_split_gelman_rubin_agree_with_gelman_rubin
0.00s setup    tests/distributions/test_delta.py::TestDelta::test_log_prob_sum
0.00s setup    tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 2 --tmc --tmc-num-samples=2]
0.00s setup    tests/infer/reparam/test_conjugate.py::test_beta_binomial_hmc
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-1-7]
0.00s setup    tests/contrib/autoname/test_named.py::test_nested
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -e=2 -k=1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-3]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape1-event_shape1]
0.00s setup    tests/ops/test_stats.py::test_hpdi
0.00s setup    tests/contrib/randomvariable/test_random_variable.py::test_tensor_ops
0.00s setup    tests/contrib/randomvariable/test_random_variable.py::test_abs
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -e=2 -k=1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-6]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-(2, 3)-(2, 3)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-4]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-(3,)-()]
0.00s setup    tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -o=0.2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-6]
0.00s setup    tests/ops/test_stats.py::test_pi
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_jit.py::test_scatter
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-3-7]
0.00s setup    tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 3]
0.00s setup    tests/poutine/test_poutines.py::test_plate_error_on_enter
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-subsample-nonreparam]
0.00s setup    tests/infer/test_enum.py::test_elbo_scale
0.00s setup    tests/test_examples.py::test_cpu[contrib/cevae/synthetic.py --num-epochs=1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-5]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-(2, 1)-(3,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-(4,)-3-7]
0.00s setup    tests/test_examples.py::test_cpu[contrib/autoname/scoping_mixture.py --num-epochs=1]
0.00s setup    tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-4]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_in_model_plate_reuse_ok
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-300-parallel-None]
0.00s setup    tests/test_examples.py::test_cpu[contrib/timeseries/gp_models.py -m imgp --test --num-steps=2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-3]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-(2, 3)-(2, 3)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-3]
0.00s setup    tests/poutine/test_poutines.py::test_decorator_interface_primitives
0.00s setup    tests/poutine/test_poutines.py::ConditionHandlerTests::test_stack_overwrite_behavior
0.00s setup    tests/ops/test_jit.py::test_varying_unhashable_kwargs
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-()-(3,)]
0.00s setup    tests/contrib/autoname/test_scoping.py::test_mutual_recur
0.00s teardown tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-6]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-1-1]
0.00s setup    tests/infer/test_smcfilter.py::test_gaussian_filter
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-6]
0.00s setup    tests/poutine/test_poutines.py::LiftHandlerTests::test_random_module_prior_dict
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-()-(3,)]
0.00s setup    tests/contrib/gp/test_models.py::test_inference_deepGP
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-1-1]
0.00s setup    tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 5 --tmc --tmc-num-samples=2]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape0-event_shape1]
0.00s setup    tests/poutine/test_poutines.py::LiftHandlerTests::test_random_module_warn
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-(2, 3)-(2, 3)]
0.00s setup    tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-2]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-()-(3,)]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_discrete_cosine
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-2-1]
0.00s teardown tests/test_examples.py::test_jit[baseball.py --num-samples=200 --warmup-steps=100 --jit]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-5]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-(2, 1)-(3,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-unscaled]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_batchnorm
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-6]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-2-7]
0.00s setup    tests/ops/test_stats.py::test_waic
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-(3,)-()]
0.00s setup    tests/infer/test_jit.py::test_scatter_workaround
0.00s setup    tests/contrib/gp/test_kernels.py::test_combination
0.00s teardown tests/test_examples.py::test_jit[eight_schools/mcmc.py --num-samples=500 --warmup-steps=100 --jit]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-5]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-()-()-(5,)-()]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_conditional_radial
0.00s setup    tests/distributions/test_reshape.py::test_independent_entropy
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-3-1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-(5,)-()-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-()-(3,)]
0.00s setup    tests/infer/test_jit.py::test_masked_fill
0.00s setup    tests/test_examples.py::test_jit[contrib/gp/sv-dkl.py --epochs=1 --num-inducing=4 --jit]
0.00s teardown tests/test_examples.py::test_jit[contrib/autoname/mixture.py --num-epochs=1 --jit]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-10]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-()-(3,)]
0.00s setup    tests/infer/test_jit.py::test_multi_output
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-unscaled]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-7-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-2]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-3-7]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_conditional_generalized_channel_permute
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-1-7]
0.00s setup    tests/contrib/autoname/test_named.py::test_eval_str
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-3]
0.00s call     tests/infer/test_autoguide.py::test_empty_model_error
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape2-event_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-2-7]
0.00s setup    tests/poutine/test_poutines.py::test_trace_score_parts_err_msg
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-(2, 3)-(2, 3)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-2]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-(2, 3)-(2, 3)]
0.00s setup    tests/infer/mcmc/test_valid_models.py::test_log_prob_eval_iterates_in_correct_order
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-True-0-1000-False-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-3-1]
0.00s setup    tests/contrib/gp/test_kernels.py::test_active_dims_overlap_ok
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-2]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-(2, 3)-(2, 3)]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_chains': 2, 'mp_context': 'spawn'}-7-7]
0.00s setup    tests/contrib/autoname/test_scoping.py::test_no_param
0.00s teardown tests/test_examples.py::test_cpu[toy_mixture_model_discrete_enumeration.py  --num-steps=1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-2]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-(2, 1)-(3,)]
0.00s setup    tests/distributions/test_transforms.py::TransformTests::test_conditional_neural_autoregressive
0.00s setup    tests/contrib/gp/test_parameterized.py::test_inference
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-1-7]
0.00s setup    tests/contrib/autoname/test_scoping.py::test_only_withs
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape3-event_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-3-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-(2, 1)-(3,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-3-7]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape0-event_shape1]
0.00s setup    tests/poutine/test_poutines.py::BlockHandlerTests::test_block_partial_expose
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-(2, 3)-(2, 3)]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-1]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-()-2]
0.00s setup    tests/optim/test_multi.py::test_multi_optimizer_overlap_error
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-(2, 3)-(2, 3)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-unscaled]
0.00s setup    tests/contrib/autoname/test_scoping.py::test_simple_recur
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-()-Real()]
0.00s setup    tests/poutine/test_poutines.py::test_trace_log_prob_err_msg
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-6]
0.00s call     tests/distributions/test_categorical.py::test_support_dims[dim=2-probs0]
0.00s setup    tests/ops/test_jit.py::test_varying_kwargs
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_valid_models.py::test_enum_in_model_ok
0.00s teardown tests/distributions/test_mvt.py::test_log_prob[1-(4,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-3-7]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-4]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-(5,)-()-()-(4,)]
0.00s setup    tests/poutine/test_poutines.py::BlockHandlerTests::test_block_hide_fn
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-()-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-()-(3,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-3-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-1-7]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape2-event_shape0]
0.00s setup    tests/poutine/test_poutines.py::UnconditionHandlerTests::test_undo_uncondition
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-(3,)-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-3-1]
0.00s setup    tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=20]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape3-event_shape1]
0.00s setup    tests/poutine/test_poutines.py::EscapeHandlerTests::test_discrete_escape
0.00s setup    tests/poutine/test_poutines.py::ConditionHandlerTests::test_trace_data
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-(3,)-()]
0.00s setup    tests/distributions/test_util.py::test_sum_rightmost
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-1-2]
0.00s setup    tests/contrib/autoname/test_named.py::test_named_dict
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-3]
0.00s setup    tests/ops/test_stats.py::test_autocorrelation
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-()-(3,)]
0.00s setup    tests/nn/test_module.py::test_torch_serialize_attributes
0.00s setup    tests/poutine/test_poutines.py::QueueHandlerDiscreteTest::test_queue_max_tries
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-(2, 1)-(3,)]
0.00s setup    tests/nn/test_module.py::test_submodule_contains_torch_module
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceMeanField_ELBO-full-nonreparam]
0.00s setup    tests/contrib/tracking/test_hashing.py::test_lsh_remove
0.00s setup    tests/contrib/gp/test_models.py::test_inference_sgpr
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-(3,)-()]
0.00s setup    tests/nn/test_module.py::test_bayesian_gru
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-1-1]
0.00s setup    tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=3 --tmc --tmc-num-samples=2 --jit]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-6]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[RejectionExponential]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-(2, 3)-(2, 3)]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-3]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-5]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-(3,)-()]
0.00s teardown tests/test_examples.py::test_jit[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -m=1 --enum --jit]
0.00s teardown tests/test_examples.py::test_jit[contrib/cevae/synthetic.py --num-epochs=1 --jit]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=1]
0.00s setup    tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=21]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-10]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[VonMises]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-(3,)-()]
0.00s setup    tests/contrib/autoname/test_scoping.py::test_recur_multi
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-4]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-6]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-(3,)-()]
0.00s call     tests/ops/test_stats.py::test_gelman_rubin
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-unscaled]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-3-7]
0.00s setup    tests/contrib/autoname/test_scoping.py::test_nested_traces
0.00s setup    tests/contrib/test_util.py::test_tensor_to_dict
0.00s teardown tests/test_examples.py::test_jit[eight_schools/svi.py --num-epochs=1 --jit]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1239]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-5]
0.00s call     tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-()-Real()]
0.00s setup    tests/poutine/test_poutines.py::test_arg_kwarg_error
0.00s setup    tests/poutine/test_poutines.py::QueueHandlerDiscreteTest::test_queue_single
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-1-7]
0.00s setup    tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --tmc --tmc-num-samples=2]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-()-2]
0.00s call     tests/distributions/test_categorical.py::test_support_dims[dim=3-probs0]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-(3,)-()]
0.00s setup    tests/nn/test_module.py::test_nested
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-unscaled]
0.00s setup    tests/infer/test_autoguide.py::test_unpack_latent
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-()-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-()-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-()-(3,)]
0.00s setup    tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --num-iafs=1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-3-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-2-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-(2, 1)-(3,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2, 'num_chains': 2, 'mp_context': 'spawn'}-7-7]
0.00s setup    tests/contrib/autoname/test_scoping.py::test_basic_scope
0.00s teardown tests/test_examples.py::test_cpu[rsa/schelling_false.py --num-samples=10]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-5]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-()-(3,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-1-1]
0.00s setup    tests/contrib/test_util.py::test_rexpand
0.00s setup    tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=4 --tmc --tmc-num-samples=2 --jit]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-3-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-()-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-sequential-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-3-1]
0.00s setup    tests/test_examples.py::test_jit[baseball.py --num-samples=200 --warmup-steps=100 --jit]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-4]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-(2, 3)-(2, 3)]
0.00s setup    tests/infer/test_jit.py::test_grad
0.00s setup    tests/infer/test_inference.py::SafetyTests::test_duplicate_obs_name
0.00s setup    tests/distributions/test_categorical.py::TestCategorical::test_mean_and_var
0.00s setup    tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --jit]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-2]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[HalfCauchy]
0.00s setup    tests/nn/test_module.py::test_hierarchy_prior_cached
0.00s setup    tests/contrib/test_util.py::test_rdiag
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-2]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-()-(3,)]
0.00s setup    tests/infer/test_jit.py::test_scale_and_mask
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-full-nonreparam]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-3-2.5]
0.00s setup    tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=1 --tmc --tmc-num-samples=2 --jit]
0.00s teardown tests/test_examples.py::test_cpu[lda.py --num-steps=2 --num-words=100 --num-docs=100 --num-words-per-doc=8]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-()-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-1-7]
0.00s teardown tests/test_examples.py::test_jit[contrib/epidemiology/sir.py --jit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-()-()]
0.00s teardown tests/test_examples.py::test_cpu[rsa/schelling.py --num-samples=10]
0.00s setup    tests/test_examples.py::test_cpu[eight_schools/mcmc.py --num-samples=500 --warmup-steps=100]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape2-event_shape1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-()-(3,)]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-False-0-1000-False-2]
0.00s setup    tests/infer/test_jit.py::test_backward
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-3-2]
0.00s setup    tests/test_examples.py::test_cpu[toy_mixture_model_discrete_enumeration.py  --num-steps=1]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-()-()-(5,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-(2, 3)-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-3-7]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=1 --tmc --tmc-num-samples=2]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-(2, 1)-(3,)]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape0-0.5-0.6-1.1]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Multinomial]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1273]
0.00s setup    tests/poutine/test_poutines.py::IndirectLambdaHandlerTests::test_scale_factors
0.00s setup    tests/poutine/test_poutines.py::LiftHandlerTests::test_prior_dict
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-(2, 1)-(3,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-3-1]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-ssmgp-2-2.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-1-1]
0.00s teardown tests/test_examples.py::test_jit[minipyro.py --backend=pyro --jit]
0.00s teardown tests/test_examples.py::test_cpu[inclined_plane.py --num-samples=1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-(3,)-()]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-2-3-(4,)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-2-3-()]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-2-3-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceTMC_ELBO-subsample]
0.00s teardown tests/test_examples.py::test_jit[sir_hmc.py -t=2 -w=2 -n=4 -p=10000 --jit]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-ij->ij]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-()-(3,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-2-1-0-()-(3,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-3-7]
0.00s teardown tests/test_examples.py::test_jit[minipyro.py --jit]
0.00s setup    tests/test_examples.py::test_jit[dmm/dmm.py --num-epochs=1 --jit]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-(5,)-()-()-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-2-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-()-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-1-1]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape1-event_shape0]
0.00s call     tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-()-2]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-(2, 3)-(2, 3)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-2-3-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 2)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-2-3-()]
0.00s setup    tests/infer/test_inference.py::SafetyTests::test_extra_samples
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-parallel-parallel-None]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-(2, 1)-(3,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-1-2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-3-1]
0.00s call     tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[0-False]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape1-event_shape1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-1-1]
0.00s setup    tests/test_examples.py::test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide easy]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -nb=4]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-parallel-None-None]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6 --raftery-parameterization]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-3-1]
0.00s setup    tests/test_examples.py::test_cpu[inclined_plane.py --num-samples=1]
0.00s call     tests/distributions/test_delta.py::test_shapes[True-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-1-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-True-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-sequential-sequential-None]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-1-1]
0.00s setup    tests/test_examples.py::test_cpu[sparse_gamma_def.py --num-epochs=2 --eval-particles=2 --eval-frequency=1 --guide auto]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-3-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-()-()]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-dmgp-1-1.5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-1-1]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-2-3-(4,)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-2-3-()]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-dmgp-3-1.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-()-()]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceTMC_ELBO-subsample]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-1-1]
0.00s setup    tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=4]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-2-3-()]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-4]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-()-()]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-2-3-(3, 2)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-scaled]
0.00s teardown tests/test_examples.py::test_jit[contrib/epidemiology/regional.py --jit -ss=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --svi]
0.00s setup    tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1 --aux-loss]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Stable]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-2-3-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-()-()-(5,)-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-2-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-()-(3,)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-2-3-(4,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-sequential-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-3-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-(2, 1)-(3,)]
0.00s setup    tests/test_examples.py::test_jit[lda.py --num-steps=2 --num-words=100 --num-docs=100 --num-words-per-doc=8 --jit]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=2]
0.00s setup    tests/test_examples.py::test_cpu[vae/vae_comparison.py --num-epochs=1]
0.00s teardown tests/test_examples.py::test_cpu[minipyro.py --backend=pyro]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-()-()]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-2-1-7]
0.00s setup    tests/test_examples.py::test_jit[eight_schools/mcmc.py --num-samples=500 --warmup-steps=100 --jit]
0.00s teardown tests/test_examples.py::test_cpu[rsa/semantic_parsing.py --num-samples=10]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=0]
0.00s teardown tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 2 --tmc --tmc-num-samples=2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-2-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-None-parallel-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-1-7]
0.00s setup    tests/test_examples.py::test_cpu[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel]
0.00s call     tests/distributions/test_shapes.py::test_categorical_shape
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Bernoulli]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-1-7]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-None-sequential-None]
0.00s teardown tests/test_examples.py::test_jit[contrib/epidemiology/sir.py --jit -np=128 -ss=2 -n=4 -d=20 -p=1000 -f 2 --svi]
0.00s teardown tests/test_examples.py::test_cpu[smcfilter.py --num-timesteps=3 --num-particles=10]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-3-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-sequential-sequential-None]
0.00s teardown tests/test_examples.py::test_cpu[rsa/generics.py --num-samples=10]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -hfm=3]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-3-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-5-4-2-(2, 1)-(3,)]
0.00s setup    tests/test_examples.py::test_cpu[neutra.py -n 10 --num-warmup 10 --num-samples 10]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=0 --tmc --tmc-num-samples=2]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=7]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape1-event_shape0]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape3-event_shape1]
0.00s call     tests/distributions/test_delta.py::test_shapes[True-2-2]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-2-()-()]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-3-1.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-3-1]
0.00s teardown tests/test_examples.py::test_cpu[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -m=1 --enum]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=5]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-3-7]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-None-sequential-None]
0.00s setup    tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=7 --jit]
0.00s setup    tests/test_examples.py::test_cpu[vae/vae.py --num-epochs=1]
0.00s call     tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-3]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-2-()-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-parallel-parallel-None]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoNormal]
0.00s teardown tests/test_examples.py::test_cpu[rsa/hyperbole.py --price=10000]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-2-1]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-2-3-(4,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-subsample-nonreparam]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -e=2]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-3-(5,)]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-lcmgp-2-2.5]
0.00s setup    tests/test_examples.py::test_cpu[rsa/generics.py --num-samples=10]
0.00s teardown tests/test_examples.py::test_cpu[minipyro.py]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 -nb=4]
0.00s call     tests/distributions/test_delta.py::test_shapes[False-2-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-False-unscaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-sequential-None]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-dmgp-1-1.5]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -ss=2 -n=4 -d=20 -p=1000 -f 2 --svi]
0.00s call     tests/distributions/test_delta.py::test_shapes[False-1-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-2-()-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-sequential-None-None]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-dmgp-2-1.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-3-1]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --haar]
0.00s teardown tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 3 --tmc --tmc-num-samples=2]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-False-Size]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-None-parallel-None]
0.00s teardown tests/test_examples.py::test_cpu[sir_hmc.py -t=2 -w=2 -n=4 -d=100 -p=10000 -f 2]
0.00s teardown tests/test_examples.py::test_cpu[eight_schools/svi.py --num-epochs=1]
0.00s teardown tests/test_examples.py::test_cpu[contrib/oed/ab_test.py --num-vi-steps=10 --num-bo-steps=2]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1381]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-None-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-None-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-None-None-None]
0.00s teardown tests/test_examples.py::test_cpu[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -p=10000 --sequential]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -o=0.2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-2-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-2-()-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-False-0-1000-False-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-parallel-None-None]
0.00s setup    tests/test_examples.py::test_jit[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel --jit]
0.00s teardown tests/test_examples.py::test_jit[lkj.py --n=50 --num-chains=1 --warmup-steps=100 --num-samples=200 --jit]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=3]
0.00s teardown tests/test_examples.py::test_cpu[contrib/timeseries/gp_models.py -m lcmgp --test --num-steps=2]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape2-event_shape1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-(2, 1)-(3,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-sequential-parallel-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-3-7]
0.00s teardown tests/test_examples.py::test_cpu[contrib/timeseries/gp_models.py -m imgp --test --num-steps=2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-2-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-2-0-()-()]
0.00s teardown tests/test_examples.py::test_cpu[contrib/autoname/mixture.py --num-epochs=1]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape1-event_shape1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-3-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-2-1-()-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-4]
0.00s setup    tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=5 --jit]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6 --tmc --tmc-num-samples=2]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-()-()-(5,)-(4,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-None-sequential-None]
0.00s teardown tests/test_examples.py::test_cpu[eight_schools/mcmc.py --num-samples=500 --warmup-steps=100]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -k=1]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-lcmgp-3-1.5]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-dmgp-2-1.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-3-1]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 --haar]
0.00s teardown tests/test_examples.py::test_cpu[contrib/autoname/tree_data.py --num-epochs=1]
0.00s teardown tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 1 --tmc --tmc-num-samples=2]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges917]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-()-()-(5,)-()]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape2-event_shape1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-True-list]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-True-tuple]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-False-tuple]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-None-sequential-None]
0.00s setup    tests/test_examples.py::test_jit[hmm.py --num-steps=1 --truncate=10 --model=3 --jit]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=4 --tmc --tmc-num-samples=2]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 -hfm=3]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-False-list]
0.00s call     tests/distributions/test_delta.py::test_shapes[False-3-0]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-sequential-parallel-None]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-dmgp-3-1.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-1-7]
0.00s teardown tests/test_examples.py::test_cpu[contrib/autoname/scoping_mixture.py --num-epochs=1]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 2)]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-2-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-2-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-parallel-parallel-None]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-full-reparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-sequential-None-None]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/regional.py --nojit -ss=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --svi]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape2-event_shape0]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-sequential-sequential-None]
0.00s setup    tests/test_examples.py::test_cpu[contrib/gp/sv-dkl.py --epochs=1 --num-inducing=4 --batch-size=1000]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-parallel-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-parallel-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-None-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-None-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-parallel-None-None]
0.00s teardown tests/test_examples.py::test_cpu[einsum.py]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-True-Size]
0.00s call     tests/distributions/test_delta.py::test_shapes[True-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-2-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-Size-sample_shape0]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-2-1-()-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-None-None-None]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-2-1-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-None-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-parallel-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-parallel-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-sequential-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-sequential-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-sequential-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-1-7]
0.00s setup    tests/test_generic.py::test_trace_handler[pyro-logistic_regression]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-None-parallel-None]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[GammaPoisson]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-4]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-None-parallel-None]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-glgssm-1-3]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-3-1.5]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-lcmgp-2-2.5]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape3-event_shape0]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-()-(3,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-None-None-None]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-lcmgp-3-1.5]
0.00s teardown tests/test_examples.py::test_cpu[lkj.py --n=50 --num-chains=1 --warmup-steps=100 --num-samples=200]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges801]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-()-()-(5,)-(4,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-None-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-None-parallel-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-3-7]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape0-event_shape0]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-parallel-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-sequential-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-None-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-parallel-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-None-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-parallel-sequential-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-None-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-parallel-sequential-None]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-glgssm-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-3-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-sequential-parallel-None]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-glgssm-1-3]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-ssmgp-2-2.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-3-7]
0.00s teardown tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 5]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-2-1-()-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-None-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-sequential-sequential-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-None-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-1-1]
0.00s teardown tests/test_examples.py::test_cpu[contrib/epidemiology/sir.py --nojit -np=128 -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -c=2]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-2-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-()-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-None-sequential]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[jit-bernoulli-2]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=3 --tmc --tmc-num-samples=2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-None-None-parallel-None-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-3-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-3-1]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-()]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-Size-sample_shape2]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-sequential-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-parallel-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-sequential-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-None-sequential-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-3-1]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=2]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1033]
0.00s call     tests/nn/test_module.py::test_submodule_contains_torch_module
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-4]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-None-None-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-3-1]
0.00s teardown tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 4 --tmc --tmc-num-samples=2]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-False-0-1000-False-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-sequential-parallel-None]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape0-event_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-parallel-None-None]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoDelta]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-3-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-(2, 1)-(3,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-parallel-None]
0.00s teardown tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 5 --tmc --tmc-num-samples=2]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-False-0-1000-False-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-parallel-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-sequential-None-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-3-7]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[FoldedNormal]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-parallel-sequential-None]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-3-7]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=4]
0.00s teardown tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 3]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape3-event_shape0]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-5-4-2-(2, 3)-(2, 3)]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-centered3-()]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-()-(5,)-()-(3, 2)]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(4,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceGraph_ELBO-subsample-reparam]
0.00s call     tests/ops/test_packed.py::test_broadcast_all[shapes1]
0.00s call     tests/ops/test_packed.py::test_unpack_pack[]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-2-()-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-None-None-None]
0.00s teardown tests/distributions/test_rejector.py::test_exponential_elbo[1.0-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-2-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-1-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-()-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-False-0-1000-False-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-list-sample_shape1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-()-(3,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges830]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape0-event_shape1]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-None-None-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges453]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-2-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-3-7]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape0-event_shape0]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape1-event_shape0]
0.00s setup    tests/test_examples.py::test_jit[air/main.py --num-steps=1 --jit]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges424]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-()-()-(5,)-()]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape2-event_shape0]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-None-batch_shape5]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-2]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-(2, 3)-(2, 3)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-1-1]
0.00s call     tests/params/test_module.py::test_param_no_grad[outest]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-()-(3,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-True-unscaled]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-3-7]
0.00s teardown tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 2]
0.00s call     tests/params/test_module.py::test_module_sequential[nn_module0]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape1-event_shape0]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-3-1]
0.00s call     tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[0-True]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-ClippedAdam-optim_kwargs1-py]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-(2, 1)-(3,)]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-2-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-1-1]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape3-event_shape0]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-5-4-2-(2, 1)-(3,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-True-scaled]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-3-1]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-Adam-optim_kwargs0-jit]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-None-batch_shape3]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-(2, 3)-(2, 3)]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-None-None]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges47]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-(2, 1)-(3,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-None-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-1-7]
0.00s teardown tests/test_examples.py::test_cpu[capture_recapture/cjs.py --num-steps=1 -m 4]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-3-3-()-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-3-3-3-()-(3,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-parallel-None]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-3-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges888]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges192]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-(2, 3)-(2, 3)]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-False-0-1000-False-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-sequential-None]
0.00s teardown tests/distributions/test_rejector.py::test_exponential_elbo[1.0-1.0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-3-7]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-5-4-2-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-2-1-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-5-4-2-(2, 1)-(3,)]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-3]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-full-reparam]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-()-(3,)]
0.00s teardown tests/distributions/test_rejector.py::test_exponential_elbo[1.0-2.0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-(4,)-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges627]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-5-4-2-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-5-4-2-(2, 3)-(2, 3)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges221]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape3-event_shape1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-(2, 3)-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-2-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1120]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges482]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1239]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges598]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-1-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-(2, 3)-(2, 3)]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-sequential-parallel-None]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-3-7]
0.00s setup    tests/test_generic.py::test_model_sample[minipyro-logistic_regression]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges337]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges18]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape2-event_shape0]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-3-2-1-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-(3,)-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(4,)-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges540]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(3, 2)-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges105]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Geometric]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-()-(3,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[cold-None-10-None-10-1]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-2-1]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[GammaPoisson]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-sequential-None-None]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-2-0.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1062]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges743]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges395]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges76]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-(3,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1149]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges946]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges308]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-3-3-()-(3,)]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-3-7]
0.00s setup    tests/test_generic.py::test_trace_handler[minipyro-eight_schools]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1236]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges714]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-(2, 3)-(2, 3)]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-None-None]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1439]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-Adam-optim_kwargs0-py]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-5-4-2-(2, 3)-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-2-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-3-1]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-()-()-(5,)-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[RejectionExponential]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-2-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-()-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-2-1-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(3, 2)]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[jit-studentt-2]
0.00s setup    tests/test_generic.py::test_mcmc_interface[pyro-neals_funnel]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1265]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1178]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-3-3-3-(2, 1)-(3,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-full-nonreparam]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges569]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-list-sample_shape0]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-3-3-3-(3,)-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-3-3-3-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-(5,)-()-()-(3, 2)]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-None-sequential-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges279]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8--1.0]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-2-1]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-2-0.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1352]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-->]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_overwrite
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(4,)-3-1]
0.00s call     tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape3-event_shape0]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-False-sample_shapes1]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-None-parallel-None]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Stable]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-tuple-sample_shape1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-(2, 3)-(2, 3)]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-sequential-None]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-2-3-MixtureOfDiagNormalsSharedCovariance]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-1-7]
0.00s teardown tests/test_examples.py::test_jit[air/main.py --num-steps=1 --jit]
0.00s call     tests/distributions/test_delta.py::test_shapes[True-2-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-3-2-1-()-()]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-1-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-False-initial_shape2-proposed_shape2]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-list-sample_shape2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(3, 2)-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1091]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges366]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[2-1.0]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-(2, 3)-(2, 3)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(3, 2)-3-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges134]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-False-unscaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-parallel-None-None]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-1-7]
0.00s setup    tests/test_examples.py::test_cpu[air/main.py --num-steps=1 --no-baseline]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1294]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges603]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-2-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-None-sequential]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges685]
0.00s call     tests/distributions/test_shapes.py::test_normal_shape
0.00s call     tests/distributions/test_delta.py::TestDelta::test_log_prob_sum
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTraceEnum_ELBO-full-nonreparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-None-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-parallel-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-3-7]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Beta]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-3-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-tuple-sample_shape0]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-3]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-3-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1004]
0.00s call     tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-tuple-sample_shape2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges975]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges656]
0.00s call     tests/distributions/test_delta.py::test_shapes[False-2-0]
0.00s call     tests/distributions/test_delta.py::test_shapes[False-1-0]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-3-7]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-3]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-None-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-None-None-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-False-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1410]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1323]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-()-()-(5,)-(4,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-None-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-None-None-parallel-300]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges250]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[LowRankMultivariateNormal]
0.00s call     tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-False-sample_shapes0]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-3]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-None-sequential-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-3-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges772]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-ClippedAdam-optim_kwargs1-py]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-1-7]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-1-7]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-Adam-optim_kwargs0-py]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-True-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-parallel-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model4-True-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-1-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-subsample-nonreparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-sequential-sequential]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-3-1]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-3-2.5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-3-7]
0.00s setup    tests/test_generic.py::test_rng_seed[minipyro-logistic_regression]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-sequential-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-None-parallel-None]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-Size-sample_shape1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-1-1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-None-batch_shape2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-1-1]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-False-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-3-7]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_noprior-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-nonreparam-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-sequential-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-3-1]
0.00s call     tests/distributions/test_delta.py::test_shapes[True-2-0]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-3-1]
0.00s setup    tests/test_generic.py::test_trace_handler[minipyro-logistic_regression]
0.00s call     tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-2-3-MixtureOfDiagNormalsSharedCovariance]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-1-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-(3,)-()]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-6-()-1-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-(3,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-1-1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-list-sample_shape2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-3-2]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-4-3-GaussianScaleMixture]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(3, 2)-1-1]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-ClippedAdam-optim_kwargs1-jit]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-()-()]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-2-3-GaussianScaleMixture]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-1-7]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Uniform]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_optimizer[pyro-Adam-optim_kwargs0-jit]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-(2, 1)-(3,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-parallel-300]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-4-3-GaussianScaleMixture]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-3-1]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-parallel-parallel-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-(3, 2)-1-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-(3,)-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-4]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-1-1]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-4-3-MixtureOfDiagNormalsSharedCovariance]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-False-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-1-1]
0.00s setup    tests/test_generic.py::test_trace_handler[pyro-neals_funnel]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-1-0-0-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-(3,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-1-1]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-4-3-MixtureOfDiagNormals]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-(3,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-3-1]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-2-3-GaussianScaleMixture]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-4-3-MixtureOfDiagNormals]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-False-1-1-()-1-7]
0.00s setup    tests/test_generic.py::test_rng_state[minipyro-logistic_regression]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-False-unscaled]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-None-None]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-3-2]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-2-3-MixtureOfDiagNormals]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-2-3-MixtureOfDiagNormalsSharedCovariance]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-6-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-False-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-3-7]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(4,)-(4,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-3-1]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-2-3-MixtureOfDiagNormalsSharedCovariance]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-(2, 1)-(3,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-1-3]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-2.0-0.0-0.5]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[LinearHMM]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-3-2]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-4-3-MixtureOfDiagNormals]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-3-7]
0.00s call     tests/distributions/test_delta.py::test_shapes[False-0-0]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(5, 1)-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-(3,)-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-3]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-()-2]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-4-3-MixtureOfDiagNormalsSharedCovariance]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model3-True-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-3-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-(3,)-()]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-sequential-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.06-0.03-0.29-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-1-1]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-False-4-3-MixtureOfDiagNormals]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-2-3-MixtureOfDiagNormals]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-6]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[False-parallel-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-3-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-1-1]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-EnergyDistance_prior]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-4]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-None-parallel-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-parallel-300]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-3-1]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-sequential-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(6,)-()-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(5,)-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-1-1]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.06-0.04-0.29-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-(5, 1)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-3-1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-tuple-sample_shape0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-2-1-0-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-1-7]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[FoldedNormal]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(5, 1)-(6,)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(5, 1)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(5, 1)-(6,)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-1-7]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-(5, 1)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(5, 1)-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-(6,)-(6,)-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-()-(6,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-1-1]
0.00s setup    tests/test_generic.py::test_model_sample[minipyro-neals_funnel]
0.00s call     tests/distributions/test_delta.py::test_shapes[True-0-0]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-(3,)-()]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.04-0.04-0.29-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-()-(6,)-()-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-()-(6,)-(5, 1)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(6,)-()-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-(6,)-()-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-()-()-3-2]
0.00s setup    tests/test_examples.py::test_jit[dmm/dmm.py --num-epochs=1 --num-iafs=1 --jit]
0.00s call     tests/distributions/test_reshape.py::test_sample_shape_order
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-5]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_non_tree_elbo_gradient[True-parallel-0.32-0.04-0.03-0.29-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-()-()-()-(6,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-(6,)-(6,)-(6,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(5, 1)-(6,)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-()-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-3-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-1-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-None-None]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-(5,)-()-()-()-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-(6,)-(5, 1)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(5, 1)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-()-1-2]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-4-3-MixtureOfDiagNormalsSharedCovariance]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-False-2-3-MixtureOfDiagNormals]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-2-3-MixtureOfDiagNormals]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-1-7]
0.00s call     tests/poutine/test_poutines.py::test_plate_error_on_enter
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Cauchy]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(5, 1)-(6,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-(5, 1)-(6,)-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(6,)-(5, 1)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(5, 1)-(6,)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-(6,)-(6,)-(6,)-(6,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-3-7]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-True-scaled]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-(5,)-(6,)-()-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-(6,)-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-()-()]
0.00s teardown tests/contrib/gp/test_models.py::test_inference[SparseGPRegression_y2D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(4,)-3-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[jit-studentt-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-nonreparam-unscaled]
0.00s teardown tests/distributions/test_tensor_type.py::test_float_type
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[(5,)-(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(5, 1)-(6,)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-()-()-(6,)-(5, 1)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[full-()-()-()-(5, 1)-(6,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-1-1]
0.00s call     tests/distributions/test_delta.py::test_shapes[True-3-0]
0.00s call     tests/test_primitives.py::test_observe_warn
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-2-1-0-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-()-(3,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-2-()]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-4-3-MixtureOfDiagNormalsSharedCovariance]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-()-3-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-2-1-0-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-(6,)-()-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-2-(5,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-(4,)-1-7]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-3]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-3-7]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-sequential-sequential]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-()-(6,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-3-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-3-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-(3,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_shape[diag-(5,)-()-()-()-(6,)-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-6-(4,)-3-1]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=5 --tmc --tmc-num-samples=2]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-None-batch_shape5]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-2-1-0-()-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-3]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_shape[()-()-()-()-(6,)-()-1-2]
0.00s teardown tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-3-NcpContinuous]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-3-1]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[jit-laplace-2]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0--1.0]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape1-0.5-0.6-1.1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(4,)-3-7]
0.00s setup    tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=2 --tmc --tmc-num-samples=2]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-None-batch_shape4]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-False-unscaled]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-()-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-1-()-3-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-(3,)-()]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape0-0.5-0.6-0.4]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-3-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-(3,)-()]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-True]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-EnergyDistance_noprior]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-True-unscaled]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-1-0-()-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-2-1-0-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-True-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-1-0-(3, 2)-3-7]
0.00s setup    tests/test_generic.py::test_rng_seed[minipyro-neals_funnel]
0.00s call     tests/contrib/epidemiology/test_util.py::test_cat2_scalar[(2,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-()-(3,)]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-3]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-parallel-10000]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-1-1]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_gradient[OMTMultivariateNormal-1-sample_shape1-0.5-0.6-0.4]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(5,)-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model2-False-2-1-(3, 2)-3-1]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-2-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-1-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-1-1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-2]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-5]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-3-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-2-0-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-3-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-0-0-()-(3,)]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-4]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-None-sequential]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-2-(5,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-()-(3,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-1-1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-None-batch_shape4]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-4]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-True-unscaled]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-1-7]
0.00s setup    tests/test_examples.py::test_jit[contrib/autoname/mixture.py --num-epochs=1 --jit]
0.00s setup    tests/test_examples.py::test_cpu[sparse_regression.py --num-steps=2 --num-data=50 --num-dimensions 20]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-False-0-1000-False-1]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-3-(5,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-3-7]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-False-0-1000-False-1]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape0-0.5-0.6-1.1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-()-(5,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-3-1]
0.00s call     tests/params/test_module.py::test_param_no_grad[outer]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-2-1-0-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-0-2-0-()-()]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_gradient[AVFMultivariateNormal-1-sample_shape0-0.5-0.6-0.4]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-2-0-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-2-1-0-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-()-(3,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-None-sequential]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-3-7]
0.00s setup    tests/test_generic.py::test_mcmc_interface[pyro-beta_binomial]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-2-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-None-parallel]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges157]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-2-0-()-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-4]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-parallel-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-1-1]
0.00s call     tests/distributions/test_delta.py::test_shapes[True-1-0]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-1-0-0-()-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-parallel-sequential]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-parallel-parallel]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(3, 2)-1-7]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-3-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-1-0-0-()-()]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_predictive[False-100]
0.00s teardown tests/infer/test_inference.py::test_energy_distance_multivariate[0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-3-7]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=6 --tmc --tmc-num-samples=2]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-sequential-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-3-7]
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-ai->a-i]
0.00s call     tests/distributions/test_shapes.py::test_dirichlet_shape
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[LogNormal]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-4-1-0-0-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-1-3]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-parallel-None]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-10]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-3-7]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::ExponentialGammaTests::test_elbo_nonreparameterized
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-None-None-parallel-None]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-1-1]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Gamma]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-True-scaled]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-1-7]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[rsvi-ReweightedWakeSleep]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-3-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-None-sequential]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-3-(5,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-3-1]
0.00s call     tests/distributions/test_reshape.py::test_idempotent[1-2]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-2-0-()-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-nonfactorized-0-200-False-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-3-7]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-None-parallel]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-2-0-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(5,)-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-3-1]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_local_param_ok[minipyro-py]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-2-2]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_diagnostics[2]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-3]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-2.0-0.5-1.5]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-list-sample_shape0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-3-7]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-sequential-parallel]
0.00s teardown tests/distributions/test_mvn.py::test_shape[1-cov_shape0-loc_shape0]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-4000]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(2, 3)-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-1-7]
0.00s setup    tests/test_generic.py::test_rng_seed[pyro-logistic_regression]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-Size-sample_shape2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-False-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-1-1]
0.00s teardown tests/test_examples.py::test_cpu[contrib/gp/sv-dkl.py --binary --epochs=1 --num-inducing=4 --batch-size=1000]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-3-2-(4,)]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-None-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-parallel-parallel]
0.00s teardown tests/distributions/test_mvn.py::test_shape[1-cov_shape0-loc_shape2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-3-7]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[HalfCauchy]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-None-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-False-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-1-()-1-1]
0.00s call     tests/contrib/autoname/test_named.py::test_eval_str
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-1-0-0-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-3]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-sequential-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(3, 2)-3-1]
0.00s setup    tests/test_generic.py::test_trace_handler[pyro-eight_schools]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-2-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(4,)-3-7]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-sequential-parallel]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-2.0-0.5-0.5]
0.00s teardown tests/distributions/test_mvn.py::test_shape[1-cov_shape0-loc_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-3-(5,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(4,)-3-7]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-None-parallel]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-list-sample_shape1]
0.00s teardown tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-2-NcpContinuous]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(4,)-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-4]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-nonfactorized-0-200-False-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-None-parallel-sequential]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-1-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-scaled]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-6-(3, 2)-3-1]
0.00s teardown tests/test_examples.py::test_cpu[mixed_hmm/experiment.py --timesteps=1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-4]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-None-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-3-()]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-1000]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[svi-None-10-None-10-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-0-()-3-7]
0.00s setup    tests/test_generic.py::test_rng_state[pyro-eight_schools]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-()-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-2-2]
0.00s teardown tests/infer/test_inference.py::test_energy_distance_multivariate[1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-True-unscaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-parallel-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-1-()-1-7]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-3-0.37-0.51-0.33]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-2-0-()-3-7]
0.00s setup    tests/test_generic.py::test_rng_state[pyro-beta_binomial]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-False-1-6-(4,)-1-7]
0.00s setup    tests/test_examples.py::test_cpu[lkj.py --n=50 --num-chains=1 --warmup-steps=100 --num-samples=200]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-sequential-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-()-()]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-tuple-sample_shape1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(4,)-1-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-1-0-0-0-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-2-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-None-None-parallel]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-False-2-0-()-3-1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-False-0-1000-False-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model0-True-1-0-(4,)-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-()-(3,)]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-factorized-True-0-1000-False-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-None-sequential]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[jit-bernoulli-1]
0.00s teardown tests/test_examples.py::test_cpu[dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=20]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-factorized-True-0-1000-False-1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-5]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-3-2]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Poisson_2D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-5]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-3-2-(3, 2)]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(3, 2)-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-1-0-0-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-3-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-1-3]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-4-0.37-0.77-0.33]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-3-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-2-0-()-()]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-scaled]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-0-(4,)-3-1]
0.00s setup    tests/test_generic.py::test_model_sample[pyro-neals_funnel]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[GammaPoisson]
0.00s call     tests/contrib/tracking/test_measurements.py::test_PositionMeasurement
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-2-3]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-4-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(4,)-(5,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-3-1]
0.00s teardown tests/test_examples.py::test_cpu[contrib/gp/sv-dkl.py --epochs=1 --num-inducing=4 --batch-size=1000]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_plate_ok[pyro-py]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-1-0-0-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-2-3]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-3-0.37-0.77-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(5, 1)-()-()-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-3-1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[svi-Model1-True-2-6-(3, 2)-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-1-0-0-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-factorized-0-200-False-4]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-None-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-None-None]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoMultivariateNormal]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-3-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-1-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-2-1]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoMultivariateNormal]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-1-1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-False-tuple-sample_shape2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-3-7]
0.00s setup    tests/test_generic.py::test_rng_state[pyro-neals_funnel]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Normal]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_predictive[True-100]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-3-0.37-0.77-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-()-()-()-(6,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-3-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[warm-None-10-None-10-1]
0.00s setup    tests/test_generic.py::test_rng_state[pyro-logistic_regression]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[minipyro-jit]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-False-0-1000-False-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-unscaled]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-3-1]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional[3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1287]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-()-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-3-7]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-full-reparam-unscaled]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges513]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-2-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.6-1-1-2]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-prior-0-200-False-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[cold-14-1-7-1-1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape2]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_predictive[False-200]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(5, 6)-(5, 6)-(5, 6)-(5, 6)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(5, 1)-(6,)-()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-(6,)-(6,)-(6,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(6,)-()-()-()-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-()-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-1-2]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges216]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-3-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-14-1-7-1-1]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-None-10-None-10-1]
0.00s setup    tests/test_generic.py::test_model_sample[pyro-eight_schools]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-(3,)-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-2-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[(5,)-(6,)-()-()-()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-()-(6,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(3, 2)-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-1-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-0-0-0-()-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-factorized-0-200-False-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-()-()-()-(4,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(5, 1)-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-3-1]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional[5]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-1-7]
0.00s setup    tests/test_generic.py::test_rng_state[minipyro-neals_funnel]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-()-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[4-4-0-0-0-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-2-1]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-2.0-0.9-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-(3,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-4-0-0-0-(3,)-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-2-2]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist2]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(5, 1)-(4,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-()-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[svi-14-1-7-1-1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-3-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-3]
0.00s teardown tests/infer/test_util.py::test_psis_diagnostic[1-0.95-krange1-500000]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-3-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(6,)-(5, 1)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-()-(5, 1)-(6,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(5, 1)-(4,)-()-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-1-1]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_predictive[False-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-parallel-None]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-2.0-0.5-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-1-0-0-0-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-2-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-(5, 1)-()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-(5, 1)-(6,)-()-()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-(4,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-()-()-1-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-1-7]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-True-list]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_local_param_ok[minipyro-jit]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[4-4-0-0-0-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-0-0-0-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-2-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-3]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-3-2-(4,)]
0.00s teardown tests/infer/test_util.py::test_psis_diagnostic[5-0.95-krange1-250000]
0.00s teardown tests/infer/test_util.py::test_psis_diagnostic[1-0.5-krange0-500000]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-(5, 1)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-2-2]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional[0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-1-7]
0.00s setup    tests/test_generic.py::test_rng_seed[pyro-beta_binomial]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-1-1]
0.00s setup    tests/test_generic.py::test_rng_seed[minipyro-eight_schools]
0.00s teardown tests/test_examples.py::test_cpu[contrib/forecast/bart.py --num-steps=2 --stride=99999]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[skewed-1.9-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-1-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_independent_hmm_shape[()-()-(6,)-()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(5, 4)-(5, 4)-(5, 4)-(5, 4)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(5, 1)-(4,)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-(4,)-(4,)-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-1-1]
0.00s teardown tests/test_examples.py::test_cpu[hmm.py --num-steps=1 --truncate=10 --model=1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.9-2-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-(5, 1)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(5, 1)-(4,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-()-()-()-(4,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[(5,)-(4,)-()-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-(4,)-()-()-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-3-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-1-1]
0.00s setup    tests/test_generic.py::test_model_sample[pyro-beta_binomial]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-False-list]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-2.0-0.0-1.5]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(4,)-(5, 1)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(5, 1)-(4,)-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-()-(4,)-(5, 1)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-3-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-2-1]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_predictive[True-200]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-3-1]
0.00s call     tests/distributions/test_reshape.py::test_idempotent[1-0]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-(5, 1)-(4,)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-()-(4,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(3, 2)-(5,)]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_MultiClass_1D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-1-1]
0.00s call     tests/ops/einsum/test_adjoint.py::test_shape[map-i->i]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-()-()-(4,)-()-1-1]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional_whiten[5]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-None-1-None-1-8]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_distribution[symmetric-1.6-1-1-2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-False-unscaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-parallel-2000]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[(5,)-(4,)-()-()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_studentt_hmm_shape[()-(4,)-()-()-()-3-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-3-1]
0.00s setup    tests/test_generic.py::test_rng_state[minipyro-beta_binomial]
0.00s teardown tests/distributions/test_hmm.py::test_stable_hmm_shape[()-()-(4,)-(5, 1)-()-3-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-3-1]
0.00s setup    tests/test_generic.py::test_trace_handler[minipyro-neals_funnel]
0.00s setup    tests/test_generic.py::test_rng_seed[pyro-eight_schools]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-3-7]
0.00s teardown tests/infer/test_util.py::test_psis_diagnostic[5-0.5-krange0-500000]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-True-unscaled]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Poisson_2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-10-1-None-3-5]
0.00s setup    tests/test_generic.py::test_mcmc_interface[pyro-eight_schools]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1223]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-3-2-(3, 2)]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[2-None-parallel]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-2.0-0.9-1.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-()-(5,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-1-7]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[cold-None-1-None-1-8]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[jit-normal-2]
0.00s setup    tests/test_generic.py::test_model_sample[minipyro-beta_binomial]
0.00s setup    tests/test_generic.py::test_model_sample[minipyro-eight_schools]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-2-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-3-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1137]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1132]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(6),:,z(5,1)]-(5, 6, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(4,)]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-2.0--0.5-1.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-2-2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges870]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-3-2-(3, 2)]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-parallel-parallel-None]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/contrib/gp/test_models.py::test_gplvm[VariationalSparseGP_y1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-1-7]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5),z(5)]-(5, 2)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-unscaled]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-1-1]
0.00s setup    tests/test_generic.py::test_rng_seed[minipyro-beta_binomial]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-3-1]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-3-2-(4,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-unscaled]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-scaled]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(4,)-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-3-1]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-2-200]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1090]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-3-1]
0.00s setup    tests/test_generic.py::test_rng_seed[pyro-neals_funnel]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-(5,)-()-()-(4,)]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional_whiten[0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-3-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-(5,)-()-()-(4,)]
0.00s teardown tests/infer/test_util.py::test_psis_diagnostic[5-0.5-krange0-250000]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges70]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(4,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-1-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-(5,)-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-prior-0-200-False-1]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-3-7]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_plate_ok[pyro-jit]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-nonreparam-unscaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-sequential-parallel-None]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[warm-None-1-None-1-8]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-(5,)-()-()-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-5]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-1000]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalGP_y2D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-7-7]
0.00s setup    tests/test_generic.py::test_rng_state[minipyro-eight_schools]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges841]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-(5,)-()-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-(5,)-()-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-3-2-(3, 2)]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[svi-None-5-10-1-5]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[cold-10-1-None-3-5]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-1-7]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-None-5-10-1-5]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[minipyro-jit]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(3, 2)-(5,)]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-3-7]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[svi-10-1-None-3-5]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5,1,1)]-(5, 2, 3)]
0.00s call     tests/distributions/test_reshape.py::test_idempotent[2-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 4, 1)-(1, 4, 5)-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-prior-0-200-False-1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-factorized-0-200-False-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-True-unscaled]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-1-1]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[hmc-7-1-7-1-7]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[warm-None-5-10-1-5]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-1-200]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5,1),:]-(5, 2, 4)]
0.00s call     tests/distributions/test_reshape.py::test_idempotent[2-2]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_plate_ok[minipyro-jit]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[1-None]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(4,)]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5,1),:,z(6)]-(5, 6, 3)]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_local_param_ok[pyro-jit]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-3-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-1-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-1-7]
0.00s call     tests/distributions/test_reshape.py::test_idempotent[1-1]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_plate_ok[minipyro-py]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-4-0.37-0.51-0.33]
0.00s teardown tests/contrib/gp/test_models.py::test_gplvm[GPRegression_y2D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-1-1]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape1-sample_shape1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-(5,)-()-()-(3, 2)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-scaled]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-()-()]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[jit-normal-1]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5,1),z(6)]-(5, 6, 2)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-True-scaled]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoMultivariateNormal]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(4,)-(5,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-1-1]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[svi-7-1-7-1-7]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),:,z(5)]-(5, 3)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-(5,)-()-()-()-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Uniform]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-(5,)-()-(4,)]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-4-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-4-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-parallel-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(3, 2)-()]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[cold-7-1-7-1-7]
0.00s call     tests/distributions/test_reshape.py::test_idempotent[2-0]
0.00s call     tests/test_primitives.py::test_sample_ok
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-(5,)-()-()-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-parallel-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-sequential-parallel-None]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[cold-None-5-10-1-5]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Geometric]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[pyro-jit]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-3-2-(4,)]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-3-0.37-0.77-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(3, 2)-(5,)]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-0-3]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[minipyro-py]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_nested_plate_plate_ok[pyro-py]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[pyro-jit]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-(5,)-()-()]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-3-(5,)]
0.00s setup    tests/test_generic.py::test_trace_handler[pyro-beta_binomial]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-parallel-sequential-None]
0.00s setup    tests/test_generic.py::test_trace_handler[minipyro-beta_binomial]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-3-2-(3, 2)]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_poisson[svi-None-1-None-1-8]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2,1)]-(2, 3)]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-3-0.37-0.77-0.33]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.7--0.5-0.5]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-Size-sample_shape1]
0.00s setup    tests/test_generic.py::test_model_sample[pyro-logistic_regression]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-4]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-4-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-4-0.37-0.51-0.33]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,:,z(5)]-(5, 2, 3)]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_local_param_ok[pyro-py]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-full-reparam-False-scaled]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-4-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-5-3-0.37-0.77-0.33]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(3, 2)-(5,)]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-3-2-7]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5),:]-(5, 2, 4)]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_constraints[minipyro-jit]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-4]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-4]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-3-0.37-0.77-0.33]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.7-0.9-1.5]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-1000]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[exponential-2-200]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges811]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-True-scaled]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-3-0.37-0.77-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-3-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-()-(5,)]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges806]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-()-(4,)]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-4]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-3-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-sequential-parallel-None]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges217]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-False-Size]
0.00s teardown tests/infer/reparam/test_neutra.py::test_reparam_log_joint[dirichlet_categorical-kwargs1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-False-factorized-0-200-False-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-nonreparam-scaled]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(4,)-()]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-list-sample_shape2]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2),:]-(2, 4)]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(3)]-(2, 3)]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(2)]-(2,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-()-()]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-False-tuple]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-False-0-1000-False-1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-5]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-nonreparam-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-sequential-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-sequential-sequential-None]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-7-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1167]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[pyro-py]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-5]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-4-0.37-0.77-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-4-0.37-0.77-0.33]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional_whiten[3]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges756]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[mixture-True-nonfactorized-0-200-False-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-nonreparam-scaled]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-5-4-0.37-0.77-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-sequential-6-4-0.37-0.51-0.33]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Exponential]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-()-()]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-3-0.37-0.77-0.33]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional[4]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[laplace-1-200]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-(5,)-()-(4,)]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_constraints[pyro-jit]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[True-parallel-6-4-0.37-0.77-0.33]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-2.0-0.9-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(4,)-(5,)]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional[2]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:,z(4)]-(4, 2)]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-True-Size]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_nonempty_model_empty_guide_ok[minipyro-py]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-False-scaled]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-nonreparam-unscaled]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-True-unscaled]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-4-0.37-0.77-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-5-4-0.37-0.77-0.33]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-2.0--0.5-0.5]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges193]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-(5,)-()-()-()-()-(4,)]
0.00s call     tests/distributions/test_distributions.py::test_distribution_validate_args[True-Gamma-args1]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_constraints[pyro-py]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-2]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-4-0.37-0.77-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-5-3-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(3, 2)-()]
0.00s teardown tests/test_examples.py::test_cpu[air/main.py --num-steps=1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-True-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-parallel-parallel-sequential-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-2-(2, 3)]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2,3)]-(2, 3)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-nonreparam-unscaled]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-parallel-6-3-0.37-0.51-0.33]
0.00s teardown tests/infer/test_enum.py::test_bernoulli_pyramid_elbo_gradient[False-sequential-6-3-0.37-0.51-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-2-(2, 3)]
0.00s teardown tests/contrib/oed/test_linear_models_eig.py::test_nmc_eig_linear_model
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges613]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_plate_ok[TraceEnum_ELBO]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-False-prior-0-200-False-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-full-reparam-False-unscaled]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-2.0-0.0-0.5]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[laplace-1-100]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges26]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-sequential-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-sequential-sequential-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.25]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-3-(2, 3)]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges97]
0.00s setup    tests/test_examples.py::test_cuda[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel --cuda]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-(2, 1)-(3,)]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.7-0.5-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-3-(2, 3)]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-sequential-sequential-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.75]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.7-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.7-0.5-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.7-0.0-1.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-()-2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-3-(2, 3)]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-tuple-sample_shape2]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-0-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-(5,)-()-()-()-()-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Poisson]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_constraints[minipyro-py]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-TraceTMC_ELBO]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.55-0.44]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-parallel-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-parallel-sequential-parallel-sequential-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-()-2]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 16}-0-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1433]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-()-(3,)]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-nonfactorized-True-0-1000-False-1]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9--0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-2-(2, 3)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-full-reparam-False-scaled]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-2.0-0.0-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.7-0.5-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-()-2]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-2.0-0.9-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.7--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.7-0.9-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-()-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-(5,)-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_noprior-scale2]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-mask3]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.7-0.0-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.7-0.5-1.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(5,)-3]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[jit-laplace-1]
0.00s call     tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-True-tuple]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-3]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_prior-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-sequential-parallel-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.25]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.7-0.0-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-()-3]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[jit-exponential-2]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-2-(4,)-()]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-EnergyDistance_noprior]
0.00s teardown tests/infer/test_jit.py::test_svi[JitTraceMeanField_ELBO-10]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-parallel-sequential-None]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.7-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-2.0--0.5-0.5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-()-3]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Binary_2D]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[python-bernoulli-2]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-(5,)-()-(4,)]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=3 --tmc --tmc-num-samples=2 --cuda]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-(3,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-(5,)-()-()-()-()-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-False-0-1000-False-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-False-scaled]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-14]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-()-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-EnergyDistance_prior]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_gradient[diagonal-True-nonfactorized-0-200-False-1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-1.0]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.7--0.5-1.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-3-(5,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-(5,)-(3, 2)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-(3,)-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-True-unscaled]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape0-df_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(2, 3)-2]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[studentt-1-200]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[laplace-2-100]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[exponential-2-100]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_logistic_regression[None-1-None-True-True-True]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-prior-True-0-1000-False-1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-2.0--0.5-1.5]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-()-3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-list-sample_shape1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-3-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-7-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-(5,)-()-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-()-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-(5,)-()-()-()-()-(4,)]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_logistic_regression[0.0855-None-4-False-True-False]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceTMC_ELBO-subsample]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-()-2]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1428]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.0-()]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_noprior-0.1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[mixture-prior-True-0-1000-False-1]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-2.0-0.5-1.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-()-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-()-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-()-2]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-4000]
0.00s teardown tests/contrib/gp/test_models.py::test_hmc[SparseGPRegression_y1D]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[exponential-1-100]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.25]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-()-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(2, 3)-3]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional[6]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional[1]
0.00s teardown tests/test_generic.py::test_trace_handler[minipyro-eight_schools]
0.00s teardown tests/test_generic.py::test_rng_state[minipyro-eight_schools]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-parallel-sequential-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(5,)-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-3]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-(5,)-()-()-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-(5,)-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-4]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-TraceTMC_ELBO]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-unscaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-sequential-parallel-parallel-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-()-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(5,)-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-0-3]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_logsumexp[1-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-1]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_logistic_regression[None-1-None-True-True-False]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_logistic_regression[None-1-None-True-False-False]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_noprior-mask2]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-2]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-1.0]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.7-0.9-1.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(2, 3)-2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-list-sample_shape1]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-2-100]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-(5,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-(5,)-()-()-()-()-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceEnum_ELBO-subsample]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-()-2]
0.00s teardown tests/contrib/gp/test_models.py::test_model[GPRegression_y1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-1-1]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[exponential-1-200]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1095]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[1-1.0]
0.00s teardown tests/ops/test_gaussian.py::test_gaussian_tensordot[1-1-3-3-3-(2, 1)-(3,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-(5,)-()]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.0-()]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-False-scaled]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.75]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(5,)-2]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-7]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[1-0.1]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-None-parallel]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5--0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(5,)-2]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-4000]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 12}-7-3]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_gaussian_tensordot[1-1-3-3-3-(2, 3)-(2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-(5,)-()-()-()-()-()]
0.00s teardown tests/infer/test_util.py::test_psis_diagnostic[1-0.95-krange1-250000]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-sequential-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-None-parallel]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-19]
0.00s teardown tests/test_generic.py::test_rng_state[minipyro-beta_binomial]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5,1),0]-(5, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-(5,)-()-(4,)]
0.00s teardown tests/infer/reparam/test_haar.py::test_normal[(2, 5)--1-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-parallel-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_plate[2-2-sequential-parallel-sequential-parallel-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.25]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-()-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-()-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(2, 3)-2]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-()-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_prior-0.1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-True]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-()-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(7,)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(7,)-(5,)-3]
0.00s teardown tests/test_generic.py::test_rng_seed[minipyro-beta_binomial]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-()-()-()-()]
0.00s call     tests/distributions/test_pickle.py::test_pickle[TruncatedPolyaGamma]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-(5,)-()-()-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-(5,)-()-()-()-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-(5,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-(5,)-()]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-1.0-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-3]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-4000]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[jit-exponential-1]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[FoldedNormal]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-(5,)-()-(4,)]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-0.5]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[EnergyDistance_prior-scale2]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-False]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-()-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-()-3]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-2-(3, 2)-(4,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-reparam-False-unscaled]
0.00s teardown tests/distributions/test_one_hot_categorical.py::test_batch_log_dims[dim=3-probs0]
0.00s teardown tests/distributions/test_one_hot_categorical.py::test_batch_log_dims[dim=2-probs0]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(11, 7)-()-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(7,)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-18]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-tuple-sample_shape1]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[normal-1-100]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[1-10.0]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-(5,)-()-()-()-()-()]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-centered3-()]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-mask3]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-False-scaled]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-full-nonreparam-unscaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-parallel-sequential]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.7--0.5-1.5]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-1-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-()-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-()-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-14]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-3]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-1000]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-(5,)-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-(5,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-(5,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-()-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[EnergyDistance_prior-mask2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-()-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-()-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(7,)-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-18]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-3-1]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5),0]-(5, 2)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-5]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[2-None]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[3-None]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-(5,)-()-()-()-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-(5,)-()-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_normals_chain_iwae[diagonal-nonfactorized-True-0-1000-False-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-3-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-1-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-3-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(1,)-(1,)-(2, 3)-2]
0.00s call     tests/distributions/test_reshape.py::test_idempotent[0-0]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-(5,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-(5,)-()-()-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-None-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-sequential-sequential-sequential-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-1-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(7,)-()-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(7,)-(11, 7)-()-3]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,0,z(5)]-(5, 2)]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[2-0.1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-(5,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-()-()]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-centered3-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-6]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-2-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(11, 7)-(7,)-(5,)-3]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-0-1-1]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[laplace-2-200]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-(5,)-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-(5,)-()-()-()-()-()]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=2 --cuda]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate_iplate[3-2-parallel-sequential-sequential-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-3-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-2-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-3-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(11,)-(7,)-(11, 7)-()-2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-list-sample_shape1]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-4000]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Poisson_2D]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-(5,)-()-(3, 2)]
0.00s teardown tests/distributions/test_mvt.py::test_rsample[1-9.1]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-3-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-()-(1,)-()-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-1-1]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-(5,)-()-(3, 2)]
0.00s call     tests/distributions/test_reshape.py::test_idempotent[0-2]
0.00s call     tests/distributions/test_reshape.py::test_idempotent[0-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-3-7]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-2.0]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(7, 1)-2-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-2-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(7, 1)-1-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-1-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-()-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-(4, 1, 1)-(3, 1, 7)-(2, 7)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-17]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0-1.0--1.0]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional_whiten[2]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-7-7]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges732]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5),:]-(5, 4)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-(5,)-()-()-()-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-()-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Dirichlet]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-(5,)-()-(3, 2)]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9--0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(7, 1)-(1,)-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(7, 1)-(1,)-1-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-1-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-2-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(1,)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-()-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(1,)-()-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-1000]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-3-7]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[2-None]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-(5,)-()-()-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-(5,)-()-()-()-()]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-3-(2, 3)]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-1000]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0--1.0--1.0]
0.00s teardown tests/contrib/gp/test_models.py::test_hmc[VariationalSparseGP_y1D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-1-7]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4,1)]-(4, 2)]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[2-10.0]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[1-1.0]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-(5,)-()-()-()-(3, 2)]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0-0.9]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-2-3-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-2-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-()-3-3-()]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_MultiClass_2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges989]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-(5,)-()-()-()-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-(5,)-()-()-()-(3, 2)]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[1-0.1]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[1-None]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-(5,)-()-()-()-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0-1.0]
0.00s teardown tests/infer/test_util.py::test_psis_diagnostic[1-0.5-krange0-250000]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.25]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-2-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(7, 1)-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(7,)-(1,)-(1,)-1-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-1-3-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-3-(2, 3)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[python-normal-2]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-1-100]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-(5,)-(3, 2)]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-3-3-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-2-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-()-1-2-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[True-()-(11, 7)-(11, 7)-()-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-13]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-6-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1228]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-()-(3, 2)]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[(4, 1, 1)-(3, 1, 1)-(2, 1)-1-3-(2, 3)]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-Size-sample_shape0]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),0,z(5)]-(5,)]
0.00s call     tests/distributions/test_distributions.py::test_batch_entropy_shape[Poisson]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-()-(1,)-2-3-(2, 3)]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-4000]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[normal-1-200]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-(5,)-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-(5,)-()-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-(5,)-()-()-()-(4,)]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-1-2]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_logistic_regression[0.0855-None-4-False-False-False]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.5]
0.00s teardown tests/distributions/test_one_hot_categorical.py::test_batch_log_dims[dim=1-probs0]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_homogeneous_trick[()-(1,)-(7, 1)-2-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(2, 3)-2]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges828]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-(5,)-()-(4,)]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[1-10.0]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[3-()-(5,)-()-()-()-()]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-unscaled]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.75]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-17]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-False-initial_shape1-proposed_shape1]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Binary_2D]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-()-(5,)-()-()-()-()]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-True-unscaled]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-1-2]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Binary_2D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-7]
0.00s teardown tests/test_generic.py::test_model_sample[minipyro-eight_schools]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2),0]-(2,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-3-2]
0.00s teardown tests/infer/test_gradient.py::test_plate[TraceEnum_ELBO-nonreparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-None-parallel-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.5]
0.00s teardown tests/distributions/test_mvt.py::test_log_prob[1-()]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-13]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-False-initial_shape1-proposed_shape1]
0.00s teardown tests/test_generic.py::test_rng_seed[minipyro-eight_schools]
0.00s teardown tests/test_generic.py::test_model_sample[minipyro-beta_binomial]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,:,z(5)]-(5, 3)]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_remove
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-2-(3, 2)-()]
0.00s teardown tests/infer/test_discrete.py::test_prob[grad]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.5]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape0-df_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-10]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-tuple-sample_shape2]
0.00s teardown tests/contrib/gp/test_models.py::test_hmc[VariationalGP_y1D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-3-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-(5,)-()-()-()-()-(4,)]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-1.0]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-unscaled]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-nonreparam-unscaled]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.17-0.33]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.0]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.9-0.95-1.1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape0-df_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-5]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-1000]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-4000]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges67]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges604]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges539]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-(5,)-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-2-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_expand[1-(5,)-()-()-()-()-(4,)]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.6-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-5]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.39-0.33]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-tuple-sample_shape0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-0-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[laplace-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-3-7]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_mean_field_ok[minipyro]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-7]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-False-Size-sample_shape2]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[python-laplace-1]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges817]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-1.0]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-reparam-False-unscaled]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.75]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape1-df_shape0]
0.00s teardown tests/contrib/gp/test_models.py::test_model[VariationalSparseGP_y2D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges132]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-(5,)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[3-()-()-()-()-()-(4,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-nonreparam-scaled]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-3-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-(5,)-()]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.75]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-7]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-3]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-2-0-3-7]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_haar.py::test_normal[(2, 5)--1-True]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.25]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-10]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-False-initial_shape2-proposed_shape2]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional_whiten[6]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-(5,)-()-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-(5,)-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_distribution_validate_args[True-Exponential-args2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-full-reparam-False-unscaled]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-19]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel4-1-6-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges655]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-3-7]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 1, 1)-(7,)]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-17]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-Size-sample_shape1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-3-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-1-1]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[studentt-2-100]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges691]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-(5,)-()]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-(5,)-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-3]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.0]
0.00s teardown tests/distributions/test_mvt.py::test_rsample[1-3.9]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape2-df_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-16]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges131]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-3-7]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-False-False]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.25]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-15]
0.00s teardown tests/contrib/gp/test_models.py::test_inference[GPRegression_y2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Binary_1D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-1-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[studentt-2-200]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5,1),z(6)]-(5, 6)]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Cauchy]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.6-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-2]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.75]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-2]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Binary_1D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-3-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges751]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-False-subsample-nonreparam-unscaled]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.39-0.44]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[2-parallel-sequential-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-8]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0--1.0-0.0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-0-7]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges267]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-3-7]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-True-scaled]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.55-0.44]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-6]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape2-event_shape2]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges707]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceMeanField_ELBO-False-subsample-reparam-True-scaled]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.75]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-18]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-True-initial_shape2-proposed_shape2]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-0-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-0-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1220]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-()-()-()-()-(4,)]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-nonreparam-unscaled]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-1.0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-2-6-1-1]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-3-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges922]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges154]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges437]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges165]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges150]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->a-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-3]
0.00s teardown tests/infer/reparam/test_haar.py::test_normal[(4, 2)--2-False]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-parallel-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-1.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5-0.9]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-14]
0.00s teardown tests/contrib/gp/test_models.py::test_hmc[VariationalSparseGP_y2D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1108]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(6),0,z(5,1)]-(5, 6)]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5,1),0,z(6)]-(5, 6)]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(6),z(5,1)]-(5, 6)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-(5,)-()-()-()-(3, 2)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-(5,)-()-()-()-()-(3, 2)]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[BetaBinomial]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-10]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-3-2]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-None-(3, 2)]
0.00s teardown tests/infer/test_jit.py::test_infer_discrete[map-2]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.55-0.33]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-18]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-list-sample_shape2]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional_whiten[1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1334]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1316]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges224]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-300-parallel-300]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(4,)-()]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1085]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges744]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges203]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges43]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-3-7]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-None-batch_shape4]
0.00s teardown tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape0-True]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-None-None]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-19]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-1]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-4000]
0.00s teardown tests/contrib/gp/test_conditional.py::test_conditional_whiten[4]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1001]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges996]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges739]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges608]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges514]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges107]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-1-7]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-1-1-(3, 2)]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-False-False]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(4,)-(5,)]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0-1.0-1.0]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-0-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1027]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges875]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges734]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges262]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges75]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges0]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-7]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-()-()]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-3-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1038]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges388]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges225]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1180]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges199]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-(5,)-()-()]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-3-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-4]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-True-scaled]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-True-unscaled]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.0-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-10.0-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(4,)-(5,)]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges959]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges397]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges299]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges505]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges403]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Gamma]
0.00s teardown tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape1-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-parallel-None]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-()-(5,)]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1329]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1159]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1122]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1081]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1048]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges38]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-(5,)-()-()-()-(4,)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-3-7]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-3]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-False-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-sequential-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-0.25]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-10]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-()-(6,)-9]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1304]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1208]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges960]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges351]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges233]
0.00s teardown tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape2-False]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-scaled]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-nonreparam-scaled]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.55-0.33]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-()-(5,)]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1324]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1127]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges356]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1148]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-(5,)-()-()-()-()-(4,)]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.6-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-scaled]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.25]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-()-(5,)]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-False-list]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0--1.0-1.0]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_MultiClass_1D]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_subsample_smoke[SubsampleModel3-1-6-1-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[python-laplace-2]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1043]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges497]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges301]
0.00s call     tests/poutine/test_poutines.py::EscapeHandlerTests::test_all_escape
0.00s call     tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Dirichlet]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-1-7]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-reparam-False-unscaled]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-2-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape0-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1206]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges573]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges443]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges204]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges198]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges149]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges335]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_mean_field_ok[pyro]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.6-(3, 2)]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-7-5-1]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9-0.0]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-2.0]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-14]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-10]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-()-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 2)-()-9]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1164]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges487]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges271]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges252]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges892]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-(5,)-()-()-()-()]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-centered3-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-4]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-3-2-(4,)]
0.00s teardown tests/infer/test_gradient.py::test_plate[DiffTrace_ELBO-nonreparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-sequential-sequential-None]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-1000]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[normal-2-200]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges492]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges445]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1084]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-3-1]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 5)--1-0.0]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-parallel-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-parallel-None]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-()-(5,)]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[normal-2-100]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1201]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1018]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges571]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[:,z(4)]-(4, 2)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-1-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(4,)-()]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges68]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges796]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-3-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-2]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-scale3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-2-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(3, 2)-()]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1277]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1168]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges341]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges304]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges369]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-nonreparam-unscaled]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-False-sample_shapes1]
0.00s teardown tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalSparseGP_y2D]
0.00s teardown tests/contrib/gp/test_models.py::test_model[VariationalGP_y1D]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1282]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges764]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),z(5,1)]-(5, 4)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[2-()-(5,)-()-()-()-(4,)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-()-(4,)]
0.00s setup    tests/infer/mcmc/test_hmc.py::test_hmc_conjugate_gaussian[dim=10_chain-len=4_num_obs=1]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-None-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-6]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-0.0]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceEnum_ELBO-True-subsample-nonreparam-scaled]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.39-0.44]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-1-1-1-(4,)-()]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-1000]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0-1.0--1.0]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1346]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges702]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges660]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges435]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges346]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges112]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-3-7]
0.00s teardown tests/infer/reparam/test_studentt.py::test_moments[()]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-centered3-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-5]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-3-2-()]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.39-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-4-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-3-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-4-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape0-2]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1213]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1117]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges383]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-(5,)-()-()-()-()-()]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-1-7]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.0-(4,)]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-None-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-4]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-sequential-sequential]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-0.25]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-False-Size]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges645]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges294]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1052]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1020]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges956]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-1-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-2]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-0.5]
0.00s teardown tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape0-False]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-8]
0.00s teardown tests/contrib/gp/test_models.py::test_model[VariationalGP_y2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-3-2-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[studentt-1-100]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges650]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges393]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges11]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-3-7]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-1.0-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-4]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[nonreparam-RenyiELBO]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-False-scaled]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-1-2]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Binary_2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-2-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-7-3]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[python-studentt-1]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[python-exponential-2]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges440]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges6]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(5,1),z(4)]-(5, 4)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-3-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-True-scaled]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-nonreparam-unscaled]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[diag-1-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-()-(5,)]
0.00s teardown tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalGP_y1D]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[python-studentt-2]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges484]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-3-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-6]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-nonreparam-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-None-None]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-()-()]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0--1.0--1.0]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Poisson_2D]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[python-bernoulli-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges117]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges668]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-3-7]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges756]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-5]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-reparam-False-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-parallel-parallel]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[LowRankMultivariateNormal]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-3-3]
0.00s teardown tests/contrib/oed/test_finite_spaces_eig.py::test_vnmc_finite_space_model
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Binary_1D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Binary_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-1-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1196]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1053]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1000]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges749]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges425]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1116]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-3-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->c-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-10]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-centered3-(4,)]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.6-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[TraceGraph_ELBO-False-subsample-reparam-True-unscaled]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5--0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_distribution[full-3-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(2, 3)-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0--1.0--1.0]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Binary_2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Binary_2D]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[python-exponential-1]
0.00s setup    tests/test_generic.py::test_not_implemented[pyro]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1235]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges880]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges816]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5),z(5)]-(5,)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-1-7]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape3]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-0.0-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[2-None-parallel-None]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-3-2]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 1)-(4, 5)-9]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-1-1-1]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[exponential-2-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges991]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges257]
0.00s call     tests/distributions/test_unit.py::test_shapes[batch_shape0]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-1.0-(4,)]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-1.0-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-1]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(4, 2)--2-2.0]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-True]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[Trace_ELBO-False-subsample-reparam-False-scaled]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-2-2]
0.00s teardown tests/contrib/gp/test_models.py::test_inference[SparseGPRegression_y1D]
0.00s teardown tests/contrib/gp/test_models.py::test_forward[GPRegression_y1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-1-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1423]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1408]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges832]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges430]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges16]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(5,)-3]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-()-(4, 5)-9]
0.00s teardown tests/contrib/gp/test_models.py::test_gplvm[SparseGPRegression_y1D]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalGP_y1D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_MultiClass_2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-3-2-7]
0.00s setup    tests/test_examples.py::test_cpu[air/main.py --num-steps=1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1240]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges697]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges289]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1212]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-1-7]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-1.0-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-3]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-mask3]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient[DiffTrace_ELBO-False-subsample-nonreparam-scaled]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-sequential-sequential]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-()-3]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-True-sample_shapes1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 1)-()-9]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_Poisson_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-4]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-scale2]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-3-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0-1.0-1.0]
0.00s teardown tests/contrib/gp/test_models.py::test_gplvm[VariationalSparseGP_y2D]
0.00s teardown tests/contrib/gp/test_models.py::test_forward[SparseGPRegression_y1D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Poisson_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-2-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1419]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1376]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges159]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),z(5,1)]-(5, 4)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-2-1]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-3]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 2)-()-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0-1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0--1.0-1.0]
0.00s teardown tests/contrib/gp/test_models.py::test_gplvm[VariationalGP_y1D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Poisson_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-2-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1180]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges899]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges566]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges521]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges333]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges236]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-3-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-2]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_prior-0.1]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-3-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(5,)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(5,)-2]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-1]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalSparseGP_y2D]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges681]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(5,1),z(4)]-(5, 4)]
0.00s call     tests/ops/test_gamma_gaussian.py::test_expand[1-()-()-()-()-(5,)-(4,)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-3-1]
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_reparameterized_N_is_17
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01-0.5]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[StudentT-None-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-6]
0.00s teardown tests/infer/reparam/test_haar.py::test_normal[(4, 2)--2-True]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.5-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(5,)-2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-10]
0.00s teardown tests/contrib/gp/test_models.py::test_gplvm[SparseGPRegression_y2D]
0.00s teardown tests/contrib/gp/test_models.py::test_forward[VariationalGP_y2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Poisson_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-1-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges942]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges580]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges561]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges305]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-2-(3, 2)-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-6]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-mask2]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-False]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-1-3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-2]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-()-(4, 5)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0--1.0-1.0]
0.00s teardown tests/contrib/gp/test_models.py::test_hmc[SparseGPRegression_y2D]
0.00s teardown tests/contrib/gp/test_models.py::test_model[SparseGPRegression_y1D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Poisson_1D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_MultiClass_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-1-1-7]
0.00s teardown tests/test_examples.py::test_cuda[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --haar --cuda]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1383]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges861]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges394]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges700]
0.00s call     tests/contrib/forecast/test_evaluate.py::test_simple[warm-10-1-None-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-2]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist2]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.55-0.44]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-()-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-(2, 3)-3]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0--1.0-1.0]
0.00s teardown tests/contrib/gp/test_models.py::test_mean_function_GPR
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalGP_y1D]
0.00s teardown tests/contrib/gp/test_models.py::test_forward[VariationalSparseGP_y1D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_Poisson_2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-1-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1371]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges826]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges729]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-1-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-4]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[parallel-False-None]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-7-1-1]
0.00s teardown tests/infer/test_autoguide.py::test_predictive[AutoLaplaceApproximation]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[0.2-5.0]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[2-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-()-2]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-10]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0--1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0-1.0--1.0]
0.00s teardown tests/contrib/gp/test_models.py::test_gplvm[VariationalGP_y2D]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[VariationalSparseGP_y1D]
0.00s teardown tests/contrib/gp/test_models.py::test_forward[VariationalGP_y1D]
0.00s teardown tests/contrib/gp/test_models.py::test_forward[GPRegression_y2D]
0.00s teardown tests/contrib/gp/test_models.py::test_model[VariationalSparseGP_y1D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_MultiClass_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-2-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1414]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges927]
0.00s call     tests/distributions/test_util.py::test_sum_rightmost
0.00s call     tests/test_primitives.py::test_deterministic_ok
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-3-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-()-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-()-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(2, 3)-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-6]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-4-3-GaussianScaleMixture]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0-1.0--1.0]
0.00s teardown tests/contrib/gp/test_models.py::test_hmc[GPRegression_y2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-3-2-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1059]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges933]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges637]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges618]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges463]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges419]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges311]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges287]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges988]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-2-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(3, 7)-(4, 7)-()-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(5,)-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-18]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-2-3-GaussianScaleMixture]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(2,)-()-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0--1.0-1.0]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-2-lgssmgp]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Poisson_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-2-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1098]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges699]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges664]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges576]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges373]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-3-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-2]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[nonreparam-TraceEnum_ELBO]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.55-0.33]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.55-0.33]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(7,)-(4, 7)-(2, 3)-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-(3,)-(4, 7)-(7,)-()-3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-(5,)-3]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-False-True-4-3-GaussianScaleMixture]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mean_gradient[quadratic-True-True-2-3-GaussianScaleMixture]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0-1.0-1.0]
0.00s teardown tests/contrib/gp/test_models.py::test_gplvm[GPRegression_y1D]
0.00s teardown tests/contrib/gp/test_models.py::test_forward[VariationalSparseGP_y2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Binary_1D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_MultiClass_2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Binary_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-2-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1339]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges652]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges48]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Normal]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->a-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-3]
0.00s teardown tests/distributions/test_mvt.py::test_log_prob[1-(3, 2)]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[3-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[2-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-1-3]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-10]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0-1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0--1.0-0.0]
0.00s teardown tests/contrib/gp/test_models.py::test_inference[VariationalSparseGP_y2D]
0.00s teardown tests/contrib/gp/test_models.py::test_inference[GPRegression_y1D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_MultiClass_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-2-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-7-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1437]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1421]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1084]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges956]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges889]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges700]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges636]
0.00s call     tests/contrib/forecast/test_evaluate.py::test_simple[warm-7-1-7-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-3-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-6]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-None-parallel-None]
0.00s teardown tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[iplate-3-7000-0.0008]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-True-True]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-1-2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-1-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1344]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1093]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1064]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges496]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges328]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges324]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges160]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges924]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),z(4)]-(4,)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-3-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-3-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-2-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-2-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-2-(4,)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-2-3]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-10]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0--1.0--1.0]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_MultiClass_2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Poisson_1D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalGP_Poisson_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-1-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1335]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1252]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1171]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1146]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1058]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1049]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges932]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges831]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges269]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-1-7]
0.00s teardown tests/ops/test_ssm_gp.py::test_matern_kernel[1.5-2]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[2-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-()-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-3-2-1]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-3-2-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-1-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-3-1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-False-sample_shapes0]
0.00s teardown tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-2-NcvContinuous]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Binary_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[warm-7-1-7-1-7]
0.00s teardown tests/test_examples.py::test_cpu[baseball.py --num-samples=200 --warmup-steps=100 --num-chains=2]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges985]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges964]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges936]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges923]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges892]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges450]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges254]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges208]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-4]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.0-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceEnum_ELBO-full]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[3-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[2-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-2-1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 1)-(4, 5)-9]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-4000]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-3-2-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1432]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1291]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1191]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1113]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges670]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges12]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-1-1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-3-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-3-1-4]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.39-0.44]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.55-0.44]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-1-2-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[4-1-2-(4,)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob_block_diag[1-1-2-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_shape[False-()-(7,)-(6,)-()-2]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0--1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0-1.0-1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0--1.0-1.0-1.0]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalSparseGP_MultiClass_2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_MultiClass_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[IndependentHMM-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-3-2-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1290]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1288]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1182]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1172]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1169]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1054]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges904]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges725]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges550]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges327]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-1-7]
0.00s teardown tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape2-effective_sample_size]
0.00s teardown tests/infer/reparam/test_loc_scale.py::test_normal[Normal-0.0-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-4]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-7-5-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-0-None]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape1-2]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalSparseGP_y1D]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalGP_y2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_Binary_2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-3-2-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-7-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1350]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1348]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1311]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1234]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1185]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1170]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1128]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges715]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges581]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges354]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges326]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges267]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges133]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-()-()-()-(4,)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(3, 2)-3-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-6]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-3-2]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-3-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-2]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_gradient.py::test_plate[TraceGraph_ELBO-reparam]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.25]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[4-2-1-(3, 2)-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-3-3]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_Binary_2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_Poisson_2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1373]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1082]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges938]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges721]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges720]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges579]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges526]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges439]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges244]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges138]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges103]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges102]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges89]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-(5,)-()-()-(4,)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-1-1]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_generate_data[pyro]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-3-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-4]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[1-parallel-parallel-10000]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.1-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0-1.0--1.0]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalSparseGP_Poisson_2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference_with_empty_latent_shape[VariationalGP_MultiClass_2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_inference[VariationalSparseGP_MultiClass_2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1345]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1298]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1250]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1248]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1177]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1080]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1056]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges969]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges907]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges894]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges778]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges774]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges753]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges727]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges712]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges671]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges537]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges166]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges162]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges46]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges7]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5),0]-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-(5,)-()-()-(4,)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_logsumexp[2-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-3-1]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-3-2-()]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_restrictions[False-None-2-True]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-0-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-0.25]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-2-2]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-()-()-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0-1.0-1.0]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalGP_Poisson_2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GaussianHMM-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[warm-14-1-7-1-1]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_simple[warm-10-1-None-3-5]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoMultivariateNormal]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1077]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges833]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges773]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges606]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges552]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges238]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges87]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges74]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges72]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges30]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-3-1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-3-2]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.39-0.44]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-None-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-sequential-None-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.1-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(1,)-2-1]
0.00s teardown tests/contrib/gp/test_models.py::test_hmc[GPRegression_y1D]
0.00s teardown tests/contrib/gp/test_models.py::test_model[GPRegression_y2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_MultiClass_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoLaplaceApproximation]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1404]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1347]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1260]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1255]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1231]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1218]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1200]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1183]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1078]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1051]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges876]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges663]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges623]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges481]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges414]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges235]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges128]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges123]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges104]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges40]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges36]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),0,:]-(5, 4)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-3-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-5]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-1-1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.1-0.5]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[dense-batch_shape2-3]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape1-2-1-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[studentt-2-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1349]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1342]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1289]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1278]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1272]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1253]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1251]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1221]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1219]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1121]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1116]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1102]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1083]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1057]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1036]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1020]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1013]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges928]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges836]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges769]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges740]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges724]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges719]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges689]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges596]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges41]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges17]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-6]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_logsumexp[2-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-3-3-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-0-2000]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-True-False]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9-0.9]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-list-sample_shape0]
0.00s teardown tests/contrib/gp/test_models.py::test_hmc[VariationalGP_y2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape1-2-1-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[bernoulli-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1411]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1336]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1306]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1243]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1233]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1044]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1003]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges951]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges915]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges912]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges891]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges828]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges775]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges768]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges765]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges665]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges578]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges483]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges460]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges457]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges385]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges361]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges230]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges167]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges118]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges101]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges85]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges77]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges60]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges53]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges28]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges14]
0.00s call     tests/distributions/test_pickle.py::test_pickle[ZeroInflatedPoisson]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-3-1]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-1-1]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.39-0.44]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[1-None-parallel-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-0.9-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-17]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-tuple-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(2,)-(6,)-9]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-0-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[laplace-2-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1403]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1247]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1203]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1161]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1046]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges967]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges957]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges953]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges943]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges897]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges807]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges802]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges786]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges709]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges586]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges558]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges494]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges321]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges264]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges196]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges144]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges100]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-1-7]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-3-1]
0.00s teardown tests/infer/test_svgd.py::test_conjugate[IMQSteinKernel-multivariate]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-list-sample_shape0]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_whiten[VariationalSparseGP_y2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-2-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-0-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1398]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1355]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1309]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1300]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1232]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1175]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1154]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1123]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1088]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1041]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1039]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1030]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges995]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges987]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges976]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges972]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges961]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges949]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges948]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges879]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges864]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges855]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges797]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges776]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges763]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges748]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges684]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges647]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges643]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges632]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges622]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges429]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges367]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges364]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges322]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges316]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges265]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges213]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges201]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges172]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-4]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-3-1]
0.00s teardown tests/infer/reparam/test_stable.py::test_stable[LatentStableReparam-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-None-sequential-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.1-1.0]
0.00s teardown tests/distributions/test_mvt.py::test_log_prob[2-(3, 2)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-1-3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-list-sample_shape1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-3-2-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[normal-2-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1321]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1270]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1224]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1209]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1166]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1151]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1143]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1075]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1005]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges990]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges962]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges952]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges935]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges890]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges835]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges821]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges814]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges738]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges732]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges730]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges717]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges661]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges624]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges619]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges616]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges588]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges532]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges298]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges288]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges246]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges195]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges121]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges49]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges2]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-3-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-4]
0.00s teardown tests/ops/test_gaussian.py::test_logsumexp[2-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[marginal-a,abi,bcij->-ij]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.55-0.44]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.22-0.55-0.33]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-1.1-0.75]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9--0.9]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape2-1]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-1000]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape2-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-7-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1374]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1360]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1301]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1267]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1193]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1187]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1157]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1129]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1100]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1092]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1087]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1074]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges966]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges944]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges913]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges874]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges787]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges658]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges639]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges629]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges611]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges601]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges589]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges529]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges519]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges509]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges508]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges501]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges493]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges448]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges406]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges331]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges319]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges270]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges239]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges191]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges185]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges156]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges109]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges90]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges61]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges51]
0.00s call     tests/distributions/test_pickle.py::test_pickle[GammaGaussianHMM]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-()]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-2.0-0.0]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.03--1.5707963267948966]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0-0.5]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-True-Size]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(2,)-(6,)-9]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedNegativeBinomial-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[NegativeBinomial-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[MultivariateNormal-batch_shape0-3-2-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1434]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1406]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1275]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1261]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1229]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1214]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1210]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1205]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1198]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1188]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1156]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1141]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1133]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1115]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1114]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1103]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1069]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1025]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1015]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1010]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1008]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges998]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges858]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges853]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges851]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges846]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges817]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges804]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges733]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges673]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges555]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges527]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges499]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges426]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges412]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges380]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges365]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges303]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges272]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges220]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges218]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges215]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges210]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges115]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges108]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges94]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges5]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges4]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-()-(5,)-()-()]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-3-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(4,)-(3, 2)]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-0-2000]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.5]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.1-0.0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape1-sample_shape1]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward_with_empty_latent_shape[VariationalSparseGP_MultiClass_2D]
0.00s teardown tests/contrib/gp/test_likelihoods.py::test_forward[VariationalGP_MultiClass_1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[FoldedDistribution-batch_shape2-2-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-0-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1422]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1395]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1385]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1378]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1354]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1296]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1293]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1285]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1246]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1174]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1134]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1131]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges997]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges974]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges958]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges918]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges910]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges900]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges885]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges868]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges838]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges834]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges827]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges820]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges735]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges688]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges679]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges662]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges575]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges495]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges488]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges468]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges427]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges387]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges360]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges357]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges355]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges352]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges318]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges317]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges247]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges234]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges231]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges180]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges164]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges148]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges126]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges122]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges92]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges23]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-()-()-()-(3, 2)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-3-7]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_generate_data_plate[minipyro]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-0-3-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-1]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-1-2-1]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-False-True]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-1.0-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.75]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9--0.9]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-0-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1429]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1401]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1396]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1391]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1363]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1340]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1332]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1262]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1259]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1249]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1216]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1195]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1165]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1162]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1147]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1144]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1139]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1097]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1060]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1012]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1002]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges999]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges971]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges941]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges940]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges916]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges895]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges845]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges823]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges792]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges789]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges781]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges710]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges675]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges583]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges568]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges545]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges544]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges506]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges498]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges491]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges485]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges465]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges437]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges420]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges418]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges408]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges403]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges370]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges329]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges280]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges275]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges228]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges211]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges184]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges158]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges120]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-()-()-(5,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-()-(5,)-()-(3, 2)]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.55-0.44]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.39-0.44]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[None-None-0.29-0.39-0.33]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.55-0.33]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5--0.5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-17]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-False-tuple]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-0-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_serialization[python-normal-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1370]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1369]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1357]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1303]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1295]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1292]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1283]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1145]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1130]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1079]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1072]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1061]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges994]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges984]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges954]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges886]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges877]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges871]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges866]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges860]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges856]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges812]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges747]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges742]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges706]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges666]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges625]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges593]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges574]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges570]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges516]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges490]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges458]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges398]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges362]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges332]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges293]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges282]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges274]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges259]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges241]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges226]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges97]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges84]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges79]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges71]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges13]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-3-1]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-()-()-(5,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-3-1-2]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.55-0.44]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.29-0.39-0.33]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-2.0-0.75]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-tuple-sample_shape2]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-2-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1420]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1402]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1399]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1386]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1380]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1319]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1269]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1245]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1244]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1204]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1067]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1035]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1007]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges986]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges979]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges977]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges925]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges909]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges850]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges848]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges825]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges779]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges766]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges750]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges746]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges676]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges668]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges642]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges609]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges565]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges563]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges547]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges534]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges480]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges470]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges447]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges444]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges436]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges421]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges410]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges391]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges375]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges283]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges249]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges223]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges197]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges187]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges135]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges125]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges113]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges99]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges64]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges54]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,0,z(5)]-(5,)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-1-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_logsumexp[2-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceTMC_ELBO-full]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-10.0-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5-0.0]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1427]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1388]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1368]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1331]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1237]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1126]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1037]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1017]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1006]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges920]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges905]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges882]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges796]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges794]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges761]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges737]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges704]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges635]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges594]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges557]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges542]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges504]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges503]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges473]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges449]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges434]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges359]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges342]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges334]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges295]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges292]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges277]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges229]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges190]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges179]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges177]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges136]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges65]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges58]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges33]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges25]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges22]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges15]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges10]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_logsumexp[3-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-()-()-(5,)-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-3-1-1]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[None-True-True-True]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.22-0.39-0.33]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-sequential-sequential]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.9-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.9-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-1.0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-False-sample_shapes0]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[normal-1-1]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoLaplaceApproximation]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1424]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1413]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1375]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1365]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1327]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1313]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1308]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1242]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1226]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1152]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1118]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges992]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges982]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges931]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges887]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges865]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges840]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges822]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges809]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges805]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges784]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges726]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges701]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges686]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges648]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges640]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges614]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges612]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges599]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges584]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges472]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges442]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges438]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges416]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges413]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges401]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges310]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges306]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges285]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges261]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges256]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges251]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges243]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges233]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges182]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges169]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges141]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges95]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges59]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges35]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-3-1]
0.00s teardown tests/infer/reparam/test_stable.py::test_stable[StableReparam-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-parallel-None-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.5-0.75]
0.00s teardown tests/distributions/test_mvt.py::test_log_prob[2-(4,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-(11, 7)-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1435]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1416]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1409]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1362]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1314]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1280]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1254]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1192]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1190]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1138]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1136]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1105]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1026]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1023]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1021]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges914]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges902]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges881]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges869]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges863]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges810]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges799]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges791]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges771]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges722]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges705]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges630]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges591]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges517]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges452]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges390]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges339]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges323]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges242]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges214]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges151]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges131]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges110]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges82]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges66]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges20]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-3-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-10]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-()-()-(5,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-(3, 1, 1)-(1, 4, 1)-(1, 1, 5)-()]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-1-1-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-1-2000]
0.00s teardown tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[plate-None-7000-0.0008]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-10.0-1.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5-0.9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0--1.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-1-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1337]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1264]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1257]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1239]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges945]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges898]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges760]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges683]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges536]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges507]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges467]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges396]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges382]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges309]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges263]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges206]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges202]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges200]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges146]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges21]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-6-()-1-7]
0.00s call     tests/pyroapi/test_pyroapi.py::test_local_param_ok[minipyro-py]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-4]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-()-()-(5,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-()-()-(5,)-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[2-parallel-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-0.1-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-10.0-0.75]
0.00s teardown tests/distributions/test_mvn.py::test_shape[5-cov_shape2-loc_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-True-sample_shapes0]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1438]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1426]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1418]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1393]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1377]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1211]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1160]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1158]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1111]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1110]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1066]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges930]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges884]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges843]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges653]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges607]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges585]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges560]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges535]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges478]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges411]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges377]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges369]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges336]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges313]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges232]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges175]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges130]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges69]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-4]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-3-(3, 2)-(3, 2)]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[None-True-True-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_8[iplate-plate-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-parallel-sequential]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.1-0.25]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0--0.9]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-False-sample_shapes0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-7-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1384]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1227]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1199]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1179]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1089]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1031]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges896]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges893]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges878]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges678]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges604]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges539]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges486]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges431]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges409]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges400]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges205]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0,z(4)]-(4,)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-1-7]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape4]
0.00s teardown tests/ops/test_gaussian.py::test_logsumexp[3-(4,)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[sample-ab,bc->-]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-3]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-False-False]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-0.5-1.0]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.1--1.5707963267948966]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5-0.9]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1405]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1382]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1372]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1367]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1302]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1202]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1109]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1028]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges934]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges911]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges815]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges756]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges698]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges638]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges562]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges551]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges524]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges462]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges378]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges349]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges344]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges174]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges139]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges119]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges56]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges42]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges26]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges9]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-()-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-()-()-(5,)-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-1.1-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.9-0.25]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.5-0.5-0.0]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[3.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-13]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1417]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1390]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1379]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1353]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1351]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1320]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1318]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1310]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1163]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1142]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1112]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1011]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges950]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges785]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges758]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges755]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges745]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges741]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges694]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges617]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges590]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges587]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges553]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges510]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges502]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges455]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges454]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges372]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges301]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges290]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges245]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges161]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges155]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges153]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges143]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-1-7]
0.00s call     tests/pyroapi/test_pyroapi.py::test_local_param_ok[minipyro-jit]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-5]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-()-()-(5,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-0-1-(3, 2)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-1-1-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_nonnested_plate_plate_ok[TraceEnum_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-1-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-sequential-parallel-None]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[4-2-2-(3, 2)-(5,)]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-7-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[studentt-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1397]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1189]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1176]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1173]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1104]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1009]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges955]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges819]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges783]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges723]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges634]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges600]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges572]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges548]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges461]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges62]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-(5,)-()-()-(3, 2)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-3-7]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_logsumexp[3-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-3]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[0.1-False-False-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-sequential-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-None-sequential-None]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[-0.9-2.0-1.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9-0.9]
0.00s teardown tests/distributions/test_spanning_tree.py::test_make_complete_graph[cpp-4-expected_grid2]
0.00s teardown tests/distributions/test_conjugate.py::test_dirichlet_multinomial_log_prob[sparse-batch_shape1-3]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1241]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1094]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1073]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges988]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges968]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges713]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges696]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges644]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges621]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges522]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges347]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges320]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges253]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges222]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges171]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges31]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges27]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-()-()-(5,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-1-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-3]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-0.0]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_8[plate-plate-1-1]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.39-0.44]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[0.5-1.0-0.75]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5-0.5]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-1-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-0-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1425]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1389]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1341]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1326]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1153]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1119]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1040]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1019]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges901]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges854]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges829]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges736]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges711]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges659]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges628]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges475]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges441]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges392]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges314]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges173]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges73]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-6-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-3-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_logsumexp[3-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-3]
0.00s teardown tests/infer/reparam/test_haar.py::test_uniform[(2, 5)--1-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-parallel-sequential]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-0-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-sequential-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-sequential-sequential-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-False-True]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_mean_variance[1.9-2.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5--0.9]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-1-1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 1)-(4, 5)-9]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1307]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1299]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1258]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1148]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1024]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges993]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges903]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges752]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges703]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges667]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges582]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges505]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges432]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges399]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges363]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges284]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges268]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges240]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges186]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges140]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges129]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges127]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges83]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges34]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges32]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),z(4)]-(4,)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-5]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-3-5-1]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.55-0.33]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-None-0.29-0.39-0.33]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-parallel-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-sequential-sequential-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-parallel-sequential-None]
0.00s teardown tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[plate-3-7000-0.0008]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-True-True]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-True-True]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5--0.9]
0.00s teardown tests/distributions/test_spanning_tree.py::test_make_complete_graph[cpp-3-expected_grid1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1436]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1392]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1358]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1322]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1317]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1312]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1256]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1212]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1197]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1107]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1099]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1076]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1016]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges965]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges947]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges849]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges844]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges818]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges788]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges767]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges718]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges682]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges680]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges657]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges636]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges620]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges595]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges531]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges500]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges384]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges376]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges340]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges209]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges207]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges189]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges181]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges111]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges80]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-3-1]
0.00s call     tests/infer/test_inference.py::test_exponential_gamma[rsvi-ReweightedWakeSleep]
0.00s teardown tests/infer/reparam/test_stable.py::test_stable[StableReparam-(2, 3)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-3]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-sequential-None-None]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0--0.5]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1407]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1400]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1364]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1361]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1263]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1217]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1096]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1063]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges937]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges808]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges762]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges754]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges728]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges677]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges633]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges597]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges592]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges577]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges549]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges512]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges286]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges281]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges278]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges273]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges152]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges88]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges86]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges45]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges37]
0.00s call     tests/contrib/forecast/test_evaluate.py::test_simple[warm-14-1-7-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-1-7]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_generate_data_plate[pyro]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-6]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.39-0.44]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-parallel-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-None-sequential-None]
0.00s teardown tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[range-None-100-0.011]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-True-True]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[1.0--1.5707963267948966]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-3]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-3-2-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1431]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1356]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1343]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1266]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1184]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1071]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1070]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1047]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1045]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1042]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges983]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges939]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges862]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges777]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges672]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges626]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges530]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges489]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges423]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges422]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges417]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges374]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges358]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges335]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges296]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges188]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges137]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges124]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges91]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges63]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges50]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges8]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges3]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-6-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(4,)-3-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-6]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[2-parallel-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-None-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-sequential-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-None-None-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-False-True]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-1-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-0-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1366]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1338]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1330]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1284]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1274]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1225]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1194]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1150]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1135]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1125]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1124]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1106]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1068]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1055]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1052]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1050]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1022]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges919]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges873]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges872]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges824]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges800]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges790]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges782]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges764]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges731]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges708]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges669]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges654]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges641]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges631]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges567]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges515]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges405]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges350]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges291]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges227]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges219]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges194]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges96]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges81]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges52]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges24]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-6]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-6]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-1-5-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-None-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-parallel-parallel-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-True-True]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[3.0-1.5707963267948966]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9-0.5]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Stable-False-sample_shapes1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 2)-(4, 5)-9]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-7-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1333]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1297]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1279]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1276]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1101]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1086]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1065]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges970]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges963]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges852]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges847]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges674]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges651]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges610]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges554]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges543]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges541]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges528]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges525]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges446]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges428]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges325]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges315]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges300]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges276]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges255]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges248]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges178]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges147]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges142]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges114]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges98]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges57]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges44]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges29]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[0,z(4)]-(4,)]
0.00s call     tests/distributions/test_delta.py::test_expand[batch_shape0]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-3-7]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-5]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-7-5-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[1-None-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-None-parallel-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-True-False]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-True-True]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-False-False]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-True-False]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0--0.9]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-1-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1430]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1415]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1305]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1286]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1271]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1186]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges978]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges973]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges921]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges908]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges906]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges857]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges839]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges795]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges770]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges759]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges716]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges690]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges538]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges415]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges368]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges353]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges348]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges330]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges297]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges266]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges176]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges168]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges78]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Uniform]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-3-7]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-()-(6, 5)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-4]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[5-1-1-()]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-7-2-2]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.39-0.33]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-parallel-None-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-True-True]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9--0.5]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-3-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-7-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1359]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1315]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges980]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges929]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges924]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges867]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges798]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges793]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges646]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges533]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges520]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges479]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges476]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges474]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges464]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges456]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges433]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges407]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges371]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges312]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges212]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges183]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges145]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges116]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges106]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges93]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges55]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-(3, 2)-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-2]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-1-2-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_iplate[2-2-parallel-sequential-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-sequential-parallel]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-True-False]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-True-False-False]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9-0.5]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(11, 7)-1-3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-True-initial_shape0-proposed_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0--1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0-1.0-1.0]
0.00s teardown tests/contrib/oed/test_linear_models_eig.py::test_vnmc_linear_model
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s teardown tests/contrib/cevae/test_cevae.py::test_smoke[exponential-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1238]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1155]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1014]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges837]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges687]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges605]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges564]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges559]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges466]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges459]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges237]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges170]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges39]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),0,0]-(5,)]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-()-(5,)-()-(4,)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-6-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-0-()-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-5]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-6]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-sequential-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-None-parallel-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-True-False-True]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-True-False]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[3.0--1.5707963267948966]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 1)-(6,)-9]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-7-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1394]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1328]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1268]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1034]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1029]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges926]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges842]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges813]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges780]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges695]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges692]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges546]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges477]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges471]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges402]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges389]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges338]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges150]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges19]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-6-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-5]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist1]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.29-0.55-0.33]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[parallel-2-0.22-0.55-0.44]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-0-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-sequential-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-parallel-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-parallel-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-True-False]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-False-False-True]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-1-1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 2)-(4, 5)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-1.0-1.0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-0-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1412]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1325]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1281]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1222]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges757]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges602]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges523]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges469]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges386]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges381]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges165]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-6-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-6-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-()-(6, 5)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-2-6]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-3-2-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-sequential-parallel-sequential]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-parallel-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-True-False-True-False]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[1-2-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-()-()]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(11, 7)-1-1]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1230]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1181]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges615]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges404]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges379]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges345]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges302]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges258]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges199]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-(5,)-()-()-(3, 2)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-10]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-()-(6, 5)]
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_reparameterized_four_layers
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-6]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-3-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[2-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[1-None-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-1-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-0-2000]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-sequential-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-parallel-None-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[loss_and_grads-sequential-None-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-parallel-None-sequential-None]
0.00s teardown tests/infer/test_discrete.py::test_prob[value]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-False-True]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9-0.0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Stable-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(2,)-(4, 5)-9]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-4000]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0-1.0--1.0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-0-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges556]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-6-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-3]
0.00s teardown tests/ops/test_indexing.py::test_value[()-()-(2,)-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-4]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[None-True-False-False]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_ok[TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-0-2000]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-1-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-sequential-parallel-None]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-list-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0--1.0-1.0--1.0]
0.00s teardown tests/contrib/gp/test_models.py::test_model[SparseGPRegression_y2D]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize2]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-2-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-7-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1140]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges883]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges803]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges693]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges518]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges451]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(3, 2)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-()-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-()-(7,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-4]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist1]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-sequential-parallel-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-False-False]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-0.5]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[0.2-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(7,)-1-1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(2,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(2,)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 1)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 2)-(4, 5)-9]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-4000]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0--1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1215]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges649]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges260]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-()-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-0-()-1-7]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(3, 2)-()]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01-0.5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-6]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_8[plate-iplate-1-1]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_normal_elbo_gradient[sequential-None-0.22-0.39-0.33]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-sequential-sequential-None]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5--0.9]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(11, 7)-(7,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(11,)-(7,)-(7,)-1-3]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 2)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 1)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 2)-()-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0--1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-2-1-7]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1387]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1032]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges981]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges343]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-()-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-10]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_logsumexp[1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[1-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(4,)-(7,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(2, 3)-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-2]
0.00s teardown tests/infer/test_valid_models.py::test_plate_iplate_ok[TraceTMC_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceTMC_ELBO-subsample]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-sequential-sequential]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-1-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-sequential-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-sequential-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-parallel-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-None-sequential-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-False-False]
0.00s teardown tests/distributions/test_rejector.py::test_rejector[0.25-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(11, 7)-(11, 7)-1-1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 2)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(2,)-(6,)-9]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-1-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-0-3]
0.00s call     tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges307]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-()-(5,)-()-(3, 2)]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_add[0.1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(3, 2)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-6-(4,)-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-()-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(4,)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-(4,)-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_prior-mask4]
0.00s teardown tests/infer/test_valid_models.py::test_plate_reuse_ok[TraceEnum_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-1-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-1-2000]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-None-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-None-None-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[True-False-True-False-True]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 2)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(2,)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(2,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-()-(4, 5)-9]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-3-2-7]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs4-dims4-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(3, 2)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-2-1-(4,)-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-10]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-1-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(4,)-1-1-6]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-1-1-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceTMC_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-None-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-sequential-None]
0.00s teardown tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[plate-8-7000-0.0008]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape2-event_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_gamma_gaussian_hmm_log_prob[3-1-1-(3, 2)-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_log_prob[1-1-1-()-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-1-1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-True-Size]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-()-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 1)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(2,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-()-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-()-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 2)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 2)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(2,)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(2,)-()-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0-1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0--1.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-3-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-7-3]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-1-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(4,)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-10]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape5]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-1-(4,)-(3, 2)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0--1.0]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[skewed-()-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-1]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_beta_bernoulli[0.5-False-True-False]
0.00s teardown tests/infer/test_valid_models.py::test_no_log_prob_ok[EnergyDistance_prior]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-7-2-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-None-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-False-False]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[2.0-5.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[(4, 1, 1)-(3, 1, 7)-(2, 7)-1-1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 1)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(2,)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(2,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 2)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 2)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 2)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 1)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(2,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 2)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(2,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 1)-(6,)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0-1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0-1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-1-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-0-3]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-1-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-1-(3, 2)-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-5]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-0-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(4,)-(7,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(2, 3)-1-1-4]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_plate_ok[TraceTMC_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[1-parallel-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-parallel-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-None-sequential-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-False-False-True-False]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape0-event_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 1)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 2)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 2)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(2,)-(4, 5)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0--1.0--1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-3-2-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-0-3]
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-,bi,abi->b-i]
0.00s call     tests/distributions/test_util.py::test_sum_leftmost
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_add[1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-1-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-1-0-(3, 2)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-10]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-1-1-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-1-3]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceTMC_ELBO-subsample]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-3-1-2]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-3-1-2]
0.00s teardown tests/infer/test_predictive.py::test_posterior_predictive_svi_manual_guide[True]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.44-0.41]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-1-2000]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-0-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-parallel-sequential-None]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-True-True-True]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(1,)-(7,)-1-1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[RejectionExponential]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 2)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 2)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 2)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(2,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-()-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(2,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-()-(4, 5)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0--1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0-1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none--1.0--1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0-1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-2]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-1-1-7]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),0,:]-(5, 4)]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_add[100]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-2-0-(4,)-3-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-0-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-()-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-1-(4,)-3-1]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-2-(3, 2)-(4,)]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-3-2-2]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-3-5-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-None-None-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-parallel-parallel-10000]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 2)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 2)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(2,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 2)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 1)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 1)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 1)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 2)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 2)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 1)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(2,)-(4, 5)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0-1.0-1.0--1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-1-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-0-7]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),:,0]-(5, 3)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-6-()-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-2-3]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-()-(6, 5)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(2, 3)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-()-1-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-5]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-7-1-2]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-3-2-1]
0.00s teardown tests/infer/test_jit.py::test_dirichlet_bernoulli[JitTraceEnum_ELBO-False]
0.00s teardown tests/infer/test_jit.py::test_beta_bernoulli[JitTraceEnum_ELBO-False]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_restrictions[True-None-1-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-1-2000]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-1-2000]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-0-2000]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-0-2000]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-parallel-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-None-parallel-None-None]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Binomial-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 1)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(2,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(3, 1)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 1)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 2)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(2,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-()-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-()-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 1)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(2,)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 1)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-()-(6,)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0--1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0-1.0--1.0--1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0--1.0-0.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-3-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-0-7]
0.00s call     tests/distributions/test_unit.py::test_shapes[batch_shape2]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-1-0-(3, 2)-1-1]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(4,)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(4,)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-()-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-1-1-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-3]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-7-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-0-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-parallel-1-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-parallel-10000]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern_bern[differentiable_loss-sequential-None-parallel-None]
0.00s teardown tests/distributions/test_stable.py::test_variance[2.0-1.5]
0.00s teardown tests/distributions/test_rejector.py::test_exponential_elbo[0.25-2.0]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-()-(7,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_mrf_shape[()-(7,)-()-1-2]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-()-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 2)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(2,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(2,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-()-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 1)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-()-(4, 5)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0-1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum-1.0--1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[1.0--1.0-1.0--1.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape0-3-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-0-3]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-1-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-2-0-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-1-(4,)-3-7]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-1-(4,)-1-1]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model0-True-1-0-(3, 2)-3-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-8]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[3-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(4,)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_logsumexp[1-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-(4,)-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[skewed-(2, 3)-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-(4,)-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-2-5]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-3-2-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-sequential-1-None]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Stable-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 1)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(2,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 2)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-()-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 1)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(3, 1)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 1)-(3, 2)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(2,)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 1)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-()-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-()-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-()-()-9]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-3-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 8}-7-3]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-()-()-()-(4,)]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-5-6]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(4,)-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-1-2]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-2-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-parallel-parallel-0-2000]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-sequential-parallel-parallel-None]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 2)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-()-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 1)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-()-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 1)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(2,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 2)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 1)-(6,)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[none-1.0-1.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-(5,)-()-()-()]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-(4,)-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-6]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-(5,)-()-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-1-1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-3-1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_independent_hmm_shape[symmetric-()-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape[symmetric-()-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-4]
0.00s teardown tests/infer/reparam/test_haar.py::test_uniform[(6,)--1-False]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-1-1-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-None-False-None]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-3-1-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate_plate[2-2-parallel-sequential-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[1-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[1-parallel-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-0-2000]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[sequential-sequential-sequential-1-2000]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.03-0.0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-list-sample_shape2]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 1)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-(3, 1)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(2,)-()-9]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-3-2-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-0-3]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_add[10]
0.00s teardown tests/pyroapi/test_pyroapi.py::test_generate_data[minipyro]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-6]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(4,)]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-2.0]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-False-HMC-_empty_model]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-parallel-False-None]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-sequential-True-None]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-7-1-2]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-7-2-2]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-1-2000]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(3, 2)-(5,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 1)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(2,)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-(3, 1)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-(3, 1)-()-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_exact_5_4_3[momentum--1.0-1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-0-3]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model4-True-1-1-()-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-2]
0.00s teardown tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape2-split_gelman_rubin]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-()-(2,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-2-(3, 2)-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-4-1-1-1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-1-2]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-parallel-False-None]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist0]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-7-1-1]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-7-2-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-parallel-parallel-None-None]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(2,)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(3, 2)-(2,)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-()-(3, 2)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(2,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 2)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-()-(3, 2)-(6,)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_2[-1.0--1.0-1.0--1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[DirichletMultinomial-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-2-1-7]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-()-()-()-(3, 2)]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-5-3]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-0-1-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-4]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-3-5-2]
0.00s teardown tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceEnum_ELBO-reparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[2-sequential-parallel-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-parallel-0-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-sequential-sequential-1-2000]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[100.0--1.5707963267948966]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0-0.9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-()-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-()-()-9]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-4000]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-2-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-7-7]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1027]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-()-()-()-()]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model2-True-2-0-(4,)-3-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-4]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-()-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(4,)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-()-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(4,)-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[ubersum-1-1-1-1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-1-2]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-1-1-(3, 2)]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist2]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-7-5-2]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-3-5-1]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-3-2-2]
0.00s teardown tests/infer/test_predictive.py::test_posterior_predictive_svi_manual_guide[False]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-list-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(2,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-()-()-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[7-(2,)-()-()-9]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape0-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-7-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-3]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-2-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(4,)-(4,)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8--0.5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-5]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist3]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-3-5-2]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-7-1-2]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-7-2-2]
0.00s teardown tests/infer/test_gradient.py::test_plate[TraceEnum_ELBO-reparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-sequential-0-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[None-loss_and_grads-10]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_big_model_guide_pair[False-True-False-False-True]
0.00s teardown tests/distributions/test_von_mises.py::test_log_prob_normalized[0.03]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 2)-(3, 1)-(6,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(3, 1)-(3, 1)-()-9]
0.00s teardown tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalGP_y2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-2-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{'haar': False}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-7-3]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),:,:]-(5, 3, 4)]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-(5,)-()-()-(3, 2)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model1-True-2-1-()-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-6]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-2-(4,)-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-6]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-parallel-sequential-None]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-5-uniform]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-2]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-10]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 1)-(2,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(2,)-()-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[2-(2,)-(3, 2)-()-9]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-2-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-0-3]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),:,:]-(5, 3, 4)]
0.00s call     tests/contrib/forecast/test_forecaster.py::test_smoke[hmc-Model3-True-1-0-(4,)-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-10]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape5]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-2-(4,)-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-2]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-sequential-False-None]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-3-2-1]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-7-5-1]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5--0.9]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-2-sparse]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-list-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_with_rate_smoke[11-(3, 2)-(3, 2)-()-9]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-4000]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape2-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-3-2-1]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),:,0]-(5, 3)]
0.00s call     tests/ops/test_contract.py::test_ubersum_size_error[ubersum-ab,bc->-shapes0-]
0.00s call     tests/distributions/test_delta.py::test_expand[batch_shape3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-5]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-2-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(4,)-(3, 2)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01--0.5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-4]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-1-1-()]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-sequential-False-None]
0.00s teardown tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-None-2-False]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[1.0-1.5707963267948966]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9--0.5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-18]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-list-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-2-unclamped]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'jit_compile': True}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-7-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-10-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-7]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(2,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-1-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-0-3-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-5]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-parallel-True-None]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-3-5-1]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-3-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[loss_and_grads-sequential-parallel-parallel-None]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9-0.0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-list-sample_shape1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[InverseGamma-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[GammaPoisson-batch_shape1-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[BetaBinomial-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape1-3-2-7]
0.00s call     tests/distributions/test_unit.py::test_shapes[batch_shape1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-3-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-6-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-6-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-4]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-2-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-2-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-1-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-1-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-1]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-parallel-True-3]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[1-sequential-True-None]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-1-1-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-parallel-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_berns[differentiable_loss-parallel-None-None-None]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5-0.9]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape1-df_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-1000]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape1-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Exponential-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape0-2-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-0-3]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoLowRankMultivariateNormal]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-()-()-()-()]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-()-(5,)-()-(4,)]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-4]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-1-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-0-2-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-()-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-1-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-2]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-0.0]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-7-5-1]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0--0.5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-13]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-list-sample_shape0]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-1000]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-1000]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-3-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Gamma-batch_shape0-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape2-3-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-3-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-7-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-4-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-5-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-2-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-4-5]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-2-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_cat[3-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-4-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-5]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-1-2-1]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-16]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-4000]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[ZeroInflatedPoisson-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[StudentT-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape2-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Geometric-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Dirichlet-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Beta-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Bernoulli-batch_shape1-3-2-7]
0.00s call     tests/distributions/test_delta.py::test_expand[batch_shape5]
0.00s call     tests/distributions/test_delta.py::test_expand[batch_shape2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-6-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-3-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-5-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-6]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-2-1-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-2-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(4,)-(4,)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_stable[LatentStableReparam-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-6]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-parallel-True-3]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-None-False-None]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist3]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-1-1-1]
0.00s teardown tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-None-2-True]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9--0.5]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-Size-sample_shape0]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-1-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-0-3]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-(5,)-()-()-()]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-20-plate]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-7]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-0-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-2-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-()-(5,)-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-()-(5,)-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-()-()-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_sizes[naive_ubersum-1-1-3-2]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3-0.5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-1]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-1-1-()]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9--0.9]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-11]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(3, 2)-()-9]
0.00s teardown tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[VariationalSparseGP_y1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape2-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Laplace-batch_shape0-2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-3-2-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-0-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-2-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-3-3]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-()-()-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-()-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-1-(3, 2)-(4,)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-1-1-()]
0.00s teardown tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceGraph_ELBO-nonreparam]
0.00s teardown tests/distributions/test_stable.py::test_variance[2.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0--0.5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-list-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-20-unclamped]
0.00s teardown tests/contrib/gp/test_models.py::test_forward[SparseGPRegression_y2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Normal-batch_shape1-1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape1-2-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'num_quant_bins': 2}-0-8-6]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges125]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-()-(5,)-()-(4,)]
0.00s call     tests/ops/test_contract.py::test_ubersum_size_error[ubersum-ab,bc->-shapes1-b]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Poisson]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-6]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-1-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-1-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-()-()-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(4,)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-6]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_vectorized_num_particles[50-parallel-True-None]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-1-2-2]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-1-1-2]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9--0.9]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-12]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape1-10]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-20-unclamped]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-0.01]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[LogNormal-batch_shape0-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape1-1-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-7-3]
0.00s call     tests/ops/test_gaussian.py::test_expand[2-(5,)-()-()-()]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-3-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(4,)-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-10-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-5-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-4-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-4-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(4,)-2-3]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape3]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-()-(5,)-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-4]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-3-2-()]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist1]
0.00s teardown tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_delta_guide[True]
0.00s teardown tests/infer/test_enum.py::test_elbo_categoricals[parallel-parallel-parallel-1-None]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5-0.5]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape2-df_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-17]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-tuple-sample_shape2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-2-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),0]-(4,)]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-5-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-3]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape4]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(2,)-(2,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-1-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-1-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-(5,)-()-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-()-(5,)-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-1-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(4,)-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-3]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_logistic_regression[False-JIT=True]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-7-2-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[1-None-parallel]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-16]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-list-sample_shape2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True, 'num_quant_bins': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-10-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(4,)-3-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-2-6]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-()-()-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_stable.py::test_stable[LatentStableReparam-(2, 3)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-2]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-1-1-2]
0.00s teardown tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceEnum_ELBO-nonreparam]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9-0.0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-20-unclamped]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape0-3-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar': True}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar_full_mass': 4}-7-8-6]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-0-7]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs6-dims6-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-3]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-()-(5,)-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-()-(5,)-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-2-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-1-2-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_ok[EnergyDistance_prior]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-1-5-2]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[1.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9--0.9]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-11]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(4,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[2-(2, 3)-(4, 5)-9]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-1-2.5]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-3-7]
0.00s call     tests/distributions/test_delta.py::test_expand[batch_shape1]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(2,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-2-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-(5,)-()-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-()-()-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-(5,)-()-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-1-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_cat[3-(4, 7, 6)--1-(2, 1, 3)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-3]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[iplate-None-100-0.013]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9-0.5]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-13]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-list-sample_shape2]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(4,)-(5,)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-0.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[svi-{'haar': False}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-0-3]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoMultivariateNormal]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-()-()-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-3-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(4,)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-2-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(4,)-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[JitTrace_ELBO-subsample-nonreparam]
0.00s teardown tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[DiffTrace_ELBO-reparam]
0.00s teardown tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[True-None-1-False]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5--0.9]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-5.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-15]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Binomial-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-list-sample_shape2]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-1000]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-1.0-0.0]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-1-1.5]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_prefix_condition[Cauchy-batch_shape2-2-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-0-7]
0.00s call     tests/distributions/test_delta.py::test_expand[batch_shape4]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-2-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-1-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-1-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-(5,)-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(3, 2)-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-1]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist0]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-1-2-2]
0.00s teardown tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_diag_normal_guide[False]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_restrictions[False-None-2-False]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.01-0.0]
0.00s teardown tests/distributions/test_tensor_type.py::test_conflicting_types
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9--0.5]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-0.5]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape0-df_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-8]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-20-unclamped]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-1-0.5]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[SparseGPRegression_y2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{}-7-8-6]
0.00s call     tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Exponential]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,wx,x->]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-3]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(3, 1)-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-(5,)-()-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-()-()-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[1-(5,)-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-2-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(4,)-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-6]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5--0.9]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[5.0-1.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape2-df_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-9]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-tuple-sample_shape1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{}-0-8-6]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True, 'num_quant_bins': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-0-7]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),:]-(4, 3)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-3-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-1-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-()-(5,)-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-3-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-2]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enumeration_in_dag[TraceTreeEvaluator]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_num_chains[True-2-2]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[1-1-5-1]
0.00s teardown tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[range-3-100-0.018]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5--0.5]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-0.2]
0.00s teardown tests/distributions/test_rejector.py::test_exponential_elbo[0.25-1.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape0-df_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-19]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-tuple-sample_shape2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'num_quant_bins': 2}-7-8-6]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-0-3]
0.00s call     tests/ops/test_gaussian.py::test_expand[3-()-(5,)-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-1-1-()]
0.00s teardown tests/ops/test_gaussian.py::test_expand[3-()-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-()-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-1-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-1-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-()-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-2]
0.00s teardown tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_diag_normal_guide[True]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[30.0--1.5707963267948966]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0-0.9]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape2-value_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-17]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Cauchy-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-5-clamped]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-1-1]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-(5,)-()-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-1-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-1-1-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-2]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[4-1-1-()]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-1-1-(4,)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[3-1-1-()]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-0.1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_ok[EnergyDistance_noprior]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-1-5-1]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.44-0.33]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5-0.9]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-2.0]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-1.0]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-0.5]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape0-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape1-value_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-16]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-tuple-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-0.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-1.0-0.0]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Cosine]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-3-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar': True}-7-8-6]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-7-3]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs13-dims13-4]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-3-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-1-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-2-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-1-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-()-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-()-(4,)]
0.00s teardown tests/infer/reparam/test_transform.py::test_log_normal[(4,)]
0.00s teardown tests/infer/reparam/test_stable.py::test_stable[StableReparam-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-1]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-0.0]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-parallel-inf]
0.00s teardown tests/infer/test_gradient.py::test_plate[DiffTrace_ELBO-reparam]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.01--1.5707963267948966]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.4-0.6--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9--0.9]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-1.0]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-0.5]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[2.0-1.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape2-df_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-6]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-5-unclamped]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-0.0-0.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar': True}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-0-7]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-1-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-3-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-1-1-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_cat[1-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-2]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-1.0]
0.00s teardown tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[False-False-100-expected0]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[1-sequential-sequential-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[None-loss-10]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.4-0.6-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.9-0.9]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-0.5]
0.00s teardown tests/distributions/test_omt_mvn.py::test_log_prob[AVFMultivariateNormal]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape2-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape1-df_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape0-value_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-11]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LogNormal-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-Size-sample_shape0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-2-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-0-3]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs18-dims18-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-5]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-3-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-3-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-2-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-()-(5,)-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-()-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-2-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-2-1-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-2-1-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-3]
0.00s teardown tests/infer/test_enum.py::test_elbo_rsvi[parallel]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.5-0.5--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5-0.0]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-4-sparse]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-1.0]
0.00s teardown tests/distributions/test_rejector.py::test_exponential_elbo[0.5-0.5]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape1-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape1-df_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-6]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Uniform-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-list-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-0.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-3-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar': True}-0-8-6]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 4}-7-12]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-0-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-8]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-()-(4, 1, 1)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-1-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_expand[2-(5,)-()-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-2-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-()-(4,)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4--0.5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-1]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.3--1.5707963267948966]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.2-0.8-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0--0.9]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-1.0]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[5.0-5.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape1-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape1-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape1-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape1-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape2-df_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-18]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Poisson-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-tuple-sample_shape0]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_dependent_matern_gp[1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-3-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[mcmc-{}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-50]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-None-batch_shape0]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-2-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-2-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-3-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-3-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-2-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-1-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-1-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(4,)-()]
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_nonreparameterized_N_is_5
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-6]
0.00s teardown tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[False-True-50-expected1]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[rsvi-Trace_ELBO]
0.00s teardown tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[Trace_ELBO-nonreparam]
0.00s teardown tests/infer/test_gradient.py::test_plate[TraceGraph_ELBO-nonreparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_rsvi[None]
0.00s teardown tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[range-8-100-0.01]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.2-0.8-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5-0.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape1-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape2-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape0-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape2-df_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-12]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[VonMises-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Exponential-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-Size-sample_shape2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-3-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_hetero_regional_smoke[mcmc-{'haar': True}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-0-3]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),0]-(4,)]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-None-iplate]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(3, 1)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-2-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-1-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-2-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-3-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-2-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-2-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-2-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-2-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-0-1-(4,)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1-0.0]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-3-1-5]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.1-1.5707963267948966]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-0.5-0.5-1.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.1-0.9--0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.1-0.9--0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.5-0.5-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9--0.5-0.5]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[0.5-1.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape2-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape0-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape2-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape0-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape0-df_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-2]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LogNormal-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-3]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-0-3]
0.00s call     tests/contrib/oed/test_xexpx.py::test_xexpx[argument0-output0]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-6]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-3-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-2-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-2-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-2-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-1-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-3-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-3-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-1-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-3-2-()]
0.00s teardown tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-True-280-expected4]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[1-sequential-parallel-None]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[30.0-1.5707963267948966]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.3-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.5--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.4-0.6--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.2-0.8-0.0--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9-0.9]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-3]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-5.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape2-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape2-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape1-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape2-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape2-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape2-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape1-df_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-16]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[VonMises-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Normal-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Normal-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LogNormal-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Uniform-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-0.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'num_quant_bins': 4}-7-7]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-2-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-0-1-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-2-3]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[10.0--1.5707963267948966]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.1-0.9-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.01-0.5-0.5-0.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8-0.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9-0.0]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-5.0]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[0.2-2.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape0-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape0-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape0-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape1-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape0-df_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-3]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Dirichlet-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-True-tuple-sample_shape1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{'haar': False}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_unknown_start_smoke[{'haar_full_mass': 4}-0-8-6]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-wxi,xyi,yzi,zwi->i]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-8]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-2-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-2-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-1-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-0-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-3-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-1-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-2-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-1-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-2-1-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-1-3-()]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-1-1-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[sequential-False-None]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[100.0-1.5707963267948966]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[100.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-0.5-0.0-1.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.4-0.6--0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.1-0.9-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.5--0.5]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[2.0-2.0]
0.00s teardown tests/distributions/test_rejector.py::test_exponential_elbo[0.5-1.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape1-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape1-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape0-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape2-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape1-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape2-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape1-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape0-df_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-10]
0.00s teardown tests/distributions/test_empirical.py::test_sample_examples[sample1-weights1-expected_mean1-expected_var1]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Categorical-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[OneHotCategorical-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-Size-sample_shape1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0--1.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-3-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(2,)-(2,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-2-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-1-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-3-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-2-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-3-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-3-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-1-(3, 2)-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-2-1]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-2.0]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-1-1-(3, 2)]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5--0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9--0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.9--0.5]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[5.0-0.2]
0.00s teardown tests/distributions/test_rejector.py::test_exponential_elbo[0.5-2.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape1-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape0-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape2-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape1-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape0-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape2-df_shape1]
0.00s teardown tests/distributions/test_ig.py::test_log_prob[0.5-3.0-4.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-5]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Bernoulli-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_log_prob[1e-06]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-7-3]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),:]-(4, 3)]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...,z(3)]-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-3-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-1-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-2-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-0-1-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[3-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-0-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-0-1-(4,)]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[2-1-1-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[None-True-None]
0.00s teardown tests/infer/test_jit.py::test_svi[JitTraceMeanField_ELBO-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[1.0-0.9]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[2-None-sequential]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[30.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.5-0.5-0.5--0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.9-0.0]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-3-uniform]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape2-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape1-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape0-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape2-df_shape2]
0.00s teardown tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-3]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Poisson-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Categorical-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Bernoulli-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Cauchy-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Exponential-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateNormal-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-list-sample_shape1]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-2-lgssm]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-3-1]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoMultivariateNormal]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs17-dims17-2]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-3-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-2-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-1-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[2-1-(3, 2)-()]
0.00s teardown tests/infer/test_elbo_mapdata.py::test_elbo_mapdata[iplate-8-100-0.018]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.99-0.5-0.5-0.9-0.9]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.4-0.6--0.5-0.0]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-5.0]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[0.5-5.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape2-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape1-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape0-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape1-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape1-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape2-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape2-loc_shape1-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape0-df_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(4,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-10]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Bernoulli-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LKJCorrCholesky-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-Size-sample_shape2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-2-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_regional_smoke[svi-{}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'arrowhead_mass': True}-0-3]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-()-(5,)-()-(3, 2)]
0.00s call     tests/distributions/test_pickle.py::test_pickle[MixtureSameFamily]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-6]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape5]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-3-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-0-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-3-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-1-3-()]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-5-sparse]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-2.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape0-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape0-loc_shape0-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape2-df_shape1]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape0-loc_shape2-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape0-df_shape0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape2-df_shape2]
0.00s teardown tests/distributions/test_mvn.py::test_shape[3-cov_shape1-loc_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape0-value_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-(3, 2)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-18]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Beta-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[FoldedNormal-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-list-sample_shape0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-0-7]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),0,0]-(5,)]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-6]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-3-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-2-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-1-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(4,)-()]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-1-1-()]
0.00s teardown tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-True-200-expected3]
0.00s teardown tests/infer/test_svgd.py::test_conjugate[RBFSteinKernel-multivariate]
0.00s teardown tests/infer/test_predictive.py::test_posterior_predictive_svi_auto_delta_guide[False]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[1-sequential-None-None]
0.00s teardown tests/distributions/test_stable.py::test_additive[1.9-0.2-0.8-0.5-0.5]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.1-0.9-0.0-0.0]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-100]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape0-df_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-8]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Categorical-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Stable-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Cauchy-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape2-10]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[2-()-()-9]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{}-7-7]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs14-dims14-3]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs8-dims8-2]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(2,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-1-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-2-2-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-2]
0.00s teardown tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-True-18-expected5]
0.00s teardown tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[DiffTrace_ELBO-nonreparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-parallel-parallel-None]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[10.0-1.5707963267948966]
0.00s teardown tests/distributions/test_stable.py::test_normal[2-0]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-5.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape2-loc_shape0-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape0-df_shape2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape1-loc_shape0-df_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(4,)-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-16]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Dirichlet-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Beta-False-initial_shape0-proposed_shape0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-1-7]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-()-()-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_condition[1-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-0-2-(4,)]
0.00s teardown tests/infer/mcmc/test_adaptation.py::test_adaptation_schedule[True-False-150-expected2]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[10.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_additive[0.5-0.2-0.8--0.9-0.5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-2-uniform]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-0.2]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape1-loc_shape1-df_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(4,)-1-1]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Geometric-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-tuple-sample_shape2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-3-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-0-7]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs15-dims15-3]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(3, 2)-()]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[(3, 2)-(2, 3, 1)-[1, 2]--2]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-4]
0.00s teardown tests/infer/test_gradient.py::test_plate[Trace_ELBO-nonreparam]
0.00s teardown tests/distributions/test_mvt.py::test_shape[1-cov_shape0-loc_shape1-df_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-(3, 2)-3-1]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Geometric-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Cauchy-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Uniform-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-tuple-sample_shape1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[MultivariateNormal-batch_shape0-3-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-7-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-7]
0.00s teardown tests/ops/test_gaussian.py::test_logsumexp[1-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-2-2-()]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_nonreparameterized_decaying_baseline
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[0.1-0.9]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.03-1.5707963267948966]
0.00s teardown tests/distributions/test_mvt.py::test_shape[5-cov_shape1-loc_shape2-df_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-8]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Bernoulli-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Dirichlet-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-Size-sample_shape0]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[GPRegression_y1D]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{}-7-7]
0.00s teardown tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_rsample[True-relu]
0.00s teardown tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-4]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist3]
0.00s teardown tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[Trace_ELBO-reparam]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.2-0.2]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[AVFMultivariateNormal]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-10]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Normal-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Gamma-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-list-sample_shape0]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoMultivariateNormal]
0.00s teardown tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-4]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-2-2-()]
0.00s teardown tests/infer/test_autoguide.py::test_predictive[AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.01-0.5]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape0-value_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(4,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-15]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Geometric-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Dirichlet-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Gamma-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-10-unclamped]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape2-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'num_quant_bins': 2}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.00s teardown tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-6-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-8]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-1-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-2-1-()]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8--1.0]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-6]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_gaussian_hmm_elbo[1-1-1-(4,)]
0.00s teardown tests/infer/test_enum.py::test_vectorized_importance[100000]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.01-1.5707963267948966]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-5-random]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[1.0-2.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-(3, 2)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[(3, 2)-()-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-1-4]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Categorical-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-list-sample_shape1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.00s teardown tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_rsample[False-relu]
0.00s call     tests/contrib/oed/test_xexpx.py::test_xexpx[argument1-output1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-6]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-1-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_add[1-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-5]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-full-inf]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-5]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-5]
0.00s teardown tests/distributions/test_rejector.py::test_exponential_elbo[0.25-0.5]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_log_prob[1.0]
0.00s teardown tests/contrib/oed/test_finite_spaces_eig.py::test_posterior_finite_space_model
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-7-3]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs16-dims16-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-5-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-5]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 1, 1)-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-4]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[2-parallel-parallel-10000]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[0.5-0.2]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[5.0-2.0]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[2.0-2.0]
0.00s teardown tests/distributions/test_mvt.py::test_shape[3-cov_shape2-loc_shape1-df_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_filter_sample[()-()-2-1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Beta-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LowRankMultivariateNormal-True-Size-sample_shape0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[mcmc-{}-7-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-2-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-8]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(2,)-(3, 1)-()]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-200-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[None-differentiable_loss-10]
0.00s teardown tests/distributions/test_von_mises.py::test_sample[0.3-1.5707963267948966]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Categorical-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Bernoulli-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-list-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0--1.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 2}-7-7]
0.00s call     tests/ops/test_gaussian.py::test_expand[1-()-(5,)-()-()]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-10-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-3-5]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-()-()-()]
0.00s teardown tests/infer/reparam/test_transform.py::test_log_normal[(2, 3)]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-1.0]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-2]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-2]
0.00s teardown tests/infer/test_gradient.py::test_plate[Trace_ELBO-reparam]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-Size-sample_shape2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-4-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-3-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-7]
0.00s teardown tests/ops/test_special.py::test_log_beta_stirling[0.02]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-0-3-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-3]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[rsvi-RenyiELBO]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.17-0.41]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.44-0.41]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-2.0]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[0.5-2.0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-list-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(3, 2)-()-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-0.0-0.0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Gamma]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-2-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-6-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-7]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-1-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-2-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-1-3-(4,)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0-1.0]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[(3, 2)-(6,)-[2, 1, 3]--1]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_feasible-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-30]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_gamma_elbo[1.0-1.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-3]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-6-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-7]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-None-batch_shape2]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(4,)-(6, 5)]
0.00s teardown tests/infer/reparam/test_stable.py::test_symmetric_stable[()]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-parallel-inf]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-1]
0.00s teardown tests/infer/test_gradient.py::test_plate_elbo_vectorized_particles[TraceGraph_ELBO-reparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[1.0-0.1]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[1.0-5.0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-Size-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-3-2]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'num_quant_bins': 2}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-7-7]
0.00s teardown tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-0]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-6-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-6-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-5-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-5-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-7]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(2,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-0-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-1-2-()]
0.00s teardown tests/ops/test_contract.py::test_ubersum[abi,b->,b,ai,abi-i]
0.00s teardown tests/infer/reparam/test_unit_jacobian.py::test_normal[(4, 5)]
0.00s teardown tests/infer/reparam/test_haar.py::test_normal[(6,)--1-False]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-1-1-(3, 2)]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gaussian_hmm[30]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-5]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-3]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-()-(5,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(3, 2)-(5,)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-2-2]
0.00s teardown tests/contrib/gp/test_models.py::test_inference[VariationalGP_y1D]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_heterogeneous_sir_smoke[svi-{}-7-3]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs11-dims11-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-4-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-3-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-6]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-0-2-(3, 2)]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[(3, 2)-(2, 5)-[2, 3]--1]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[()-(6,)-[2, 1, 3]--1]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-multivariate-latent_dist0]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[0.1-0.75]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_sample-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_reshape.py::test_idempotent[2-1]
0.00s teardown tests/distributions/test_rejector.py::test_gamma_elbo[1.0-1.0]
0.00s teardown tests/distributions/test_omt_mvn.py::test_log_prob[OMTMultivariateNormal]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-13]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'haar': False}-7-7]
0.00s teardown tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-10-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-3-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-2-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-7]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-0-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-0-1-()]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8-0.0]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-3-None-3]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-4]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-2-unclamped]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-2]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-1-1.5]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-7-3]
0.00s teardown tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-6]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_logsumexp[3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-0-1-()]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(2, 3, 1)--2-0.5]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-parallel-1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-200-1]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-2]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-1]
0.00s teardown tests/distributions/test_ig.py::test_sample[2.5-3.3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Gamma-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-10-clamped]
0.00s teardown tests/contrib/oed/test_finite_spaces_eig.py::test_marginal_finite_space_model
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-3-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-5-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-5-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-6]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(4,)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-2-3-()]
0.00s teardown tests/infer/reparam/test_unit_jacobian.py::test_normal[(6,)]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[(4,)-(4, 2)-[1, 3]--2]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[(4,)-(6,)-[2, 1, 3]--1]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-2-2]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-None-1]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[10-1-5-2]
0.00s teardown tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-sequential-1-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[2-sequential-sequential-None]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_mean-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape2-value_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-11]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Gamma-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[VonMises]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(3, 2)-()-9]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[GPRegression_y2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-3-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{}-7-7]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-10-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-10-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-5-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-4-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-3-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-10-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-10-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-6-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-4-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-3-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-4-8]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[3-2-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-3]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[differentiable_loss-None-parallel-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[None-loss_and_grads-1]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-100]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-1-0]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_beta[4.0-0.5]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[MultivariateNormal]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Beta-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-list-sample_shape2]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.05]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-0.1-100]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-3-lgssmgp]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_whiten[SparseGPRegression_y1D]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[mcmc-{}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-0-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-5-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-5-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-4-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-2-5]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape3]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(3, 2)-(6, 5)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-5]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-2-1]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_num_chains[False-1-2]
0.00s teardown tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[sequential-True-None]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-1-1-1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape1-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-5-unclamped]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_whiten[SparseGPRegression_y2D]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_whiten[GPRegression_y1D]
0.00s teardown tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[GPRegression_y1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-3-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'jit_compile': True}-7-7]
0.00s teardown tests/ops/test_welford.py::test_welford_arrowhead[True-1000-5-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-(2, 3)-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-2-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-5]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-1-()-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceTMC_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-Trace_ELBO-full]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-3-None-2]
0.00s teardown tests/infer/test_enum.py::test_backwardsample_posterior_restrictions[False-parallel-1-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[sequential-loss_and_grads-1]
0.00s teardown tests/distributions/test_mvt.py::test_rsample[2-3.9]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.2]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-3-lgssmgp]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-3-lgssmgp]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_whiten[GPRegression_y2D]
0.00s teardown tests/contrib/gp/test_models.py::test_inference[VariationalSparseGP_y1D]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-3-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-7-3]
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-bi,abi->b-i]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs7-dims7-3]
0.00s teardown tests/ops/test_welford.py::test_welford_arrowhead[False-1000-5-0]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-5-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-6-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-10-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-3-5]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(3, 2)-(7,)]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[(3, 2)-(4, 2)-[1, 3]--2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-6]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-0.0]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-1-1-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-Trace_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[mixture-2-200-1]
0.00s teardown tests/infer/test_predictive.py::test_deterministic[event_shape1-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-None-parallel-parallel]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_mean-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Normal-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(4,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-()-()-9]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.02]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[37-imgp-1-0.5]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-7-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-5-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(6,)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-()-2-6]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-1-2-(3, 2)]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[(4,)-(2, 3, 1)-[1, 2]--2]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[()-(4, 2)-[1, 3]--2]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-1-1-(3, 2)]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[nonreparam-ReweightedWakeSleep]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.17-0.41]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[cpp-10]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-9]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[LKJCorrCholesky]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[5-(2, 3)-(7,)-9]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.01]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-1.0-0.0]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-3-lgssm]
0.00s teardown tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[SparseGPRegression_y2D]
0.00s teardown tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[SparseGPRegression_y1D]
0.00s teardown tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[GPRegression_y2D]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar': True}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[mcmc-{'haar_full_mass': 2, 'jit_compile': True}-7-7]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,ixj->ji]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(3, 2)-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-6]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(3, 2)-(7,)]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[()-(2, 5)-[2, 3]--1]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-1-1-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-parallel-inf]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-1-2-1]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_restrictions[False-sequential-1-False]
0.00s teardown tests/distributions/test_mvn.py::test_shape[1-cov_shape1-loc_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape0-value_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-5-unclamped]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-10]
0.00s teardown tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-2-NcpContinuous]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-1.0-1.0]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-2-lgssm]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-2-lgssmgp]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-4-2-lgssm]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-6]
0.00s teardown tests/ops/test_special.py::test_log_beta_stirling[1e-08]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape4]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(2,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-0-1-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-2]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s teardown tests/infer/test_inference.py::test_energy_distance_univariate[0.0001]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(3, 2)-(5,)-9]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.10]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.0001]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_log_prob[1e-08]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0--1.0--1.0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[mcmc-{'haar_full_mass': 2}-7-3]
0.00s call     tests/contrib/oed/test_xexpx.py::test_xexpx[argument2-output2]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-20-plate]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-3-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-10-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-(3, 2)-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-()-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-6]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-()-(7,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-6]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-1-1-()]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-parallel-1]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-Trace_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-Trace_ELBO-full]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-5]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape1-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape0-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-list-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(4,)-(5,)-9]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-10]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0--1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-1.0-1.0]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_timeseries_models[11-imgp-1-2.5]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[SparseGPRegression_y1D]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[WhiteNoise0]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Exponential]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-20-5-plate]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-10-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-8]
0.00s teardown tests/ops/test_special.py::test_log_beta_stirling[1e-06]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-()-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-()-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-2-()-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_size_error[naive_ubersum-ab,bc->-shapes1-b]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-4]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-0.5]
0.00s teardown tests/infer/test_jit.py::test_infer_discrete[sample-10]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[1-parallel-None-None]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_mask.py::test_mask[1-2-0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-9]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(3, 2)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(4,)-(5,)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0--1.0--1.0]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-4-3-lgssm]
0.00s teardown tests/contrib/gp/test_models.py::test_inference[VariationalGP_y2D]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-3-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-6-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-5-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-6]
0.00s teardown tests/ops/test_special.py::test_log_beta_stirling[0.0001]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(4,)-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-4]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_feasible-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape0-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-9]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-10-clamped]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-10]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-10]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0--1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{'haar': False}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_seir_smoke[svi-{}-7-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-6]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(3, 1)-(2,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-2-2-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-0-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_cat[2-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-2-()-(4,)]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[(4,)-(2, 5)-[2, 3]--1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-2]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-1-1-(4,)]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-dense-3]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape2-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape2-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape2-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(4,)-(5,)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-1.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0--1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-1.0-0.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-3-1]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges150]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-(2, 3)-10-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-6-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(5, 4)-(6,)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-5]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-3-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-3-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(3, 2)-(6, 5)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-2]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.29-0.44-0.33]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[2-None-sequential-None]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape1-value_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-17]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-list-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-(3, 2)-(5,)-9]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-10]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-0.0--1.0]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize1]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Periodic]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Matern52]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Matern32]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(5, 4)-()-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-7]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-()-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-()-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-0-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(4,)-(6, 5)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(4,)-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-2-6]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_nested_plate[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-False-NUTS-_empty_model]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-Trace_ELBO-full]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-4]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-3-sparse]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape1-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape2-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape1-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-12]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-10-unclamped]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-0.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-0.0-0.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-2-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-0-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_overdispersed_seir_smoke[{}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': None}-7-3]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-None-plate]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-4-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-8]
0.00s teardown tests/ops/test_special.py::test_log_beta_stirling[0.01]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-()-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(4,)-(7,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-5]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[5-1-1-()]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[4-1-1-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape2-True-Trace_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-None-3]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-1-1-2]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-2]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_sample-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape1-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape0-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape1-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape1-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_conjugate.py::test_variance[dist0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(3, 2)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(4,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-()-(5,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[5-(6,)-(7,)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0--1.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0--1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0--1.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-1.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0--1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0--1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-1.0-0.0]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Polynomial]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Brownian]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Constant]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_simple_sir_smoke[svi-{'guide_rank': 'full'}-7-3]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-20-iplate]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-7]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-()-(3, 1)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-0-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-3-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-1]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-0.5]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-parallel-1]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-full-reparam]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape0-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape1-value_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-()-(5,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(3, 2)-()-9]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-10]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-0.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0--1.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-0.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0-0.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-0.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0--1.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-0.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AbsTransform-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_coalescent_likelihood_smoke[svi-{'haar': False}-7-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-3-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-2-8]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(2,)-(3, 1)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-1-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-0-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_cat[3-(4, 7, 6)--1-(2, 1, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-1-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[(2, 3)-1-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-2]
0.00s teardown tests/infer/reparam/test_haar.py::test_normal[(6,)--1-True]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-1-1-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape4-True-Trace_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape3-True-Trace_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-Trace_ELBO-subsample]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-100]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_sample-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.0--0.5-0.5]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape2-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape2-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape2-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape2-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-0-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-20-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-10-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-10-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[2-(4,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(3, 2)-(5,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(4,)-(5,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[11-()-(7,)-9]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-100]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-0.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0--1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-1.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-1.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0--1.0-0.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-3-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-7-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-4-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-8]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-()-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-()-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(3, 1)-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-2-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-0-2-(3, 2)]
0.00s teardown tests/infer/reparam/test_split.py::test_normal[()-(2, 3, 1)-[1, 2]--2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[()-1-1-3]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-2.0]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-Trace_ELBO-full]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-multivariate-latent_dist1]
0.00s teardown tests/infer/test_smcfilter.py::test_smoke[2-1-5-2]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_restrictions[False-parallel-1-False]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_normal_bern_elbo_gradient[0.22-0.17-0.33]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape1-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape0-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape0-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape1-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape2-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape0-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-exponential-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-8]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Stable]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-10-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-20-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-()-(5,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[11-(2, 3)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[5-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[3-()-(4, 5)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0--1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-0.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-0.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[AffineTransform-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-3-7]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs5-dims5-4]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-None-iplate]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-20-plate]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-3-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-7]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(2,)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-1-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[1-1-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-0-1-()]
0.00s teardown tests/ops/test_gaussian.py::test_cat[1-(4, 7, 6)--1-(2, 1, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-2-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(4,)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-3-()-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-6]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape1-True-TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_tmc.py::test_tmc_categoricals[diagonal-2-200-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_3[masked-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[None-loss-1]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_bernoulli_entropy_grad[1.0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape2-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape1-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape2-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[normal-normal-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Multinomial-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-10-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[11-()-(5,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[5-(2, 3)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[3-(6,)-(7,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[3-()-(7,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[2-(6,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[2-()-(7,)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-3-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0--1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-0.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-0.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-1.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-0.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-1.0-1.0]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[RBF]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[SigmoidTransform-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-2-7]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->a-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-(2, 3)-4-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-6]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-()-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(4,)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_cat[2-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_cat[1-(4, 7, 6)--1-(2, 1, 3)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-5]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_normal[(6,)--1-1.0]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[2-1-1-()]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape0-True-TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-subsample-nonreparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[None-differentiable_loss-1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape0-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape0-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape0-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape1-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-exponential-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_univariate[exponential-normal-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_sample_examples[sample0-weights0-expected_mean0-expected_var0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[BetaBinomial-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-5-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-20-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_constant_rate[7-(3, 2)-(5,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[7-(4,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[2-(6,)-(7,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[2-(6,)-()-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-0.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-0.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0--1.0-0.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[PowerTransform-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-7-3]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-do-normal_normal_model]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-5-None-plate]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-4-5]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-()-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(2,)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-()-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(2,)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(2,)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-()-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-()-(2,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-3-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-1-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_cat[2-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-1-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-1-()-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-6]
0.00s teardown tests/infer/test_enum.py::test_elbo_normals[loss_and_grads-parallel-None-parallel]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[map-1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape2-value_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape1-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBetaBinomial-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-list-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-(4,)-(5,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-()-(5,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[11-(6,)-(7,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[5-(2, 3)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[3-(2, 3)-(4, 5)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-1.0-0.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0--1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0--1.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-0.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-0.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-0.0-0.0]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-2-lgssm]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[WhiteNoise1]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Linear]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-3-1]
0.00s setup    tests/test_generic.py::test_not_implemented[minipyro]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4)]-(4, 3)]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-4-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-6]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape4]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 3, 2)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-()-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(3, 1)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(2,)-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-3-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-1-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-1-1-()]
0.00s teardown tests/ops/test_gaussian.py::test_reshape[1-(6,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-3-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-3-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-2-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-3-()-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-3]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[3-1-1-(3, 2)]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-30]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape2-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape1-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape2-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape0-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape2-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape2-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape1-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape0-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape0-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape0-value_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-3]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape0-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Binomial-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Beta-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-5-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-5-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-5-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-5-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-2-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-10-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-20-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[11-(2, 3)-(7,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[11-(6,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[5-(6,)-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[5-(6,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[5-()-(7,)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[3-(6,)-(4, 5)-9]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-10.0-100]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0--1.0-0.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0--1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0--1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-0.0-0.0]
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[RationalQuadratic]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-0-7]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-5-5-plate]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(6,)-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-6]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(2,)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-()-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-()-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-1-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-1-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-0-2-()]
0.00s teardown tests/ops/test_gaussian.py::test_cat[1-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-1-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-3-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-3-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-2-1-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(2, 3)-1-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_transformed_hmm_shape[(4,)-1-1-6]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_stable_hmm_smoke[1-1-1-(3, 2)]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_observed[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[2-parallel-parallel-None]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[sample-100]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape1-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape2-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape0-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape1-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape1-value_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-17]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[DirichletMultinomial-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[FoldedNormal]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-5-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-20-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[3-(2, 3)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[2-(2, 3)-()-9]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-1.0-100]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0--1.0--1.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-2-7]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoLaplaceApproximation]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-5-plate]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-None-plate]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-3-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-(3, 2)-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-6]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-()-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(2,)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(2,)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-2-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-3-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-1-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-1-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-1-2-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-2-1-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-0-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_cat[2-(4, 7, 6)--1-(2, 1, 3)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-4]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 5)--1-1.0]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[sequential-True-None]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[nonreparam-TraceGraph_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-True-1]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-5-sparse]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-4-uniform]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape1-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape2-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape2-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape0-value_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-0-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Dirichlet-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(3, 2)-2-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-20-2-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-20-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[11-(2, 3)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[5-()-(4, 5)-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[3-(2, 3)-(7,)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-3-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-2-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-0.0-1.0]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-2-lgssmgp]
0.00s teardown tests/contrib/oed/test_finite_spaces_eig.py::test_dv_finite_space_model
0.00s teardown tests/contrib/gp/test_kernels.py::test_kernel_forward[Coregionalize0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-3-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_overdispersed_seir_smoke[{'num_quant_bins': 2}-7-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-5]
0.00s teardown tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape2-_cummin]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape3]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(3, 1)-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(2,)-(2,)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-()-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(2,)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-2-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-1-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-2-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-1-2-()]
0.00s teardown tests/ops/test_gaussian.py::test_cat[1-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[3-2-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-2-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-1-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_cat[3-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-1-6]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-1.0]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[reparam-RenyiELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[2-sequential-None-None]
0.00s teardown tests/infer/test_enum.py::test_elbo_plate[2-None-None-None]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-1-sparse]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Geometric-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedDirichlet-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-list-sample_shape2]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[()-2-20-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(4,)-10-10-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[3-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[2-()-(4, 5)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-1.0-1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[0.0-0.0-1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-1.0-1.0-1.0]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[17-2-3-lgssm]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-1-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-3-5]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 3, 2)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-()-(3, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 3, 2)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-3-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-2-2-()]
0.00s teardown tests/ops/test_gaussian.py::test_cat[2-(4, 7, 6)--1-(2, 1, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[1-1-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-2-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-1-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-3-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-0-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_cat[1-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s teardown tests/infer/reparam/test_studentt.py::test_moments[(2, 3)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_studentt_hmm_shape[()-1-2-1]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[sequential-False-None]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-1]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_median-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_spanning_tree.py::test_log_prob[6]
0.00s teardown tests/distributions/test_mvt.py::test_rsample[2-9.1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape2-component1_shape1-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape1-value_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-0-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_ig.py::test_sample[2.5-4.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-13]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[GammaPoisson-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Bernoulli-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[MultivariateStudentT-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[HalfCauchy-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-True-list-sample_shape2]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-2-5-clamped]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[11-()-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[3-(6,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[2-(2, 3)-(7,)-9]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-0.1-10]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0--1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-1.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_transform_batch[ExpTransform-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-7-3]
0.00s call     tests/distributions/test_pickle.py::test_pickle[SpanningTree]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-2-5]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 1, 1)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(4, 1, 1)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-()-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(4, 1, 1)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(2,)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(3, 1)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(3, 1)-(3, 1)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[3-1-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-3-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-1-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[2-1-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-1-2-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_cat[3-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-1-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-1-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-0-3-(3, 2)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-4]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[parallel-True-3]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-3-True]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[parallel-differentiable_loss-1]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-2-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-1-uniform]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-1-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_enumerate_support[6]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape2-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape1-value_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape2-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Binomial-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ExtendedBinomial-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Normal-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedBeta-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[RejectionExponential-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[(3, 2)-2-2-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[11-(6,)-()-9]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-10.0-100]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-2-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[1.0-0.0-1.0--1.0]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-3-lgssmgp]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-2-lgssmgp]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-2-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,a,ab,b,b,b,b->a-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(6,)-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-()-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(6,)-3-5]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(3, 1)-(4, 1, 1)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-(4, 3, 2)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-3-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-1-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-()-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-1-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-1-()-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-1-3]
0.00s teardown tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[None-False-None]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-2-True]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.0-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-0.5-0.9-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-0.5--0.5-0.5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-2-uniform]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape0-value_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape1-component1_shape0-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape0-component1_shape2-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape0-component1_shape2-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape2-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape1-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape0-value_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-9]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Poisson-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[ShapeAugmentedGamma-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Uniform-False-tuple-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-2-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0-0.0-1.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_2[-1.0--1.0-1.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-0-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[same-()-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[full-()-2-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-5]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape3]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(2,)-(4, 1, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-()-(3, 1)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-0-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-2-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-1-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-0-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_cat[2-(4, 7, 6)--2-(1, 1, 2, 3)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3--1.0]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-4]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-1-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-2-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-3]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-1-3]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[RBFSteinKernel-univariate-latent_dist3]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.99-0.0]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-5-sparse]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-2-sparse]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape2-component0_shape2-component1_shape0-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape2-value_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape2-component1_shape0-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape0-component1_shape2-value_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-10]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape1-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Binomial-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveDirichlet-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[2-(4,)-()-9]
0.00s teardown tests/distributions/test_coalescent.py::test_simple_smoke[11-()-(4, 5)-9]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-3-3]
0.00s teardown tests/contrib/timeseries/test_lgssm.py::test_generic_lgssm_forecast[11-2-3-lgssm]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-2-7]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->d-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(3, 2)-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[(7,)-(3, 2)-3-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-8]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(4, 3, 2)-(2,)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-(2,)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[2-1-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-2-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_matrix_and_mvn_to_gamma_gaussian[2-1-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[3-3-()-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-2-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-0-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_cat[3-(4, 7, 6)-1-(1, 1, 2, 3)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-1]
0.00s teardown tests/infer/reparam/test_haar.py::test_uniform[(4, 2)--2-False]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-Trace_ELBO-full]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.27-0.43]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[sequential-differentiable_loss-1]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_mean-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.0-0.5-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-0.5-0.0-0.5]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape1-value_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape0-component0_shape1-component1_shape1-value_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Cauchy-True-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[NaiveBeta-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_sequential[(4,)-2-10-unclamped]
0.00s teardown tests/distributions/test_coalescent.py::test_likelihood_vectorized[()-5-2-clamped]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-1-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-2-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-3]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-2-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-0-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[(7,)-(3, 2)-4-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-6]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-2-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-1-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-2-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[1-2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-2-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-2-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-0-3-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-1-1-1]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-1]
0.00s teardown tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[parallel-True-None]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceTMC_ELBO-subsample]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[rsvi-TraceEnum_ELBO]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[reparam-Trace_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-3-False]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.55-0.43]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[sequential-differentiable_loss-10]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_feasible-Trace_ELBO]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-1-sparse]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-4-random]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape0-value_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-12]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_conjugate.py::test_variance[dist1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-2-2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-2-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->ac-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve[valid-()-2-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-(6,)-2-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-3-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-6]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(3, 1)-(4, 3, 2)-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(4, 3, 2)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[()-(2,)-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_condition[2-2-()-(4,)]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(2, 3)-3-2-2]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-5]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-True-HMC-_empty_model]
0.00s teardown tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[parallel-True-3]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_3[masked-10]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_3[batch-1]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_median-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.0-0.5-0.5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-5-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-5-uniform]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-4-sparse]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-4-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-3-sparse]
0.00s teardown tests/distributions/test_mixture.py::test_broadcast[mask_shape1-component0_shape1-component1_shape1-value_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Exponential-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_coalescent.py::test_log_prob_unit_rate[11-()-()-9]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[0.1-10.0-100]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-1-3]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LinearHMM-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-2-8]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-0-2-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-1-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-1-2-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-1-6]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-4]
0.00s teardown tests/infer/reparam/test_haar.py::test_normal[(2, 3, 1)--2-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_normal[None-loss_and_grads]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_feasible-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-5-uniform]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-1-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-4-sparse]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-3-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-2-random]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-18]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape0-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[VonMises-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-list-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-3-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-3-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-3]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[IndependentHMM-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GammaPoisson-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-2-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_overdispersed_sir_smoke[{'num_quant_bins': 2}-7-3]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-20-iplate]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_cumsum[()-()-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-8]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-3-5]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-None-batch_shape2]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-None-batch_shape3]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-2-()-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-2-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_condition[3-1-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-0-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-()-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-2-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-0-2-()]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[(4,)-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-3-2-5]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-2]
0.00s teardown tests/infer/reparam/test_haar.py::test_uniform[(2, 3, 1)--2-True]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_shape_broadcasting_ok[parallel-True-None]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-sequential-1]
0.00s teardown tests/infer/test_multi_sample_elbos.py::test_outer_inner[RenyiELBO]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[rsvi-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-3-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-5-random]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-2]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype0-sample_shape2-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-True-initial_shape2-proposed_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-3-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-1-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-1-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-1-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[LogNormal-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[GaussianHMM-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[BetaBinomial-batch_shape0-1-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-2]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(3, 2)-(7,)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0--1.0]
0.00s teardown tests/infer/reparam/test_hmm.py::test_stable_hmm_shape_error[()-1-2-3]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-2.0]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-0.0]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-sequential-inf]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.55-0.33]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_median-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_median-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.0-0.0-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.0--0.5-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-0.5-0.5-0.5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-3-uniform]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape2-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Multinomial-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Multinomial-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[RejectionExponential-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-Size-sample_shape0]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-1.0-100]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[1.0-1.0-100]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-3-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-1-1]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(()))[...]-()]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs12-dims12-1]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-20-iplate]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->bc-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->ac-2]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-None-batch_shape0]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-0-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-2-3-(4,)]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enumeration_in_dag[TraceEinsumEvaluator]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-Trace_ELBO-subsample]
0.00s teardown tests/infer/test_autoguide.py::test_predictive[AutoMultivariateNormal]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-3-sparse]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-10]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype1-sample_shape1-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_log_prob[0.11]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-1-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-3-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-2-3]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[InverseGamma-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-3-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-0-3]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2))[z(3)]-(3,)]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->a-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-2-7]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape5]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-()-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-2-3-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-1-3-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-1-2-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-2-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-()-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-2-3-(3, 2)]
0.00s teardown tests/infer/reparam/test_stable.py::test_symmetric_stable[(2, 3)]
0.00s teardown tests/infer/reparam/test_haar.py::test_normal[(2, 3, 1)--2-True]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_null_model_with_hook[1-True-NUTS-_empty_model]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-parallel-1]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.0-0.0-0.5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-2-uniform]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-1-uniform]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-16]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Multinomial-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Gamma-True-list-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-3-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-1-2-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-2-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-4]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedPoisson-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-3-7]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->c-1]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[2-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-1-2-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_enumerate_parallel_plate_ok[parallel-False-None]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-full-0]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-Trace_ELBO-subsample]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-0.5--0.5-1.5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-4-uniform]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-2-uniform]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-5]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[VonMises-True-tuple-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-2-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-1-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-2-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-1-2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[ZeroInflatedNegativeBinomial-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[DirichletMultinomial-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-2-7]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoMultivariateNormal]
0.00s teardown tests/test_generic.py::test_mcmc_interface[pyro-beta_binomial]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs10-dims10-2]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-5-5-iplate]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->bc-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->ac-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_repeat[()-()-4-5]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape0]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-3-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-2-2-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-1-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-1-3-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.00s teardown tests/infer/reparam/test_stable.py::test_symmetric_stable[(4,)]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-Trace_ELBO-full]
0.00s teardown tests/infer/test_inference.py::test_energy_distance_univariate[0]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_2[masked-10]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_feasible-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-0.5-0.9-0.5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-3-uniform]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-3-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-3-uniform]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-2-sparse]
0.00s teardown tests/distributions/test_spanning_tree.py::test_partition_function[6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-6]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_samples[dtype2-sample_shape0-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[VonMises-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-Size-sample_shape2]
0.00s teardown tests/distributions/test_binomial.py::test_beta_binomial_approx_sample[10.0-0.1-100]
0.00s teardown tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-3-NcvContinuous]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[StudentT-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape1-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-2-7]
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-aij->ai-ij]
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-,ai->a-i]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->c-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->-2]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-1-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-1-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-1-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-2-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-0-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-1-1-(3, 2)]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_diagnostics[1]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceGraph_ELBO-subsample]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-4-sparse]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-1-uniform]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-1-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-4-uniform]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-3]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Multinomial-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[VonMises-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-False-tuple]
0.00s teardown tests/distributions/test_conjugate.py::test_mean[dist3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-3-7]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0]-(2,)]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ad-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->b-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->c-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->b-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,b,bc->ac-2]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape2]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-1-3-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-1-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-0-3-(3, 2)]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-4-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-2-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-1-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-5-uniform]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-python-1-sparse]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-15]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Categorical-True-list-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[bp-2-2-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-1-3]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[NegativeBinomial-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Laplace-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-3-7]
0.00s teardown tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[16]
0.00s teardown tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.1]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoDiagonalNormal]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->b-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->a-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->-1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape4]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[2-1-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-1-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[1-2-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-1-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-()-(7,)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceTMC_ELBO-full]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-1-sparse]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-4-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-3-random]
0.00s teardown tests/distributions/test_mask.py::test_mask[1-0-0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-True-sample_shapes1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-4]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Normal-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Gamma-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Beta-batch_shape0-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar': True, 'num_quant_bins': 2}-7-12]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->a-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->ac-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->b-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->b-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ac-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag[mat_size0-2-batch_shape1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-None-batch_shape3]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-(4, 1, 1)-()-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-()-(2,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-1-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-2-1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-0-3-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-2-1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-2-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-1-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-1-2-(4,)]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_median-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_normal[0.5-21]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-4-uniform]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-18]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Binomial-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Dirichlet-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-tuple-sample_shape1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[FoldedDistribution-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Exponential-batch_shape1-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-3-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape0-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-1-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-2-1]
0.00s teardown tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[4]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->bc-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->c-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->ac-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ad-1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-None-batch_shape5]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-3-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-2-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-1-2-(3, 2)]
0.00s teardown tests/distributions/test_stable.py::test_mean[S0-1.0-0.9-1.5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-2-sparse]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-10]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Exponential-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[Stable-True-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-Size-sample_shape2]
0.00s teardown tests/distributions/test_conjugate.py::test_gamma_poisson_log_prob[shape0]
0.00s teardown tests/distributions/test_conjugate.py::test_mean[dist1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-1-1-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-0.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape2-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape0-2-7]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-2-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-7-7]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->c-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->c-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->a-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-10]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-1-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-2-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-1-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-0-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-2-3-(4,)]
0.00s teardown tests/infer/reparam/test_haar.py::test_uniform[(2, 3, 1)--2-False]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data3-expected_log_prob3]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[5-False-TraceTMC_ELBO-subsample]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[3.0-0.25]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-5-random]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[mcmc-cpp-3-sparse]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-14]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LKJCorrCholesky-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Multinomial]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0--1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-3-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Cauchy-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape2-2-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->c-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->a-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ac-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc->ac-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->b-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->b-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-4]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-2-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[2-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-0-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-1-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-1-1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-1-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-2-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-1-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-0-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-2-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-1-2-()]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-2.0]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_feasible-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_feasible-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-cpp-1-uniform]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Exponential-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[ExtendedBetaBinomial]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-3]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Dirichlet-batch_shape1-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-0-3]
0.00s call     tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-aij->aj-ij]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[error-20-5-iplate]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->ac-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->ac-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,a,ab,b,b,b,b->a-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->b-1]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 1, 1)-()-(4, 1, 1)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-3-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-2-1-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-0-2-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-0-2-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-0-1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-2-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-2-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-1-3-(3, 2)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4-0.5]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(4, 2)--2-0.5]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_onehot_entropy_grad[1.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Poisson-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Categorical-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Multinomial-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Uniform-Size-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-4]
0.00s teardown tests/contrib/oed/test_finite_spaces_eig.py::test_marginal_likelihood_finite_space_model
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->d-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->b-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->b-1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape2]
0.00s teardown tests/ops/test_indexing.py::test_value[(4, 3, 2)-()-()-(7,)]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[1-1-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-2-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-3-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-0-2-(4,)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01--0.5]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[reparam-TraceEnum_ELBO]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[map-2]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_gof[approx-python-2-random]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateNormal-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedGamma-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[MultivariateStudentT]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-3-1]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0,:]-(3,)]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ad-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ac-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->d-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->b-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->c-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->c-2]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(3, 1)-(3, 1)-()]
0.00s teardown tests/ops/test_indexing.py::test_value[(2,)-(2,)-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[2-2-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[1-1-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-2-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-0-2-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-2-1-()]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-1-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(4,)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-2-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-1-1-(3, 2)]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data3-expected_log_prob3]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-None-inf]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_3[single-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-3-False]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[3-(2, 3)]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Poisson-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Multinomial-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[VonMises-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Beta-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-tuple-sample_shape1]
0.00s teardown tests/distributions/test_conjugate.py::test_mean[dist0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_independent_subproblems[enum-2-1-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-0.0--1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0--1.0-0.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-3-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_overdispersed_sir_smoke[{}-7-3]
0.00s teardown tests/test_generic.py::test_mcmc_interface[pyro-logistic_regression]
0.00s call     tests/ops/test_contract.py::test_ubersum_size_error[naive_ubersum-ab,bc->-shapes1-b]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->a-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc->ac-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,b,bc->ac-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,b,bc->ac-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->d-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc->ac-1]
0.00s teardown tests/ops/test_indexing.py::test_value[(3, 1)-(2,)-(3, 1)-()]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-2-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-1-2-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-TraceEnum_ELBO]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[nonreparam-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-14]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBinomial-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-tuple-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-1.0--1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-1-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->c-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->c-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,bc->a-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-5]
0.00s teardown tests/ops/test_special.py::test_log_binomial_stirling[0.10]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape4]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian_2[3-3-()-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_matrix_and_mvn_to_gaussian[3-1-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(4,)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-0-3-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-2-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[0-2-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[2-1-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[2-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-2-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-1-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-1-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[1-0-2-(4,)]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-0.0]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-sequential-inf]
0.00s teardown tests/infer/test_valid_models.py::test_plate_param_size_mismatch_error[shape6-False-TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[reparam-TraceGraph_ELBO]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-subsample-reparam]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_mean-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-9]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Stable-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[HalfCauchy-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedBeta-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Exponential-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-tuple-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-4]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 2}-7-12]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-1.0]
0.00s teardown tests/test_generic.py::test_trace_handler[pyro-beta_binomial]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-20-None-iplate]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc->ac-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,bc,cd->ad-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,,b,bc->ac-1]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[1-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_condition[1-1-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[2-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[1-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[1-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[1-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-2-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-0-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-1-3-(3, 2)]
0.00s teardown tests/infer/reparam/test_haar.py::test_uniform[(6,)--1-True]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-sequential-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_1[masked-1]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_sample-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_feasible-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.5]
0.00s teardown tests/distributions/test_mvn.py::test_shape[1-cov_shape2-loc_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ExtendedBetaBinomial-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[ShapeAugmentedDirichlet-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LogNormal-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Uniform-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-tuple-sample_shape0]
0.00s teardown tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-2-NcvContinuous]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-0.0-1.0]
0.00s teardown tests/poutine/test_mapdata.py::test_model_guide_mismatch[ok-None-5-iplate]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->ac-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-ab,b,bc->a-2]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-None-batch_shape1]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[1-3-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[1-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-()-()]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data0-expected_log_prob0]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-sequential-inf]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-subsample-reparam]
0.00s teardown tests/infer/test_autoguide.py::test_predictive[AutoNormal]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape0-event_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-2]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Stable-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LowRankMultivariateNormal-False-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-Size-sample_shape1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar_full_mass': 3, 'num_quant_bins': 2}-7-12]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-7-7]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,b,bc->b-2]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[3-3-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[3-1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[1-3-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-2-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[3-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[0-2-1-(3, 2)]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-parallel-inf]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[reparam-ReweightedWakeSleep]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-full-nonreparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-30-False]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_feasible-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_mean-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_median-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.0--0.5-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-0.5-0.0-1.5]
0.00s teardown tests/distributions/test_mvn.py::test_shape[3-cov_shape2-loc_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-8]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[OneHotCategorical-False-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[DirichletMultinomial-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Stable-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Beta-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[BetaBinomial]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-1.0--1.0]
0.00s teardown tests/contrib/oed/test_finite_spaces_eig.py::test_nmc_eig_finite_space_model
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-1-1]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[0,0]-()]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:,0]-(2,)]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->bc-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,bc,cd->ac-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-ab,,bc->a-2]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[2-1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[3-2-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[2-1-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_pad[1-2-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[2-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[2-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_pad[2-1-2-()]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs6-dims6-3]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-1.0]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data2-expected_log_prob2]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-subsample-reparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[0.1-0.1]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.27-0.33]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_median-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-0.5-0.9-0.5]
0.00s teardown tests/distributions/test_mvt.py::test_mean_var[(4,)]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-11]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[GammaPoisson-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LogNormal-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Uniform-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-Size-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0--1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-1.0-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0--1.0--1.0]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-None]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape2]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[2-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[2-1-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[3-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[3-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[1-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_reshape[3-(6,)-(3, 2)]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_2[single-10]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_1[masked-10]
0.00s teardown tests/infer/test_enum.py::test_elbo_normal[parallel-loss_and_grads]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_sample-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_median-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_median-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.25]
0.00s teardown tests/distributions/test_mvn.py::test_shape[5-cov_shape0-loc_shape2]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape1-event_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-13]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Geometric-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Gamma-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Uniform-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[BetaBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[ExtendedBinomial]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0--1.0-1.0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-7-7]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-10]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-<lambda>]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0,0]-()]
0.00s teardown tests/poutine/test_counterfactual.py::test_counterfactual_query[False-True-False]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[3-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[2-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[1-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-3-(4,)]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_continuous_sampled[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s teardown tests/infer/test_enum.py::test_mixture_of_diag_normals[MixtureOfDiagNormalsSharedCovariance-scale1]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[0.1-0.5]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_median-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-0.5--0.5-0.5]
0.00s teardown tests/distributions/test_rejector.py::test_standard_gamma_elbo[5.0]
0.00s teardown tests/distributions/test_mvn.py::test_shape[5-cov_shape1-loc_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-15]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Geometric-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Poisson-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[BetaBinomial-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[FoldedNormal-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[LogNormal-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Binomial]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-3]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-3-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar_full_mass': 3, 'num_quant_bins': 2}-7-4]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[2-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[2-1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[1-1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[1-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-3-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceEnum_ELBO-full-reparam]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-subsample-nonreparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[1.0-0.5]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.0-0.0-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.0-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-0.5-0.5-0.5]
0.00s teardown tests/distributions/test_mvn.py::test_shape[3-cov_shape2-loc_shape1]
0.00s teardown tests/distributions/test_mvn.py::test_shape[3-cov_shape0-loc_shape0]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape0-event_shape2]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape0-event_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-9]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Geometric-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Bernoulli-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Normal-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Gamma-True-initial_shape0-proposed_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LKJCorrCholesky-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-tuple-sample_shape2]
0.00s teardown tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[3-3-NcpContinuous]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-1.0-0.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-0.0-1.0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 2}-7-4]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-6]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[3-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[2-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[1-3-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[1-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[1-2-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[1-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[2-3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[1-3-(3, 2)]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceTreeEvaluator-data2-expected_log_prob2]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceEinsumEvaluator-data0-expected_log_prob0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-subsample-reparam]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceMeanField_ELBO-full-reparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[parallel-differentiable_loss-10]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[sequential-loss-10]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_sample-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_median-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_mvt.py::test_mean_var[(3, 2)]
0.00s teardown tests/distributions/test_mvn.py::test_shape[5-cov_shape0-loc_shape0]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape0-event_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(3, 2)-Real()]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-5]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Poisson-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Cauchy-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Dirichlet-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Exponential-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-True-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-list-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape1-1-1]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-2-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'num_quant_bins': 4}-7-4]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_sparse_smoke[{'haar': True, 'num_quant_bins': 2}-7-4]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-1.0]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs9-dims9-3]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-()-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[3-3-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[3-1-()]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[2-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[3-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[3-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[2-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-2-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[marginal-a,abi->-i]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-3-True]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.55-0.43]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[sample-10]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_feasible-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.75]
0.00s teardown tests/distributions/test_mvn.py::test_shape[5-cov_shape1-loc_shape2]
0.00s teardown tests/distributions/test_mvn.py::test_shape[1-cov_shape1-loc_shape2]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape2-event_shape1]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape1-event_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-12]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Binomial-False-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveBeta-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Normal-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-Size-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-2-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-1.0-1.0]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape0-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_overdispersed_sir_smoke[{'haar_full_mass': 2}-7-3]
0.00s teardown tests/contrib/test_minipyro.py::test_local_param_ok[minipyro-py]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[:,0]-(2,)]
0.00s call     tests/distributions/test_pickle.py::test_pickle[ZeroInflatedDistribution]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[2-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[1-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[3-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-()-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[3-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[3-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[1-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-3-()]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_discrete_observed[TraceEinsumEvaluator-data1-expected_log_prob1]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.27-0.33]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-flat-1]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoNormal]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_feasible-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.25]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-0.5-0.9-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-0.5-0.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-0.5--0.5-1.5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_make_complete_graph[python-2-expected_grid0]
0.00s teardown tests/distributions/test_mvn.py::test_shape[3-cov_shape2-loc_shape0]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape2-event_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-9]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Cauchy-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-Size-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-1]
0.00s call     tests/distributions/test_independent.py::test_to_event[mvn]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape2]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[2-2-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[1-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[3-1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[1-(4,)-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-3-(3, 2)]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(2, 3, 1)--2-0.5]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-0.5]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-subsample-inf]
0.00s teardown tests/infer/test_svgd.py::test_mean_variance[IMQSteinKernel-univariate-latent_dist0]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[rsvi-JitTraceGraph_ELBO]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-full-reparam]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[Trace_ELBO-full-nonreparam]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.55-0.33]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_sample-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_sample-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.75]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-0.5-0.5-1.5]
0.00s teardown tests/distributions/test_mvn.py::test_shape[3-cov_shape0-loc_shape1]
0.00s teardown tests/distributions/test_mvn.py::test_shape[1-cov_shape2-loc_shape1]
0.00s teardown tests/distributions/test_mvn.py::test_shape[1-cov_shape1-loc_shape1]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape2-event_shape1]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape2-event_shape0]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape1-event_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-13]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape4-event_shape4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_new_dim[LogNormal-False-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[OneHotCategorical-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-tuple-sample_shape1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-2-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Geometric-batch_shape1-1-1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-7-3]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[0,:]-(3,)]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:,:]-(2, 3)]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:]-(2, 3)]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[1-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(3, 2)-(7,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[3-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[2-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-1-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[sample-ai,abij,acik->i-ijk]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-sequential-inf]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-None-inf]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[1.0-0.25]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-2-True]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-2-True]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[parallel-loss_and_grads-10]
0.00s teardown tests/distributions/test_mvn.py::test_shape[5-cov_shape2-loc_shape0]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape1-event_shape2]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-7]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateNormal-Size-sample_shape1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-4]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape1-sample_shape2]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwj->ji]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[2-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[1-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[3-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[3-3-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[3-1-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[2-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[1-1-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-3-(4,)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[sample-ai,abij->i-ij]
0.00s teardown tests/infer/test_valid_models.py::test_plate_ok[TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_ok[TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[rsvi-JitTraceEnum_ELBO]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-subsample-nonreparam]
0.00s teardown tests/infer/test_gradient.py::test_subsample_gradient_sequential[TraceGraph_ELBO-full-nonreparam]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-30-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_rsvi[sequential]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[1-parallel-sequential]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.5]
0.00s teardown tests/distributions/test_mvn.py::test_shape[5-cov_shape2-loc_shape1]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape2-event_shape1]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape0-event_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-4]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape0]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[ExtendedBetaBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[DirichletMultinomial]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-2]
0.00s teardown tests/contrib/forecast/test_util.py::test_reshape_batch[Bernoulli-batch_shape0-1-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'num_quant_bins': 2}-0-7]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[3-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[3-2-()]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[1-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[3-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[1-2-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-1-(4,)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[marginal-ai,ai->i-i]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[marginal-ab,bc->-]
0.00s teardown tests/infer/reparam/test_discrete_cosine.py::test_uniform[(6,)--1-2.0]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enum_log_prob_multiple_plate[TraceTreeEvaluator-data1-expected_log_prob1]
0.00s teardown tests/infer/test_enum.py::test_svi_step_smoke[sequential-single]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_mean-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_mean-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.0-0.9-1.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.0-0.5-1.5]
0.00s teardown tests/distributions/test_mvt.py::test_log_prob_normalization[2]
0.00s teardown tests/distributions/test_mvn.py::test_shape[3-cov_shape1-loc_shape0]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape0-event_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-3-3]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape3-event_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[MultivariateStudentT-True-initial_shape1-proposed_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[NaiveDirichlet-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_error[Normal-True-initial_shape2-proposed_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-True-list]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-1-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-3-3]
0.00s call     tests/distributions/test_pickle.py::test_pickle[LinearHMM]
0.00s call     tests/distributions/test_pickle.py::test_pickle[ImproperUniform]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[2-1-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[3-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[2-(4,)-()]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[2-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-2-()]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[reparam-JitTrace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_predictive[auto_class6]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_mean-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_sample-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_mean-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.0-0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[2.0-0.5]
0.00s teardown tests/distributions/test_mvn.py::test_shape[5-cov_shape1-loc_shape1]
0.00s teardown tests/distributions/test_mvn.py::test_shape[3-cov_shape1-loc_shape2]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape1-event_shape2]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape0-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(3, 2)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-5]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-tuple-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-2]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-0-7]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...,0]-()]
0.00s call     tests/distributions/test_pickle.py::test_pickle[GaussianHMM]
0.00s call     tests/distributions/test_pickle.py::test_pickle[DiscreteHMM]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-()-(7,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[2-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[2-2-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[1-1-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_size_error[naive_ubersum-ab,bc->-shapes0-]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[1.0-0.75]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_3[batch-10]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-50-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-40-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-20-False]
0.00s teardown tests/infer/test_autoguide.py::test_predictive[auto_class7]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class12]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.5]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.0-0.5-0.5]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_bernoulli_entropy_grad[0.5]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[FoldedNormal-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Cauchy-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[FoldedNormal-tuple-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0-0.0-0.0]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[3-1-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize[2-3-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[2-1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[1-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-2-(4,)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[marginal-ai,abij->i-ij]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[marginal-a,abi,bcij,bdik->-ijk]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7--0.5]
0.00s teardown tests/infer/reparam/test_haar.py::test_uniform[(2, 5)--1-True]
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_dbn[True-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[parallel-loss_and_grads-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[sequential-loss_and_grads-10]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[sequential-loss-1]
0.00s teardown tests/distributions/test_stable.py::test_mean[S-1.0--0.5-0.5]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape1-event_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-False-Size]
0.00s teardown tests/distributions/test_conjugate.py::test_gamma_poisson_log_prob[shape2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.05]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[:,:]-(2, 3)]
0.00s call     tests/distributions/test_independent.py::test_to_event[normal]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(3, 2)-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[1-3-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[3-(4,)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[2-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[1-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-()-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize[2-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-1-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-None-inf]
0.00s teardown tests/infer/test_jit.py::test_beta_bernoulli[TraceEnum_ELBO-False]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[rsvi-JitTrace_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[parallel-loss-10]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_onehot_entropy_grad[0.3]
0.00s teardown tests/distributions/test_mvn.py::test_shape[3-cov_shape0-loc_shape2]
0.00s teardown tests/distributions/test_mvn.py::test_shape[1-cov_shape2-loc_shape0]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape1-event_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LKJCorrCholesky-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[GammaPoisson]
0.00s teardown tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[8]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar_full_mass': 2}-0-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-0-7]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape1-sample_shape0]
0.00s call     tests/distributions/test_pickle.py::test_pickle[MultivariateStudentT]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape3]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[2-3-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[2-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[1-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_reshape[2-(6,)-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-2-(4,)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[sample-ai,ai->i-i]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7-0.5]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-3-True]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[parallel-0.27-0.33]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[1-sequential-sequential]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-tuple-sample_shape2]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-(2,)]
0.00s teardown tests/ops/test_gaussian.py::test_logsumexp[3-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[3-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[2-1-()]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[3-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[3-()-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-2-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-2-()]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-,bi,abi->b-i]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-full-inf]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[reparam-JitTraceGraph_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[1-None-sequential]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.5]
0.00s teardown tests/distributions/test_rejector.py::test_standard_gamma_elbo[2.0]
0.00s teardown tests/distributions/test_mvn.py::test_shape[5-cov_shape0-loc_shape1]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape2-batch_shape0-event_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-True-Size]
0.00s teardown tests/distributions/test_conjugate.py::test_variance[dist3]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape2-sample_shape0]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[2-2-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[1-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[3-(3, 2)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[3-(4,)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-3-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[2-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-1-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[1-1-()]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[sample-a,abi->-i]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-True-2200]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-False-None]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[True-None-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[3.0-0.1]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-3-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-3-True]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_sample-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-0.75]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-3-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[ExtendedBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Binomial]
0.00s teardown tests/distributions/test_conjugate.py::test_variance[dist2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-1]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[3-(3, 2)-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[3-()-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[2-(4,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[1-()-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-mask3]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-full-inf]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_4[False-True-10]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-2-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_model[sequential-2-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_bern[parallel-loss-1]
0.00s teardown tests/distributions/test_independent.py::test_expand[sample_shape1-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-0-expand_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-False-list]
0.00s teardown tests/contrib/epidemiology/test_quant.py::test_quantization_scheme[12]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1290]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[2-1-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[1-1-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[1-(3, 2)-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_marginalize_shape[3-1-(4,)]
0.00s teardown tests/infer/reparam/test_unit_jacobian.py::test_normal[(2, 1, 3)]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_1[single-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-20-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-2-True]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[OneHotCategorical-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Bernoulli-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Dirichlet-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-Size-sample_shape0]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'num_quant_bins': 2}-7-3]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-7-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-4]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-bi,abi->b-i]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-None-inf]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_single[Bernoulli-0.2]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-2-False]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.55-0.43]
0.00s teardown tests/infer/test_enum.py::test_svi_step_smoke[parallel-single]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_feasible-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_sample-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_mean-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_feasible-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-2-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-list-sample_shape2]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{'haar': True}-7-7]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges578]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[3-1.0]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[2-2-()]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-False-1800]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-sequential-1]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-full-inf]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[None-0.27-0.43]
0.00s teardown tests/infer/test_enum.py::test_svi_step_smoke[sequential-batch]
0.00s teardown tests/infer/test_enum.py::test_svi_step_smoke[None-batch]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-flat-3]
0.00s teardown tests/infer/test_autoguide.py::test_predictive[AutoDiagonalNormal]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_ig.py::test_sample[3.0-4.0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateNormal-True-list]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-0-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-6]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape5]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[2-3-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_shape[2-()-()]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-True-None]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-subsample-inf]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-subsample-inf]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[nonreparam-JitTraceGraph_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-False-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[2-sequential-sequential]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[2-None-None]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_onehot_entropy_grad[0.5]
0.00s teardown tests/distributions/test_mvt.py::test_log_prob_normalization[1]
0.00s teardown tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_independent.py::test_independent[normal-1-batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-Size-sample_shape2]
0.00s teardown tests/test_generic.py::test_mcmc_interface[pyro-neals_funnel]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-,ai->a-i]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-True-1800]
0.00s teardown tests/infer/test_valid_models.py::test_plate_iplate_ok[TraceEnum_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-10-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_model[parallel-10-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_normal[parallel-differentiable_loss]
0.00s teardown tests/infer/test_enum.py::test_svi_step_smoke[parallel-batch]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-dense-3]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-flat-3]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_rejector.py::test_standard_gamma_elbo[1.0]
0.00s teardown tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape0-sample_shape2]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...,:]-(2,)]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixyj,iyzj->ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum_size_error[ubersum-ab,bc->-shapes0-]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-False-10]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[parallel-3-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[1-None-None]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_mean-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-list-sample_shape1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-6]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.00s teardown tests/infer/test_valid_models.py::test_dim_allocation_ok[TraceEnum_ELBO-False]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[False-None-1]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-full-inf]
0.00s teardown tests/infer/test_svgd.py::test_conjugate[RBFSteinKernel-univariate]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[sample-1]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_feasible-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_median-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_feasible-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_median-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_mask.py::test_mask[0-1-0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-17]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Categorical-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LowRankMultivariateNormal-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LowRankMultivariateNormal-Size-sample_shape1]
0.00s teardown tests/distributions/test_conjugate.py::test_mean[dist2]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar_full_mass': 2}-0-7]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-None-batch_shape1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_size_error[ubersum-ab,bc->-shapes1-b]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-None-1]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-subsample-inf]
0.00s teardown tests/infer/test_discrete.py::test_distribution_2[sample]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoIAFNormal-init_to_sample-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_median-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_feasible-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoNormal-Trace_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[2.0-1.0]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape1-sample_shape0]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(4,)-(6, 5)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1--1.0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-subsample-inf]
0.00s teardown tests/infer/test_enum.py::test_non_mean_field_bern_bern_elbo_gradient[sequential-0.27-0.43]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_median-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_median-Trace_ELBO]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-list-sample_shape0]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-<lambda>]
0.00s call     tests/distributions/test_pickle.py::test_pickle[NegativeBinomial]
0.00s teardown tests/infer/reparam/test_haar.py::test_uniform[(4, 2)--2-True]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[nonreparam-JitTrace_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[1-sequential-None]
0.00s teardown tests/infer/test_discrete.py::test_distribution_2[map]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_median-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_mean-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_median-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_median-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_mean-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_mean-Trace_ELBO]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-7-3]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges672]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,a,ab,b,b,b,b->a-1]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,ab->ab-2]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-100.0-batch_shape0]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-None-batch_shape0]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_moments[(4,)]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[reparam-JitTraceEnum_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[3.0-0.9]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_in_guide[sequential-2-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[1-parallel-parallel]
0.00s teardown tests/infer/test_enum.py::test_differentiable_loss[None-batch]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_mean-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_sample-Trace_ELBO]
0.00s teardown tests/distributions/test_ig.py::test_log_prob[1.7-3.0-4.0]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Poisson]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{'haar': True}-7-7]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag[mat_size1-1-batch_shape1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-aij->aj-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[sample-elbo-10]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_sample-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_sample-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_feasible-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_sample-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_sample-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_mean-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_feasible-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_median-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_mask.py::test_mask[0-0-0]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[5.0-clip_value-Adam]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-2.0-batch_shape0]
0.00s teardown tests/ops/test_contract.py::test_ubersum[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-subsample-0]
0.00s teardown tests/infer/test_jit.py::test_dirichlet_bernoulli[JitTraceEnum_ELBO-True]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_sample-Trace_ELBO]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape6]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Multinomial-list-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape2-sample_shape2]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plate_dependency_warning[True-None-1]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-full-0]
0.00s teardown tests/infer/test_inference.py::test_exponential_gamma[nonreparam-JitTraceEnum_ELBO]
0.00s teardown tests/infer/test_enum.py::test_differentiable_loss[parallel-batch]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape1-event_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-5]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-False-sample_shapes1]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_seir_smoke[{}-7-7]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-7-3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-1.0-0.5]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges903]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-,ab->ab-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag[mat_size2-3-batch_shape1]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplates_plate_dependency_warning[False-sequential-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[1-sequential-parallel]
0.00s teardown tests/infer/test_enum.py::test_svi_step_smoke[None-single]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_feasible-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoNormal-init_to_feasible-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_sample-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_mean-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_mean-Trace_ELBO]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-10]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-True-list]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_overdispersed_seir_smoke[{'haar_full_mass': 2}-7-3]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99-0.5]
0.00s call     tests/distributions/test_pickle.py::test_pickle[ZeroInflatedNegativeBinomial]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,ab->ab-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag[mat_size2-3-batch_shape0]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[2-0.1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-100.0-batch_shape0]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[1-1-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_ubersum[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_mcmc_interface[1-False-2200]
0.00s teardown tests/infer/test_jit.py::test_svi_enum[2-sequential-sequential]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_sample-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(4,)-GreaterThan(lower_bound=0.0)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-15]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-True-tuple]
0.00s teardown tests/contrib/epidemiology/test_models.py::test_superspreading_sir_smoke[{}-7-7]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-2]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Rejector]
0.00s call     tests/distributions/test_pickle.py::test_pickle[DirichletMultinomial]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-,ab->ab-1]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwxj,ixyj,iyzj,izwj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-w,wx,xy,yz->]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[3-3-(4,)]
0.00s teardown tests/ops/test_contract.py::test_contract_to_tensor[example5]
0.00s teardown tests/infer/test_valid_models.py::test_dim_allocation_ok[TraceEnum_ELBO-True]
0.00s teardown tests/infer/test_jit.py::test_infer_discrete[map-10]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_single[OneHotCategorical-prior5]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[sample-2]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[map-100]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-8]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LogNormal-tuple-sample_shape1]
0.00s teardown tests/contrib/oed/test_linear_models_eig.py::test_lfire_linear_model
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-20]
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_subsample_smoke[init_to_median]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[3.0-clip_norm-Adam]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-3]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-9]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[OneHotCategorical-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-tuple-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-3-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-4]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-5]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[1.0-clip_norm-Adam]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-4]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-100.0-batch_shape1]
0.00s teardown tests/ops/test_newton.py::test_newton_step[1-2.0-batch_shape1]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[2-2-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-ai->a-i]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-full-0]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_1[batch-10]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLaplaceApproximation-init_to_mean-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_mean-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDiagonalNormal-init_to_feasible-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoDelta-init_to_median-Trace_ELBO]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[4-(2, 3)]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-17]
0.00s teardown tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-2-expand_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Poisson-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Categorical-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-list-sample_shape1]
0.00s teardown tests/contrib/oed/test_glmm.py::test_log_prob_matches[model-sigmoid_example-design6]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-1]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[3.0-clip_value-Adam]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-,ab->ab-2]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[1-2-()]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_predictive[True-None]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[parallel-inf]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_2[batch-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_1[single-10]
0.00s teardown tests/infer/test_enum.py::test_differentiable_loss[sequential-batch]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape2-event_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-2-1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-0-expand_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[DirichletMultinomial-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Dirichlet-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-Size-sample_shape0]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-4-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.2]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges625]
0.00s call     tests/distributions/test_pickle.py::test_pickle[ExtendedBinomial]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-do-normal_model]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-3.0]
0.00s teardown tests/infer/test_svgd.py::test_conjugate[IMQSteinKernel-univariate]
0.00s teardown tests/infer/test_discrete.py::test_plate_smoke[2-map]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoMultivariateNormal-init_to_sample-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_mask.py::test_mask[0-2-0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-4]
0.00s teardown tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-2-expand_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Categorical-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-Size-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-1-1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-20]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[10.0-1.0-2.0]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,ab->ab-2]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-,ab->ab-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-10]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-subsample-0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_ok[TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_discrete.py::test_hmm_smoke[map-10]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide[AutoGuideList-init_to_mean]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape1-event_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-8]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Beta-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-tuple-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-4-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-3-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-2-1]
0.00s teardown tests/contrib/oed/test_linear_models_eig.py::test_dv_linear_model
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-50]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-10]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[5.0-5-0.5]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-i,wi->i]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-3]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[3-2-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[1-1-()]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs11-dims11-2]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-TraceTMC_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceEnum_ELBO-full-0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-subsample-0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceEnum_ELBO-subsample-0]
0.00s teardown tests/distributions/test_mask.py::test_mask[0-1-1]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(3, 2)-GreaterThan(lower_bound=0.0)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(5,)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-3-1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-1-expand_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_reshaped_distribution[True-0-expand_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[GammaPoisson-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateNormal-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Beta-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[MultivariateNormal]
0.00s teardown tests/contrib/oed/test_finite_spaces_eig.py::test_lfire_finite_space_model
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-1.0-1.0]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3--1.0]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[1.0-clip_value-Adam]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[finite-a,ab->ab-1]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj->ij]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-10]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_reshape[2-(5, 6)-(5, 3, 2)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0-0.5]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_enum.py::test_mixture_of_diag_normals[MixtureOfDiagNormals-scale0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-13]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Uniform-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Geometric-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Poisson-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Cauchy-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-list-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-1-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-3-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0--1.0--1.0]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-(2,)]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-20]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-50]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-10]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[5.0-5-2.0]
0.00s teardown tests/ops/einsum/test_torch_log.py::test_einsum[infinite-a,a,ab,b,b,b,b->a-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-10]
0.00s teardown tests/ops/test_newton.py::test_newton_step[3-None-batch_shape1]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[3-1-(3, 2)]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-3.0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-0.5]
0.00s teardown tests/infer/test_jit.py::test_beta_bernoulli[JitTraceEnum_ELBO-True]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[3.0-0.5]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_1[batch-1]
0.00s teardown tests/distributions/test_ig.py::test_log_prob[0.5-2.5-4.0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-16]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-0-expand_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[BetaBinomial-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Dirichlet-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Categorical-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Multinomial-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Normal-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-tuple-sample_shape1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-2-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-1]
0.00s teardown tests/contrib/oed/test_linear_models_eig.py::test_marginal_likelihood_linear_model
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-(2,)]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-20]
0.00s call     tests/distributions/test_pickle.py::test_pickle[Empirical]
0.00s call     tests/distributions/test_pickle.py::test_pickle[CoalescentTimesWithRate]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[5.0-clip_norm-Adam]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[3-size0]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-10]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[3-3-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[3-2-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-subsample-0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-full-0]
0.00s teardown tests/infer/test_autoguide.py::test_predictive[AutoDelta]
0.00s teardown tests/infer/test_autoguide.py::test_shapes[AutoLowRankMultivariateNormal-init_to_mean-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape1-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_mask[2-1-1]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(3, 2)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(3, 2)-GreaterThan(lower_bound=0.0)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-6]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Uniform-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Binomial-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Binomial-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Bernoulli-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Multinomial-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Cauchy-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedDirichlet-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[OneHotCategorical]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-2]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-()]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-None]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-10]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[5.0-5-1.0]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-5]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[2-2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(4,)-(6, 5)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99-1.0]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_4[False-True-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_3[single-10]
0.00s teardown tests/infer/test_enum.py::test_elbo_normal[None-differentiable_loss]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape0-event_shape2]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(4,)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(4,)-Real()]
0.00s teardown tests/distributions/test_ig.py::test_sample[3.0-3.3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-2-1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape1-batch_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape0]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Categorical-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBinomial-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Binomial-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Dirichlet-tuple-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-2-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[-1.0-1.0-0.0]
0.00s teardown tests/contrib/oed/test_glmm.py::test_log_prob_matches[lm_2p_10_10_1-model21-design1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-20]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-1.0-2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[10.0-1.0-1.0]
0.00s call     tests/distributions/test_pickle.py::test_pickle[GaussianMRF]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-wx,xy,yz,zw->]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-5]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape1]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[3-2-()]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[2-3-(3, 2)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4-1.0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceTMC_ELBO-subsample]
0.00s teardown tests/infer/test_enum.py::test_differentiable_loss[sequential-single]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape2-event_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape1-event_shape2]
0.00s teardown tests/distributions/test_mask.py::test_mask[2-2-1]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(4,)-GreaterThan(lower_bound=0.0)]
0.00s teardown tests/distributions/test_ig.py::test_log_prob[1.7-3.0-2.5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-7]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveBeta-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Geometric-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Poisson-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Poisson-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Poisson-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Dirichlet-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Dirichlet-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Beta-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Beta-list-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-3-3]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-None]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-50]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-50]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-10]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-1.0-2.0]
0.00s call     tests/distributions/test_distributions.py::test_distribution_validate_args[False-Gamma-args1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-10]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[3-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_reshape[1-(5, 6)-(5, 3, 2)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0--1.0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-Trace_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_jit.py::test_svi_enum[1-parallel-sequential]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-dense-2]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-dense-2]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.0]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape1-event_shape2]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape2-event_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_ig.py::test_log_prob[1.7-2.5-4.0]
0.00s teardown tests/distributions/test_ig.py::test_log_prob[0.5-3.0-2.5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[()-2-1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape0]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape3]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Geometric-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Cauchy-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Geometric-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Geometric-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Poisson-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Poisson-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Cauchy-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Dirichlet-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Normal-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Exponential-list-sample_shape2]
0.00s teardown tests/distributions/test_conjugate.py::test_gamma_poisson_log_prob[shape1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-0.0-1.0]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-0.0]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-(2,)]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-50]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-20]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-20]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-10]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-5]
0.00s call     tests/distributions/test_independent.py::test_to_event[exponential]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,xi->i]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-5-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag[mat_size1-1-batch_shape0]
0.00s teardown tests/ops/test_newton.py::test_newton_step[2-2.0-batch_shape0]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[3-1-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_reshape[3-(5, 6)-(5, 3, 2)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-,bi,abi->b-i]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-0.5]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3-0.5]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-True]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-subsample-inf]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_2[masked-1]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[2-parallel-parallel]
0.00s teardown tests/infer/test_enum.py::test_differentiable_loss[None-single]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-dense-2]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-flat-2]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape1-event_shape0]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-1-(2, 3, 5)]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-0-(4, 5)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(4,)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-3-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-2]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_logmatmulexp[(2, 4)-3-1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape0]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape0-batch_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[NaiveDirichlet-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LowRankMultivariateNormal-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Geometric-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveBeta-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Beta-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Categorical-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Binomial-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Cauchy-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Beta-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-Size-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-3-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-1-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-2-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[-1.0--1.0]
0.00s teardown tests/contrib/oed/test_linear_models_eig.py::test_posterior_linear_model
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-None]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-5.0-20]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-1.0-10]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-50]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-0.2-10]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-10.0-2.0]
0.00s teardown tests/test_generic.py::test_mcmc_interface[pyro-eight_schools]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,x->]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-10]
0.00s teardown tests/ops/test_gaussian.py::test_logsumexp[2-()]
0.00s teardown tests/ops/test_gaussian.py::test_reshape[3-(5, 6)-(5, 3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[2-(3, 2)-(6, 5)]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-0.5]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[Trace_ELBO-full-inf]
0.00s teardown tests/infer/test_jit.py::test_grad
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_4[True-True-10]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoNormal]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape1-event_shape2]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(3, 2)-Real()]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(4,)-Real()]
0.00s teardown tests/distributions/test_ig.py::test_log_prob[0.5-2.5-2.5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-9]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-4]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-12]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape1-batch_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Binomial-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[GammaPoisson-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[BetaBinomial-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Geometric-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Binomial-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Bernoulli-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Cauchy-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Gamma-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Gamma-list-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-2]
0.00s teardown tests/contrib/oed/test_glmm.py::test_log_prob_matches[model15-lr_10_10-design5]
0.00s teardown tests/contrib/oed/test_glmm.py::test_log_prob_matches[model14-normal_inv_gamma_2_2_10_10-design4]
0.00s teardown tests/contrib/oed/test_glmm.py::test_log_prob_matches[model13-nz_lm_2p_10_10_1-design3]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-()]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-(2,)]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-50]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-0.2-2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[10.0-1.0-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-10.0-1.0]
0.00s call     tests/test_primitives.py::test_param_ok
0.00s teardown tests/poutine/test_counterfactual.py::test_counterfactual_query[True-True-True]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-2-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-2-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-6-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-4]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[3-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-aij->ai-ij]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-1.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3-0.0]
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_dbn[True-True]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceTMC_ELBO-full-0]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-dense-3]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-nested_auto_guide_callable]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape2-event_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape0-event_shape2]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(4,)-GreaterThan(lower_bound=0.0)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(2, 4)-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-12]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-2-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-3-7]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-17]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-12]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape2-batch_shape3]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Bernoulli-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedDirichlet-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedBeta-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[RejectionExponential-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Categorical-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Binomial-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Cauchy-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Cauchy-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Dirichlet-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Beta-list-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[4-4-3]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[1-4-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[bp-4-1]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-(2,)]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-None]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-()]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-()]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-2-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-5-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-0.2-10]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_negative_binomial[1.0-10.0-0.5]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape1-sample_shape0]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,wi->i]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-6-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-4-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-10-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-5-2]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[3-3-()]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[2-3-(4,)]
0.00s teardown tests/ops/test_gaussian.py::test_reshape[1-(5, 6)-(5, 3, 2)]
0.00s teardown tests/ops/test_contract.py::test_ubersum[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8--1.0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[False-Trace_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceTMC_ELBO-subsample-0]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[3.0-0.75]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[2-parallel-sequential]
0.00s teardown tests/infer/test_enum.py::test_elbo_iplate[2-sequential-parallel]
0.00s teardown tests/infer/test_enum.py::test_elbo_normal[None-loss]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-dense-1]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-flat-2]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape0-event_shape2]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape0-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_mask[2-0-0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-2-5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(2, 4)-1-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-3-2]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype1-event_shape2-batch_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Beta-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBinomial-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Binomial-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[HalfCauchy-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedBeta-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[RejectionExponential-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Exponential-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Geometric-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Geometric-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Poisson-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Poisson-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[DirichletMultinomial-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Bernoulli-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Multinomial-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Multinomial-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveDirichlet-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[LKJCorrCholesky]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[3-4-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[0.0-0.0-0.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-2-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-2-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-50-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.1]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape2-sample_shape1]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape2-sample_shape0]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape2-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape2-sample_shape1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-10-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-3-6]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag[mat_size0-2-batch_shape0]
0.00s teardown tests/ops/test_gaussian.py::test_reshape[3-(6,)-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_reshape[2-(5, 6)-(5, 3, 2)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[marginal-ai,abij,acik->i-ijk]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[sample-a->-]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-bi,abi->b-i]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7-1.0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-full-inf]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape2-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape3-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape2-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape2-event_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-11]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[HalfCauchy-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[FoldedNormal-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Poisson-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ExtendedBetaBinomial-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Gamma-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Geometric-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Geometric-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[GammaPoisson-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Categorical-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Binomial-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Binomial-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Bernoulli-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Multinomial-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[VonMises-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Cauchy-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Dirichlet-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Normal-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Beta-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-Size-sample_shape1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-1]
0.00s teardown tests/contrib/oed/test_glmm.py::test_log_prob_matches[model12-lm_2p_10_10_1_w12-design2]
0.00s teardown tests/contrib/oed/test_glmm.py::test_log_prob_matches[model10-lm_2p_10_10_1-design0]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-(2,)-0.0]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-None]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-20-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-10-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-10-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-1-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-20-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-20]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-50]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[0.2-5.0-2]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs3-dims3-2]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape1-sample_shape1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-3-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-4]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-2.0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1--0.5]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_reparam_stable[NUTS-kwargs1]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-flat-2]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-flat-1]
0.00s teardown tests/distributions/test_von_mises.py::test_von_mises_3d[2.0]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape2-batch_shape0-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape3-event_shape1]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(3, 2)-GreaterThan(lower_bound=0.0)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[()-()-GreaterThan(lower_bound=0.0)]
0.00s teardown tests/distributions/test_ig.py::test_log_prob[1.7-2.5-2.5]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-7]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype2-event_shape0-batch_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ExtendedBetaBinomial-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Multinomial-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Exponential-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Cauchy-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedGamma-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Binomial-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Multinomial-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[MultivariateStudentT-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[VonMises-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[VonMises-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[VonMises-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[HalfCauchy-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Dirichlet-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Normal-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedBeta-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[NaiveBeta-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Beta-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Exponential-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Uniform-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Uniform-tuple-sample_shape1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_vs_persistent[enum-1-2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_2_1[1.0--1.0-0.0]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-0.0]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-0.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-1-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-1-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-50-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-10-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.2]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[Trace_ELBO-AutoDiagonalNormal]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...]-(2, 3, 4)]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-replay-bernoulli_normal_model]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-5]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[marginal-a,a->-]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[sample-a,abi,bcij,bdik->-ijk]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::RaoBlackwellizationTests::test_plate_in_elbo_with_superfluous_rvs
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-2.0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-0.5]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1-1.0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-full-0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-full-inf]
0.00s teardown tests/infer/test_enum.py::test_elbo_normal[parallel-loss]
0.00s teardown tests/infer/test_discrete.py::test_distribution_masked[sample]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape1-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_mask[0-2-1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-11]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-6]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[(5,)-1-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-7]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype0-batch_shape2-event_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Bernoulli-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Bernoulli-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Multinomial-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[VonMises-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[VonMises-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[VonMises-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Normal-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Exponential-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Exponential-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Uniform-Size-sample_shape1]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-()]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-None-0.0]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-None]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-50-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-5-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-50-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-2-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[5.0-1.0-2]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-ij,iwj->ij]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-5-10]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[same-3-5]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-1.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.3--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01--1.0]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_logistic_regression[False-JIT=False]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_logistic_regression[True-JIT=False]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-False]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_ok[TraceGraph_ELBO-subsample-0]
0.00s teardown tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceTMC_ELBO-True]
0.00s teardown tests/infer/test_jit.py::test_svi_enum[2-sequential-parallel]
0.00s teardown tests/infer/test_discrete.py::test_distribution_masked[sample-elbo]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class9]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoNormal-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape1-batch_shape0-event_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape0-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape0-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_mask[1-2-2]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-1-(4, 5)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-(4,)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-2]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Normal-False-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[ShapeAugmentedGamma-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Normal-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[BetaBinomial-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Bernoulli-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Multinomial-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Normal-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Normal-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Beta-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Gamma-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Exponential-tuple-sample_shape0]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-1.0-(2,)]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-None]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-20-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-1-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-20-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-1-0.5]
0.00s teardown tests/poutine/test_counterfactual.py::test_counterfactual_query[True-False-False]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[3.0-clip_norm-SGD]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwj->ij]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[valid-4-4]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-10-5]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_logsumexp[2-()]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-1.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8-0.0]
0.00s teardown tests/infer/test_valid_models.py::test_vectorized_num_particles[Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-full-0]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-flat-3]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class9]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape2-event_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_expand[sample_shape0-batch_shape1-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape0-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_mask[1-1-0]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-0-(2, 3, 5)]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-1-(2, 3, 5)]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-1-(4, 5)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(3, 2)-Real()]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-14]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[MultivariateStudentT-True-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[Gamma-True-sample_shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBinomial-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ExtendedBetaBinomial-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Uniform-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Uniform-tuple-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_bp_vs_exact[2-1-1]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(3, 2)-()-(2,)]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-0.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.8-10-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.2-20-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial[1.0-5.0-2]
0.00s teardown tests/contrib/autoguide/test_mean_field_entropy.py::test_guide_entropy[mean_field_guide-args1-expected_entropy1]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[1.0-clip_norm-SGD]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[10-event_shape1]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[3-10.0]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[3-0.1]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[sample-a,a->-]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[map-a,abi->-i]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.0--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4--0.5]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-mask4]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[EnergyDistance_noprior-mask2]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceTMC_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_ok[TraceTMC_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_ok[TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_jit.py::test_svi_enum[2-parallel-sequential]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_2[batch-10]
0.00s teardown tests/infer/test_enum.py::test_differentiable_loss[parallel-single]
0.00s teardown tests/distributions/test_tensor_type.py::test_double_type
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-0-1]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape3-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape2-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape0-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape1-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape0-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_mask[1-2-1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[ShapeAugmentedDirichlet-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[VonMises-tuple-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Gamma-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Gamma-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Uniform-list-sample_shape1]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-()-()]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-()]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-50-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.9-5-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-10-0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-5-0.1]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[1-2-(4,)]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-2.0-1.0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-3.0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-0.5]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-1.0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-3.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.7--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.99--0.5]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=True-HMC]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-subsample-0]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-dense-1]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class12]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape1-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape1-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_mask[2-2-2]
0.00s teardown tests/distributions/test_mask.py::test_mask[2-1-0]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-1-(5,)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-()-GreaterThan(lower_bound=0.0)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-16]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[VonMises-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Poisson-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Bernoulli-list-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Stable-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Normal-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[ShapeAugmentedGamma-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Gamma-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Gamma-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Gamma-Size-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Exponential-tuple-sample_shape2]
0.00s teardown tests/distributions/test_delta.py::test_shapes[False-3-3]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-2]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-1.0-0.0]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-do-trivial_model]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[5.0-clip_norm-SGD]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[1.0-clip_value-SGD]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj->ji]
0.00s teardown tests/ops/test_ssm_gp.py::test_matern_kernel[1.5-1]
0.00s teardown tests/ops/test_gaussian.py::test_marginalize_shape[1-1-()]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_gamma_and_mvn_to_gamma_gaussian[1-(4,)-(6, 5)]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-3.0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-2.0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-2.0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-2.0-1.5]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-2.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-1.01-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.4--1.0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_ok[TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_2[single-1]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_model-flat-1]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_guide_module_callable]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape2-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape0-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_mask[2-2-0]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-0-(5,)]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-0-(2, 3, 5)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-18]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-3]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-10]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-15]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Bernoulli-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-list-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[0.0--1.0]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-(2,)-()]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_clamp[(2,)-None-0.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.5-5-0.5]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs0-dims0-1]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape1-sample_shape1]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-2]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_reshape[2-(6,)-(3, 2)]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-0.1-1.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.1-0.5]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-flat-2]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-dense-2]
0.00s teardown tests/infer/test_enum.py::test_gmm_iter_discrete_traces[gmm_guide-dense-1]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_guide_list_x]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_guide_list_x]
0.00s teardown tests/distributions/test_mvt.py::test_log_prob[2-()]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape3-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape1-event_shape0]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform1-0-(4, 5)]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-1-(5,)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-8]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-16]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[VonMises-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-Size-sample_shape1]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...]-(2, 3)]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...]-(2,)]
0.00s call     tests/distributions/test_pickle.py::test_pickle[CoalescentTimes]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwxj,ixyj,iyzj,izwj->ji]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[10-event_shape2]
0.00s teardown tests/ops/test_ssm_gp.py::test_matern_kernel[2.5-3]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[1-3-(4,)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[sample-a,abi,bcij->-ij]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[map-ai,abij->i-ij]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-3.0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-0.1-1.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7--1.0]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-TraceTMC_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_ok[TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoIAFNormal]
0.00s teardown tests/distributions/test_rejector.py::test_rejector[0.5-2.0]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape2-dist_shape3-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape2-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_mask[1-1-1]
0.00s teardown tests/distributions/test_kl.py::test_kl_transformed_transformed[transform0-0-(5,)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-()-GreaterThan(lower_bound=0.0)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-2-19]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-9]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape3]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype2-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[VonMises-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Stable-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Exponential-Size-sample_shape2]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1208]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape1-sample_shape0]
0.00s call     tests/distributions/test_pickle.py::test_pickle[LKJCorrCholesky]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-condition-bernoulli_normal_model]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[5.0-clip_value-SGD]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[1.234-1.0-1.0]
0.00s teardown tests/infer/reparam/test_studentt.py::test_distribution[0.0-1.0-2.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4-1.0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[Trace_ELBO-subsample-inf]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-dense-1]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-dense-3]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_model-flat-3]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape2-event_shape0]
0.00s teardown tests/distributions/test_mask.py::test_mask[0-2-2]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal_mvn[3-(2, 3)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-(4,)-Real()]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-1-11]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_log_prob[dtype1-batch_shape0-event_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Exponential-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Uniform-tuple-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Uniform-Size-sample_shape0]
0.00s teardown tests/distributions/test_delta.py::test_shapes[True-3-2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_beta_binomial[0.1-2-0.1]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape1-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape1-sample_shape0]
0.00s teardown tests/ops/test_contract.py::test_contract_tensor_tree[example7]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1-0.5]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-scale2]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_enum.py::test_gmm_batch_iter_discrete_traces[gmm_batch_guide-flat-1]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_guide_list_x]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-9]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[RejectionExponential-tuple-sample_shape1]
0.00s teardown tests/contrib/tracking/test_distributions.py::test_EKFDistribution_smoke[2-3-NcvContinuous]
0.00s teardown tests/contrib/autoguide/test_mean_field_entropy.py::test_guide_entropy[mean_field_guide-args0-expected_entropy0]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape1-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape2-sample_shape2]
0.00s teardown tests/poutine/test_counterfactual.py::test_counterfactual_query[True-True-False]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-wi->i]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[marginal-a->-]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[map-a,abi,bcij,bdik->-ijk]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-0.8-0.5]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_logistic_regression[True-JIT=True]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[1-False]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-TraceTMC_ELBO]
0.00s teardown tests/infer/test_jit.py::test_svi_enum[1-sequential-sequential]
0.00s teardown tests/infer/test_discrete.py::test_distribution_1[map]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class9]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape3-event_shape1]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape1-dist_shape1-event_shape0]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal_mvn[3-(4,)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-19]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape0-batch_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Stable-Size-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Stable-Size-sample_shape0]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape1-sample_shape0]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-do-bernoulli_normal_model]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-2-3]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(3, 2)-(6, 5)]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=False-HMC]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-TraceTMC_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-TraceEnum_ELBO]
0.00s teardown tests/infer/test_discrete.py::test_plate_smoke[2-sample-elbo]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class11]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoDiagonalNormal]
0.00s teardown tests/distributions/test_von_mises.py::test_von_mises_3d[0.5]
0.00s teardown tests/distributions/test_rejector.py::test_rejector[0.5-1.0]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape3-dist_shape2-event_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[(5,)-3-15]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape2-batch_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_mean_var[dtype0-event_shape1-batch_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Multinomial-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Categorical-tuple-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Categorical-Size-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Gamma-tuple-sample_shape1]
0.00s teardown tests/distributions/test_delta.py::test_shapes[True-2-2]
0.00s teardown tests/distributions/test_conjugate.py::test_beta_binomial_log_prob[shape0-1]
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_subsample_smoke[init_to_mean]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_diagonal_gaussians[TraceMeanField_ELBO-AutoLowRankMultivariateNormal]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape2-sample_shape1]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwj->ij]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[1-2-(3, 2)]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[1-1-(4,)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8-1.0]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gaussian_hmm[3]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gamma_normal[True-JIT=False]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=True-NUTS]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-1]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_guide_callable]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoIAFNormal]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoNormal]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape3-event_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-13]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-12]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-False-Size]
0.00s teardown tests/distributions/test_delta.py::test_shapes[True-3-3]
0.00s teardown tests/distributions/test_conjugate_update.py::test_beta_binomial[(3, 2)-()]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoIAFNormal]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges484]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2))[:]-(2,)]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-replay-normal_normal_model]
0.00s teardown tests/poutine/test_poutines.py::test_enumerate_poutine[1--1]
0.00s teardown tests/optim/test_optim.py::test_clip_norm[3.0-clip_value-SGD]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8-0.5]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_iplate_swap_ok[TraceGraph_ELBO-subsample-inf]
0.00s teardown tests/infer/test_discrete.py::test_distribution_2[sample-elbo]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class11]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_guide_callable]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_reshaped_distribution[False-1-expand_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Stable-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Exponential-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[LowRankMultivariateNormal]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[1.0-0.0]
0.00s teardown tests/test_generic.py::test_rng_state[pyro-beta_binomial]
0.00s call     tests/distributions/test_pickle.py::test_pickle[FoldedDistribution]
0.00s teardown tests/ops/test_stats.py::test_fit_generalized_pareto[1.3-0.5]
0.00s teardown tests/ops/test_contract.py::test_adjoint_marginal[a,abi,bcij,bdik->-ijk]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8-0.0]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_guide_module_callable]
0.00s teardown tests/distributions/test_spanning_tree.py::test_make_complete_graph[python-4-expected_grid2]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-2-2]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-0-0]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape3-event_shape1]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal_mvn[1-(2, 3)]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-10]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-True-tuple]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixyj,iyzj->ji]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99-0.5]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gamma_normal[True-JIT=True]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_unnormalized_normal[JIT=False-NUTS]
0.00s teardown tests/infer/test_jit.py::test_svi_enum[1-parallel-parallel]
0.00s teardown tests/infer/test_jit.py::test_svi_enum[1-sequential-parallel]
0.00s teardown tests/distributions/test_mask.py::test_broadcast[mask_shape0-dist_shape2-event_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gamma_gaussian_tensordot[()-1-14]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-16]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Categorical-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Stable-list-sample_shape2]
0.00s teardown tests/distributions/test_delta.py::test_shapes[True-3-1]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-condition-normal_normal_model]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,ixj->ij]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(3, 2)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gaussian.py::test_affine_normal[1-3-(3, 2)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8-1.0]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[Trace_ELBO-AutoDiagonalNormal]
0.00s teardown tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[()-(3, 2)]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape1-sample_shape2]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1-1.0]
0.00s call     tests/distributions/test_pickle.py::test_pickle[IndependentHMM]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-condition-trivial_model]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,wi->i]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[2-(4,)-(6, 5)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_reshape[1-(6,)-(3, 2)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1-1.0]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[EnergyDistance_noprior-scale3]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[parallel-1]
0.00s teardown tests/infer/test_multi_sample_elbos.py::test_inner_outer[RenyiELBO]
0.00s teardown tests/infer/test_jit.py::test_dirichlet_bernoulli[TraceEnum_ELBO-False]
0.00s teardown tests/infer/test_enum.py::test_elbo_zip[0.1-0.25]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoDelta]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_guide_module_callable]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.75]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-2-0]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-0-2]
0.00s teardown tests/distributions/test_rejector.py::test_rejector[0.25-2.0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Stable-list-sample_shape1]
0.00s teardown tests/distributions/test_delta.py::test_shapes[False-3-1]
0.00s teardown tests/distributions/test_delta.py::test_shapes[False-3-0]
0.00s teardown tests/distributions/test_delta.py::test_shapes[False-1-0]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-1.0]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-replay-normal_model]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwj->ji]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1-0.5]
0.00s teardown tests/infer/test_discrete.py::test_plate_smoke[2-sample]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class11]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-auto_class10]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoMultivariateNormal-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-1-0]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-1-2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape3]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal_mvn[2-(4,)]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Stable-tuple-sample_shape0]
0.00s teardown tests/distributions/test_delta.py::test_shapes[True-1-0]
0.00s teardown tests/distributions/test_delta.py::test_shapes[False-0-0]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-condition-normal_model]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixyj,iyzj->ij]
0.00s teardown tests/ops/test_ssm_gp.py::test_matern_kernel[2.5-1]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[True-A2]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[map-ai,abij,acik->i-ijk]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.7--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99-1.0]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gamma_normal[False-JIT=True]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-auto_class10]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceEnum_ELBO-AutoDiagonalNormal]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[Stable-tuple-sample_shape2]
0.00s teardown tests/distributions/test_delta.py::test_shapes[True-3-0]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-0.01]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape1-sample_shape1]
0.00s call     tests/distributions/test_pickle.py::test_pickle[MaskedDistribution]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-wxi,xyi,yzi,zwi->i]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwxj,ixyj,iyzj,izwj->ji]
0.00s teardown tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape1-_cummin]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[3-10.0]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[1-(3, 2)-(6, 5)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.3-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8--1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.4--0.5]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gamma_normal[False-JIT=False]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-Trace_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceTMC_ELBO-subsample]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[TraceEnum_ELBO-AutoDiagonalNormal]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-2-2]
0.00s teardown tests/distributions/test_rejector.py::test_rejector[0.5-0.5]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal_mvn[2-(2, 3)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-()-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[()-(3, 2)-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[()-()-Real()]
0.00s teardown tests/distributions/test_hmm.py::test_sequential_gaussian_tensordot[()-1-13]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-False-tuple]
0.00s teardown tests/distributions/test_delta.py::test_shapes[False-2-2]
0.00s teardown tests/distributions/test_delta.py::test_shapes[False-1-1]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-1.0]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape1-sample_shape1]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape1-sample_shape2]
0.00s teardown tests/ops/test_tensor_utils.py::test_precision_to_scale_tril[batch_shape1-event_shape1]
0.00s teardown tests/ops/test_stats.py::test_fit_generalized_pareto[0.8-0.2]
0.00s teardown tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape2-_hpdi]
0.00s teardown tests/ops/test_gaussian.py::test_mvn_to_gaussian[3-(3, 2)-(7,)]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.0--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.99-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.8--0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1--0.5]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_single[OneHotCategorical-prior4]
0.00s teardown tests/distributions/test_spanning_tree.py::test_make_complete_graph[python-3-expected_grid1]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-1-1]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-2-0]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-0-1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-False-list]
0.00s teardown tests/distributions/test_delta.py::test_shapes[True-2-1]
0.00s teardown tests/distributions/test_delta.py::test_shapes[False-2-0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges423]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape2-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_distribution_validate_args[False-Exponential-args2]
0.00s teardown tests/poutine/test_properties.py::test_commutes[trace-replay-trivial_model]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixyj,iyzj->ji]
0.00s teardown tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape2-autocorrelation]
0.00s teardown tests/ops/test_ssm_gp.py::test_matern_kernel[0.5-3]
0.00s teardown tests/ops/test_ssm_gp.py::test_matern_kernel[0.5-1]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[map-a,abi,bcij->-ij]
0.00s teardown tests/infer/test_jit.py::test_grad_expand
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model[20]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class12]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[2-(2, 3)]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-0-2]
0.00s teardown tests/distributions/test_delta.py::test_shapes[False-3-2]
0.00s teardown tests/distributions/test_delta.py::test_shapes[False-2-1]
0.00s call     tests/ops/test_indexing.py::test_shape[Vindex(z(2))[0]-()]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,wxi,xyi,yzi->i]
0.00s teardown tests/infer/mcmc/test_mcmc_api.py::test_sequential_consistent
0.00s teardown tests/infer/test_jit.py::test_svi_enum[2-parallel-parallel]
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal_100-0.1]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-2-2]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-2-1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LogNormal-list-sample_shape2]
0.00s teardown tests/distributions/test_delta.py::test_shapes[True-2-0]
0.00s teardown tests/distributions/test_delta.py::test_shapes[True-1-1]
0.00s teardown tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape1-effective_sample_size]
0.00s teardown tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape2-autocovariance]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[3-0.1]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-1.01-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[StableReparam-0.1-0.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0-0.5]
0.00s teardown tests/infer/test_jit.py::test_beta_bernoulli[TraceEnum_ELBO-True]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_single[OneHotCategorical-prior6]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_class10]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-AutoDelta]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-2.0]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-1-2]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-2-0]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_beta[0.5-0.5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape3]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[()-()-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Stable-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LogNormal-Size-sample_shape1]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-0.01]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixyj,iyzj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixyj,iyzj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-,w->]
0.00s teardown tests/ops/test_tensor_utils.py::test_convolve_shape[full-4-3]
0.00s teardown tests/ops/test_stats.py::test_fit_generalized_pareto[0.8-0.5]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[2-1.0]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_guide_not_model_error[True-Trace_ELBO-full]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median[TraceGraph_ELBO-auto_guide_callable]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoDelta]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-0-0]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-1-0]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal_mvn[1-(4,)]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-True-tuple]
0.00s teardown tests/distributions/test_delta.py::test_shapes[True-0-0]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-0.1]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.05]
0.00s teardown tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape1-autocovariance]
0.00s teardown tests/ops/test_ssm_gp.py::test_matern_kernel[2.5-2]
0.00s teardown tests/ops/test_newton.py::test_newton_step_trust[3-1.0]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_add[3-(3, 2)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_add[2-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-mask3]
0.00s teardown tests/infer/test_jit.py::test_svi[TraceEnum_ELBO-10]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.25]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-2-1]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-()-Interval(lower_bound=0.0, upper_bound=1.0)]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(4,)-()-Real()]
0.00s teardown tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(3, 2)-()]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-10.0]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/test_generic.py::test_rng_seed[pyro-beta_binomial]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-i,wi->i]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-w,wx,x->]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[7-size1]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[3-size2]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[3-size1]
0.00s teardown tests/ops/test_stats.py::test_fit_generalized_pareto[1.3-0.2]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->bij-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->a-ij]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0-0.5]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gaussian_mixture_model[JIT=False]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.25]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[2-()]
0.00s teardown tests/distributions/test_stable.py::test_shape[batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[2-1-2]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-2-1]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-0-2]
0.00s teardown tests/distributions/test_rejector.py::test_rejector[1.0-1.0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[DirichletMultinomial-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LogNormal-tuple-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-1]
0.00s teardown tests/contrib/oed/test_linear_models_eig.py::test_laplace_linear_model
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges210]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwxj,ixyj,iyzj,izwj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,wxi,xyi,yzi->i]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-wxi,xyi,yzi,zwi->i]
0.00s teardown tests/ops/test_tensor_utils.py::test_precision_to_scale_tril[batch_shape0-event_shape0]
0.00s teardown tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape0-effective_sample_size]
0.00s teardown tests/ops/test_ssm_gp.py::test_matern_kernel[0.5-2]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[3-None]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[map-ai,ai->i-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[,i,i,ia,ia->,i,ia-i]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_ok[TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_single[Categorical-prior2]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.75]
0.00s teardown tests/distributions/test_stable.py::test_normal[0.5-0]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-1-1]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype0-size0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-False-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-False-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LKJCorrCholesky-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-False-Size]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwxj,ixyj,iyzj,izwj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-wi,wxi,xi->i]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[8-size1]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[2-size1]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size1-3]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_logsumexp[1-()]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.00s teardown tests/infer/test_predictive.py::test_deterministic[event_shape1-True]
0.00s teardown tests/infer/test_predictive.py::test_deterministic[event_shape0-False]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_single[Categorical-prior3]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_single[Categorical-prior1]
0.00s teardown tests/infer/test_autoguide.py::test_median[Trace_ELBO-AutoNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.75]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_normal[0.5--1]
0.00s teardown tests/distributions/test_distributions.py::test_subsequent_expands_ok[LogNormal-False-sample_shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LogNormal-list-sample_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LogNormal-tuple-sample_shape2]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[Trace_ELBO-AutoDiagonalNormal]
0.00s teardown tests/contrib/test_minipyro.py::test_constraints[minipyro-jit]
0.00s teardown tests/poutine/test_poutines.py::test_enumerate_poutine[2--1]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,ixj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-ji,iwj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwxj,ixyj,iyzj,izwj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-ji,iwj->ji]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size2-2]
0.00s teardown tests/ops/test_contract.py::test_ubersum[ab->,a,b,ab,ba-]
0.00s teardown tests/infer/reparam/test_neutra.py::test_neals_funnel_smoke[False]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_reparam_stable[HMC-kwargs0]
0.00s teardown tests/infer/test_valid_models.py::test_dim_allocation_ok[Trace_ELBO-False]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-EnergyDistance_prior]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-1.0]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[1-0-0]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-0-1]
0.00s teardown tests/distributions/test_rejector.py::test_rejector[0.25-0.5]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape3]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[MultivariateStudentT-True-Size]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges591]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape1-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape1-sample_shape1]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,ixj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-w,wx,x->]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[2-size2]
0.00s teardown tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape1-autocorrelation]
0.00s teardown tests/ops/test_ssm_gp.py::test_matern_kernel[1.5-3]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[,i,i->,i-i]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-Trace_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.0]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[0-0]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal_mvn[1-()]
0.00s teardown tests/distributions/test_improper_uniform.py::test_improper_uniform[(3, 2)-()-Real()]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-False-Size]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-100]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.1]
0.00s teardown tests/test_generic.py::test_model_sample[pyro-beta_binomial]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[8-size2]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[7-size2]
0.00s teardown tests/ops/test_contract.py::test_ubersum[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-scale3]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceEnum_ELBO-True]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[29]
0.00s teardown tests/infer/test_jit.py::test_dirichlet_bernoulli[TraceEnum_ELBO-True]
0.00s teardown tests/infer/test_jit.py::test_bernoulli_enumerate[shape2-False]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_guide_list_x]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_quantiles[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[3-4]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-100]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoDiagonalNormal]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-ij,iwj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-wx,xy,yz,zw->]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[map-a,a->-]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.00s teardown tests/infer/reparam/test_stable.py::test_subsample_smoke[SymmetricStableReparam-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[None-inf]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.25]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[2-0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LogNormal-list-sample_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LogNormal-Size-sample_shape2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-3]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs2-dims2-1]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,wx,xy,yz->]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[1-size2]
0.00s teardown tests/ops/test_newton.py::test_newton_step_converges[2-10.0]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_distribution[3-(3, 2)]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_model_error_stray_batch_dims[HMC-kwargs0]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-TraceTMC_ELBO]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[25]
0.00s teardown tests/infer/test_predictive.py::test_deterministic[event_shape0-True]
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal_100-1e-08]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoNormal-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-1.1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.5-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.1-0.75]
0.00s teardown tests/distributions/test_stable.py::test_normal[2--1]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[cpp-10]
0.00s teardown tests/distributions/test_reshape.py::test_extra_event_dim_overflow[0-1-1]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[NaiveDirichlet-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-True-Size]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[LogNormal-False-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_distribution_validate_args[True-Normal-args0]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_whiten_vsgp
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj,iwxj,ixj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-wx,xy,yz,zw->]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,ixj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwj->ij]
0.00s teardown tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape0-_cummin]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[map-ab,bc->-]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-aij->ai-ij]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7-1.0]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3-0.5]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01--1.0]
0.00s teardown tests/infer/test_jit.py::test_svi[TraceMeanField_ELBO-10]
0.00s teardown tests/infer/test_discrete.py::test_distribution_1[sample-elbo]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[10.0-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.25]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[3-()]
0.00s teardown tests/distributions/test_distributions.py::test_expand_by[LogNormal-Size-sample_shape0]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_dirichlet[TraceMeanField_ELBO-AutoIAFNormal]
0.00s teardown tests/test_generic.py::test_rng_state[pyro-eight_schools]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges908]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges328]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[replay-trivial_model]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,wxi,xi->i]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwxj,ixj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-wi,wxi,xyi,yzi->i]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[ubersum-aij->aj-ij]
0.00s teardown tests/nn/test_module.py::test_constraints[shape18-constraint_18]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_model_error_stray_batch_dims[NUTS-kwargs1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-True]
0.00s teardown tests/infer/test_discrete.py::test_distribution_3[sample-elbo]
0.00s teardown tests/infer/test_conjugate_gradients.py::ConjugateChainGradientTests::test_gradients
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal_100-0.0001]
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal-1e-08]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.0]
0.00s teardown tests/distributions/test_unit.py::test_expand[batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[4-(5,)]
0.00s teardown tests/distributions/test_stable.py::test_normal[2-20]
0.00s teardown tests/distributions/test_rejector.py::test_rejector[1.0-2.0]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_mvn_shape[2-()]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-100]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-10]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.2]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1101]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi->i]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,iwj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-ij,iwj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-wi,wxi,xi->i]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-wi,xi->i]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[1-size1]
0.00s teardown tests/ops/test_contract.py::test_ubersum[bi,ia->,i,ia,ib,iab-i]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-True]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-TraceEnum_ELBO]
0.00s teardown tests/infer/test_discrete.py::test_distribution_masked[map]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_class9]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoIAFNormal-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoMultivariateNormal-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[3-3]
0.00s teardown tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape1]
0.00s teardown tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape2]
0.00s teardown tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape1]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_mvn_shape[3-(2, 3)]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_mvn_shape[3-(4,)]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_mvn_shape[1-(2, 3)]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype2-event_shape2]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype2-event_shape1]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.9-10]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.05]
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_subsample_smoke[None]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[a,ai,abij->bij-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[,ai,abij->aij-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.00s teardown tests/ops/test_contract.py::test_ubersum[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[sequential-inf]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceGraph_ELBO-True]
0.00s teardown tests/infer/test_discrete.py::test_distribution_1[sample]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoLowRankMultivariateNormal-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.1-0.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[1.0-0.25]
0.00s teardown tests/distributions/test_stable.py::test_normal[1-21]
0.00s teardown tests/distributions/test_stable.py::test_normal[1--1]
0.00s teardown tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-True-list]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-True-Size]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1264]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-wi,xi->i]
0.00s teardown tests/ops/test_contract.py::test_adjoint_shape[map-a->-]
0.00s teardown tests/ops/test_contract.py::test_ubersum[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoMultivariateNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_mean_variance[0.9-0.5]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[2-(5,)]
0.00s teardown tests/distributions/test_stable.py::test_normal[2-1]
0.00s teardown tests/distributions/test_stable.py::test_normal[1-0]
0.00s teardown tests/distributions/test_stable.py::test_normal[0.5-1]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[2-2]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[2-(2, 3)]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_mvn_shape[2-(2, 3)]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_vsgp
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.1]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges396]
0.00s teardown tests/ops/test_gaussian.py::test_add[3-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[,i,ia->,i,ia-i]
0.00s teardown tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0--0.5]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide[AutoGuideList-init_to_median]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-auto_class6]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoDiagonalNormal]
0.00s teardown tests/distributions/test_stable.py::test_normal[0.5-20]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape3]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal_mvn[3-()]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal_mvn[2-()]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype2-event_shape0]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype1-event_shape2]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.99-10]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-10]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.9-0.05]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1015]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges545]
0.00s call     tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-ClippedAdam-optim_kwargs1-py]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-iwj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-i,wi->i]
0.00s teardown tests/ops/test_welford.py::test_welford_diagonal[1-1]
0.00s teardown tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape0-_pi]
0.00s teardown tests/ops/test_gaussian.py::test_add[1-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[ai->i-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[abi,b->,b,ai,abi-i]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[,i,i,ia,ia->,i,ia-i]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide[AutoGuideList-None]
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoMultivariateNormal-0.1]
0.00s teardown tests/distributions/test_stable.py::test_normal[1-1]
0.00s teardown tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping1-x_shape2]
0.00s teardown tests/distributions/test_lkj.py::test_corr_cholesky_transform[mapping0-x_shape4]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_mvn_shape[1-()]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.01-100]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.2]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[block-normal_model]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-i->i]
0.00s teardown tests/ops/test_welford.py::test_welford_dense[1-1]
0.00s teardown tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape0-_hpdi]
0.00s teardown tests/ops/test_contract.py::test_ubersum[i->,i-i]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceEnum_ELBO-subsample]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-auto_class7]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_class8]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoMultivariateNormal]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[3-5]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[2-5]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-True-tuple]
0.00s teardown tests/distributions/test_distributions.py::test_distribution_validate_args[True-Exponential-args2]
0.00s teardown tests/contrib/tracking/test_em.py::test_em_nested_in_svi[True]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-2]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-wi->i]
0.00s teardown tests/ops/test_gaussian.py::test_logsumexp[1-()]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[a,ai,bij->bij-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum[,i,i,ia,ia->,i,ia-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum[,i,ia->,i,ia-i]
0.00s teardown tests/nn/test_module.py::test_constraints[shape12-constraint_12]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceTMC_ELBO-full]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoDelta-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[1-()]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[2-4]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[2-1]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[2-0]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype0-event_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-False-Size]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.5-10]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.03]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1041]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-iwj,ixj->ij]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[idct--2]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[10-event_shape0]
0.00s teardown tests/ops/test_contract.py::test_ubersum_collide_implemented[naive_ubersum-True]
0.00s teardown tests/ops/test_contract.py::test_ubersum[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.00s teardown tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceTMC_ELBO-False]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_guide_callable]
0.00s teardown tests/distributions/test_unit.py::test_expand[batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[4-()]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[3-(5,)]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[3-1]
0.00s teardown tests/distributions/test_rejector.py::test_rejector[1.0-0.5]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_mvn_shape[3-()]
0.00s teardown tests/distributions/test_distributions.py::test_distribution_validate_args[True-Gamma-args1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.2-0.05]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.1-0.2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.01]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1350]
0.00s call     tests/contrib/gp/test_models.py::test_inference[VariationalGP_y1D]
0.00s teardown tests/ops/test_gaussian.py::test_add[3-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[ai->ai-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs0-dims0-1]
0.00s teardown tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceEnum_ELBO-False]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-auto_guide_module_callable]
0.00s teardown tests/distributions/test_stable.py::test_normal[2-21]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[3-2]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[1-(4,)]
0.00s teardown tests/distributions/test_distributions.py::test_expand_existing_dim[Dirichlet-False-tuple]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-ji,iwj->ji]
0.00s teardown tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape1-_quantile]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-ai->a-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->bi-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s teardown tests/nn/test_module.py::test_constraints[shape21-constraint_21]
0.00s teardown tests/nn/test_module.py::test_constraints[shape13-constraint_13]
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_nonreparameterized_N_is_3
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoMultivariateNormal-0.0001]
0.00s teardown tests/distributions/test_stable.py::test_normal[1-20]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[1-2]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[3-0]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[2-(4,)]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[1-()]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_mvn_shape[2-(4,)]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.2]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges760]
0.00s teardown tests/ops/test_gaussian.py::test_add[2-(4,)]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_add[2-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs10-dims10-2]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs4-dims4-1]
0.00s teardown tests/infer/reparam/test_stable.py::test_subsample_smoke[StableReparam-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_ok[TraceTMC_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide[AutoNormal-init_to_median]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide[AutoDelta-None]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_iplate_smoke[Trace_ELBO-AutoDiagonalNormal]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoLaplaceApproximation-TraceEnum_ELBO]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[0-0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape5]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[3-(4,)]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype0-event_shape0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.5-0.1]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges667]
0.00s teardown tests/ops/test_contract.py::test_adjoint_marginal[a,abi,bcij->-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[a,abi,bcij->cij-ij]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[,i,ia->,i,ia-i]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs17-dims17-2]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs14-dims14-3]
0.00s teardown tests/nn/test_module.py::test_constraints[shape9-constraint_9]
0.00s teardown tests/nn/test_module.py::test_constraints[shape8-constraint_8]
0.00s teardown tests/infer/test_jit.py::test_svi[Trace_ELBO-10]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoNormal]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoDelta]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[3-()]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[1-(2, 3)]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[1-(2, 3)]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype1-event_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype0-event_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_distribution_validate_args[False-Exponential-args2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_asymptote[0.8-0.1]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges967]
0.00s teardown tests/ops/test_contract.py::test_adjoint_marginal[ab,bc->-]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[ai->-i]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs9-dims9-3]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs7-dims7-3]
0.00s teardown tests/nn/test_module.py::test_constraints[shape7-constraint_7]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-mask3]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceEnum_ELBO-full]
0.00s teardown tests/infer/test_discrete.py::test_distribution_3[map]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceEnum_ELBO-AutoDelta]
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal-0.0001]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape6]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[3-(2, 3)]
0.00s teardown tests/distributions/test_binomial.py::test_binomial_approx_sample[0.1-100]
0.00s teardown tests/ops/test_contract.py::test_adjoint_marginal[ai,abij->i-ij]
0.00s teardown tests/ops/test_contract.py::test_adjoint_marginal[a,abi->-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[ab,bcdi,deij->eij-ij]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs15-dims15-3]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_beta_binomial[False]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[Trace_ELBO-subsample]
0.00s teardown tests/infer/test_jit.py::test_svi[TraceEnum_ELBO-1]
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoDiagonalNormal-0.0001]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[2-3]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges430]
0.00s teardown tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[0-False]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-wi,wi->i]
0.00s teardown tests/ops/test_gaussian.py::test_add[2-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[,i,i->,i-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum[,i->,i-i]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[sequential-1]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-TraceTMC_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide[AutoNormal-init_to_mean]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-auto_class6]
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoDiagonalNormal-1e-08]
0.00s teardown tests/distributions/test_torch_patch.py::test_lower_cholesky_transform[1-(5,)]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_delta_mvn_shape[2-()]
0.00s teardown tests/distributions/test_empirical.py::test_weighted_sample_coherence[dtype1-event_shape0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1358]
0.00s call     tests/ops/test_contract.py::test_partition_terms[inputs1-dims1-1]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs13-dims13-4]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs8-dims8-2]
0.00s teardown tests/infer/reparam/test_neutra.py::test_reparam_log_joint[neals_funnel-kwargs0]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide[AutoNormal-None]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-nested_auto_guide_callable]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_mvn_shape[1-(4,)]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges195]
0.00s teardown tests/poutine/test_poutines.py::QueueHandlerDiscreteTest::test_queue_single
0.00s teardown tests/ops/test_gaussian.py::test_add[1-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs18-dims18-1]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs3-dims3-2]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs2-dims2-1]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs1-dims1-1]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-TraceTMC_ELBO]
0.00s teardown tests/infer/test_jit.py::test_traceenum_elbo[10]
0.00s teardown tests/infer/test_jit.py::test_svi[JitTraceEnum_ELBO-10]
0.00s teardown tests/infer/test_discrete.py::test_distribution_3[sample]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide[AutoDelta-init_to_mean]
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoLowRankMultivariateNormal-0.1]
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoDiagonalNormal-0.1]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoDiagonalNormal-TraceEnum_ELBO]
0.00s teardown tests/contrib/test_util.py::test_rvv[a0-b0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1241]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges851]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges320]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceEnum_ELBO-None]
0.00s teardown tests/infer/test_jit.py::test_categorical_enumerate[shape0-True]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[30]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[4-5]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-nested_auto_guide_callable]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[3-6]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-10.0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges812]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges422]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges354]
0.00s teardown tests/poutine/test_mapdata.py::test_replay[iplate-20]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_add[1-(4,)]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs5-dims5-4]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-0.1]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplate_plate_dependency_ok[None-1]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_guide_module_callable]
0.00s teardown tests/infer/test_autoguide.py::test_init_scale[AutoMultivariateNormal-1e-08]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1088]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges726]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[8-size0]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs16-dims16-3]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gamma_poisson[False]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-TraceEnum_ELBO]
0.00s teardown tests/contrib/oed/test_linear_models_eig.py::test_marginal_linear_model
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.3]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1026]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixj->ji]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_distribution[3-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_plate_error[naive_ubersum-,ai->a-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_beta_binomial[True]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[31]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceEnum_ELBO-AutoDiagonalNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_class8]
0.00s teardown tests/contrib/test_minipyro.py::test_local_param_ok[minipyro-jit]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges825]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_add[3-(4,)]
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_two_layers
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceEnum_ELBO-sequential]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_guide_callable]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_guide_module_callable]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[2-100.0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1075]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges786]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[idct-0]
0.00s teardown tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape2-_quantile]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[,i,j,a,ij,ia,ja,ija->,a,i,j,ia,ja,ij,ija-ij]
0.00s teardown tests/ops/test_contract.py::test_partition_terms[inputs12-dims12-1]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_beta_bernoulli[JIT=True]
0.00s teardown tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape3]
0.00s teardown tests/infer/test_jit.py::test_svi[TraceGraph_ELBO-10]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_beta[4.0-1.0]
0.00s teardown tests/distributions/test_distributions.py::test_distribution_validate_args[False-Gamma-args1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[1.0-1]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1298]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[dct-2]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[dct--4]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size0-2]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_dirichlet_categorical[JIT=False]
0.00s teardown tests/infer/test_jit.py::test_categorical_enumerate[shape1-True]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoDiagonalNormal]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_beta[1.0-4.0]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-1.0]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-0.1]
0.00s call     tests/optim/test_multi.py::test_multi_optimizer_overlap_error
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceEnum_ELBO-parallel]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-4]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide_2[AutoNormal-dependent]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges997]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges913]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges906]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges617]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges610]
0.00s call     tests/distributions/test_util.py::test_weakmethod
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[4-event_shape2]
0.00s teardown tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape2-_pi]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[TraceMeanField_ELBO-1]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[Trace_ELBO-parallel]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-TraceTMC_ELBO]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[3-5]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_guide_list_x]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-AutoDelta]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoIAFNormal-Trace_ELBO]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_beta[4.0-4.0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1200]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1054]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1002]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges200]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[3-event_shape2]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[inf]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceGraph_ELBO-parallel]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceGraph_ELBO-sequential]
0.00s teardown tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape1]
0.00s teardown tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape1]
0.00s teardown tests/infer/test_jit.py::test_svi[JitTraceEnum_ELBO-1]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide[AutoDelta-init_to_median]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.1--0.5]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs4]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_mixture.py::test_masked_mixture_multivariate[batch_shape1-sample_shape2]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges401]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges367]
0.00s teardown tests/ops/test_welford.py::test_welford_diagonal[1000-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size0-1]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[4-event_shape1]
0.00s teardown tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape0-autocorrelation]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op1-JIT=True]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-mask3]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-False]
0.00s teardown tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape3]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_class9]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoLowRankMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_guide_module_callable]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-10.0]
0.00s teardown tests/contrib/tracking/test_em.py::test_em_nested_in_svi[False]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges895]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges393]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges78]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[idct-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[2-size0]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-0.1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[TraceMeanField_ELBO-scale2]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[Trace_ELBO-sequential]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-5]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_guide_list_x]
0.00s teardown tests/distributions/test_unit.py::test_expand[batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_distribution_validate_args[False-Normal-args0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges745]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges239]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges47]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size2-1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[TraceMeanField_ELBO-0.1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-False]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_ok[TraceTMC_ELBO-subsample]
0.00s teardown tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape0]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoDelta]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_guide_callable]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_guide_list_x]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoMultivariateNormal]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_beta[1.0-0.5]
0.00s teardown tests/contrib/gp/test_models.py::test_mean_function_VSGP_whiten
0.00s teardown tests/contrib/test_minipyro.py::test_local_param_ok[pyro-jit]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1388]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1363]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges846]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges838]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges497]
0.00s teardown tests/ops/test_welford.py::test_welford_dense[1000-1]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[dct--1]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[dct--2]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size1-2]
0.00s teardown tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape0-gelman_rubin]
0.00s teardown tests/ops/test_stats.py::test_statistics_B_ok_with_sample_shape[sample_shape0-autocovariance]
0.00s teardown tests/ops/test_contract.py::test_ubersum[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-True]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-mask2]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[TraceMeanField_ELBO-True]
0.00s teardown tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape2]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoNormal]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoLowRankMultivariateNormal-TraceGraph_ELBO]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1418]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges226]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1--1.0]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[1-event_shape1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-mask2]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-mask4]
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_dbn[False-False]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[28]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-auto_class7]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_guide_module_callable]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-AutoDelta]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_guide_list_x]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoNormal]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoIAFNormal-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_beta[1.0-1.0]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_beta[0.5-1.0]
0.00s teardown tests/distributions/test_conjugate_update.py::test_gamma_poisson[(3, 2)-(3, 2)]
0.00s teardown tests/test_generic.py::test_rng_state[minipyro-logistic_regression]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-w,wx,xy,yz->]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[dct-3]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[2-event_shape1]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-False]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-mask3]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-Trace_ELBO]
0.00s teardown tests/infer/test_svgd.py::test_shapes[RBFSteinKernel-shape2]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoDelta-Trace_ELBO]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Geometric]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges241]
0.00s teardown tests/poutine/test_poutines.py::test_replay_enumerate_poutine[2--2]
0.00s teardown tests/poutine/test_poutines.py::test_replay_enumerate_poutine[2--1]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-w->]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[7-size0]
0.00s teardown tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape0-split_gelman_rubin]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-scale2]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[Trace_ELBO-mask2]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_param_ok[Trace_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_ok[Trace_ELBO-subsample]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_guide_list_x]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoLowRankMultivariateNormal-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoDelta-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_rejector.py::test_shape_augmented_beta[0.5-4.0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape3]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal[(2, 3)-(4,)]
0.00s teardown tests/test_generic.py::test_trace_handler[minipyro-logistic_regression]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1308]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges983]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges968]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges380]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges260]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0--1.0]
0.00s teardown tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape2-gelman_rubin]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[ia,jb,ijab->,i,ia,j,jb,ij,ija,ijb,ijab-ij]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes2-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoNormal]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[Trace_ELBO-auto_class8]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoLaplaceApproximation-TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoDiagonalNormal-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-0.5]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal[()-(2, 3)]
0.00s teardown tests/contrib/test_minipyro.py::test_generate_data[pyro]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1221]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1127]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges435]
0.00s teardown tests/poutine/test_poutines.py::test_replay_enumerate_poutine[2--3]
0.00s teardown tests/poutine/test_poutines.py::test_replay_enumerate_poutine[1--3]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-ij->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-iwj,iwxj,ixj->ij]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_distribution[2-(4,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[,i->-i]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[TraceMeanField_ELBO-scale3]
0.00s teardown tests/infer/test_valid_models.py::test_plate_ok[TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[True-TraceGraph_ELBO]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model[30]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[4-2]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-1.1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-1.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[-1.0-1.0]
0.00s teardown tests/contrib/test_minipyro.py::test_constraints[pyro-jit]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges773]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges651]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_error[shapes0]
0.00s teardown tests/poutine/test_poutines.py::test_replay_enumerate_poutine[0--3]
0.00s teardown tests/poutine/test_poutines.py::test_enumerate_poutine[0--3]
0.00s teardown tests/poutine/test_poutines.py::test_enumerate_poutine[0--2]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-ij->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-,w->]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-w,w->]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[3-event_shape1]
0.00s teardown tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape1-split_gelman_rubin]
0.00s teardown tests/ops/test_stats.py::test_diagnostics_ok_with_sample_shape[sample_shape1-gelman_rubin]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(6),z(5,1)]-(5, 6)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_marginal[ai,abij,acik->i-ijk]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_bernoulli_latent_model[JIT=False]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op0-JIT=False]
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_dbn[False-True]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_set_has_rsample_ok[Trace_ELBO-True]
0.00s teardown tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-EnergyDistance_noprior]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-10.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-0.1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-0.5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape4]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape6]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[True-p_mask3]
0.00s teardown tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[()-(4,)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_beta_binomial[(3, 2)-(4,)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_beta_binomial[(4,)-()]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[1-100.0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-3]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges104]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1102]
0.00s teardown tests/poutine/test_poutines.py::test_replay_enumerate_poutine[1--1]
0.00s teardown tests/poutine/test_poutines.py::test_replay_enumerate_poutine[0--1]
0.00s teardown tests/ops/test_welford.py::test_welford_dense[1000-7]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[2-event_shape0]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_plate_ok[TraceMeanField_ELBO-mask4]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_ok[TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide_2[AutoNormal-independent]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class10]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoLaplaceApproximation-Trace_ELBO]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal[(2, 3)-(2, 3)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(4,)-(3, 2)]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[3-1]
0.00s teardown tests/test_generic.py::test_rng_seed[minipyro-neals_funnel]
0.00s teardown tests/poutine/test_poutines.py::test_enumerate_poutine[2--3]
0.00s teardown tests/poutine/test_poutines.py::test_enumerate_poutine[1--2]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-w,w->]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-w->]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[4-event_shape0]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[2-event_shape2]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.00s teardown tests/infer/reparam/test_stable.py::test_subsample_smoke[LatentStableReparam-subsample]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op1-JIT=False]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_beta_bernoulli[JIT=False]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_param_ok[TraceGraph_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_set_has_rsample_ok[Trace_ELBO-False]
0.00s teardown tests/infer/test_jit.py::test_svi[TraceGraph_ELBO-1]
0.00s teardown tests/infer/test_autoguide.py::test_median_module[TraceGraph_ELBO-auto_class9]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoNormal]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoMultivariateNormal-Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_factor[AutoDiagonalNormal-Trace_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-10.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-0.9]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-0.1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-10.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-0.9]
0.00s teardown tests/distributions/test_conjugate_update.py::test_beta_binomial[(3, 2)-(3, 2)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_beta_binomial[(4,)-(3, 2)]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_large[3-100.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-1.5]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_init[-1.0]
0.00s teardown tests/poutine/test_poutines.py::test_enumerate_poutine[1--3]
0.00s teardown tests/poutine/test_mapdata.py::test_cond_indep_stack[plate-5]
0.00s teardown tests/ops/test_tensor_utils.py::test_repeated_matmul[1-size0]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[3-event_shape0]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_model_with_param[1-True]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_singular_matrix_catch[op0-JIT=True]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-False]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[TraceGraph_ELBO-None]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_misuse_warning[Trace_ELBO-None]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoLaplaceApproximation]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-0.1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-2.0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape6]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-p_mask3]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[True-p_mask2]
0.00s teardown tests/distributions/test_conjugate_update.py::test_beta_binomial[()-(3, 2)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_beta_binomial[()-()]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-1]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[-1.0-0.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[1.0-1.0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1436]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges518]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges273]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1282]
0.00s teardown tests/poutine/test_poutines.py::test_replay_enumerate_poutine[1--2]
0.00s teardown tests/ops/test_welford.py::test_welford_diagonal[1000-7]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_distribution[2-(3, 2)]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[aij,abi,bc,cdj,dij->,bi,cj,aij,dij,adij-ij]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[ab->,a,b,ab,ba-]
0.00s teardown tests/infer/reparam/test_stable.py::test_subsample_smoke[LatentStableReparam-full]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gaussian_hmm[2]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_bernoulli_latent_model[JIT=True]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_plate_ok[Trace_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class12]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class11]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-0.1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-2.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-1.1]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[0-2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape4]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal[(4,)-(4,)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_beta_binomial[()-(4,)]
0.00s teardown tests/contrib/test_minipyro.py::test_constraints[pyro-py]
0.00s teardown tests/test_generic.py::test_rng_state[pyro-neals_funnel]
0.00s teardown tests/poutine/test_poutines.py::test_enumerate_poutine[0--1]
0.00s teardown tests/poutine/test_mapdata.py::test_cond_indep_stack[plate-20]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-ij->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-w,x->]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-,w->]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size1-1]
0.00s teardown tests/ops/test_special.py::test_log_beta_stirling[0.10]
0.00s teardown tests/ops/test_indexing.py::test_hmm_example[-5--3]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[aik,bij,abci,cd,dej,eij->,ai,bi,ej,aik,bij,eij-ijk]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[i,jb,ijab->,i,j,jb,ij,ija,ijb,ijab-ij]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[a->,a-]
0.00s teardown tests/ops/test_contract.py::test_ubersum[ab,bcij->,a,b,ab,i,j,ij,ai,aj,aij,bi,bj,aij,bij,cij,abij,acij,bcij,abcij-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[Trace_ELBO-0.1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[Trace_ELBO-1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_mask_ok[Trace_ELBO-mask2]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_in_model_not_guide_ok[Trace_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_plate_ok[TraceGraph_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_ok[Trace_ELBO-full]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide_2[AutoDelta-independent]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-auto_class7]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-auto_class6]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[Trace_ELBO-AutoDiagonalNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-2.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-2.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-1.1]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape0]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-True]
0.00s teardown tests/contrib/test_minipyro.py::test_local_param_ok[pyro-py]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges758]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges576]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges294]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges156]
0.00s call     tests/infer/test_jit.py::test_cond_indep_equality[x1-y1]
0.00s teardown tests/poutine/test_poutines.py::test_enumerate_poutine[2--2]
0.00s teardown tests/optim/test_multi.py::test_optimizers[<lambda>0]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_distribution[1-()]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[abi,b->,b,ai,abi-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[i->,i-i]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[ab,bcdi,deij->,a,b,ci,di,eij-ij]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[ab,bci,bdi->,a,b,ab,i,ai,bi,ci,abi,bci,bdi,cdi,abci,abdi,abcdi-i]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[abi,b->,b,ai,abi-i]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.00s teardown tests/infer/reparam/test_stable.py::test_subsample_smoke[StableReparam-full]
0.00s teardown tests/infer/reparam/test_neutra.py::test_neals_funnel_smoke[True]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_model_error_enum_dim_clash[NUTS-kwargs1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_ok[Trace_ELBO-scale2]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-TraceGraph_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes1-TraceGraph_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[Trace_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_param_ok[Trace_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_set_has_rsample_ok[TraceGraph_ELBO-False]
0.00s teardown tests/infer/test_jit.py::test_svi[TraceMeanField_ELBO-1]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_vector[False-dense]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoDelta]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-AutoDelta]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-2.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-0.1]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[1-0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape6]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-True]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[DirichletMultinomial]
0.00s teardown tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(3, 2)-(4,)]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[5.0-5.0]
0.00s teardown tests/contrib/test_minipyro.py::test_plate_ok[minipyro-py]
0.00s teardown tests/test_generic.py::test_rng_seed[minipyro-logistic_regression]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges861]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-w,x->]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[dct-0]
0.00s teardown tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape1-_pi]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0,:]-(3,)]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_distribution[3-()]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[a->,a-]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[ab,bci->,a,b,ab,i,ai,bi,ci,abi,bci,abci-i]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes1-Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes2-TraceGraph_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_subsample_primitive_ok[Trace_ELBO-full]
0.00s teardown tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-TraceTMC_ELBO]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[3-5]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoDelta]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-0.5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-10.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.1-1.0]
0.00s teardown tests/distributions/test_unit.py::test_expand[batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-10]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-1]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[0-2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape6]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape3]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-p_mask2]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal[(4,)-(2, 3)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_gamma_poisson[(4,)-(4,)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[()-()]
0.00s teardown tests/distributions/test_conjugate_update.py::test_beta_binomial[(4,)-(4,)]
0.00s teardown tests/distributions/test_conjugate.py::test_log_prob_support[dist2-values2]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[1.0-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[10.0-1.0]
0.00s teardown tests/contrib/test_minipyro.py::test_plate_ok[minipyro-jit]
0.00s teardown tests/contrib/test_minipyro.py::test_plate_ok[pyro-jit]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1166]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges784]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges414]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges91]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[trace-normal_normal_model]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[trace-normal_model]
0.00s teardown tests/poutine/test_poutines.py::test_replay_enumerate_poutine[0--2]
0.00s teardown tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape1-_hpdi]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[False-A5]
0.00s teardown tests/ops/test_indexing.py::test_hmm_example[-3--4]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[bi,ia->,i,ia,ib,iab-i]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[,i,ia->,i,ia-i]
0.00s teardown tests/nn/test_module.py::test_constraints[shape5-constraint_5]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_ok[sizes0-TraceGraph_ELBO]
0.00s teardown tests/infer/test_svgd.py::test_shapes[IMQSteinKernel-shape0]
0.00s teardown tests/infer/test_jit.py::test_infer_discrete[sample-1]
0.00s teardown tests/infer/test_autoguide.py::test_linear_regression_smoke[TraceGraph_ELBO-AutoDiagonalNormal]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class11]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class10]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_guide_module_callable]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-10.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape3]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[1-1]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape6]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape5]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-p_mask3]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype1-size0]
0.00s teardown tests/distributions/test_conjugate_update.py::test_gamma_poisson[(4,)-(3, 2)]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[2-1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.3]
0.00s teardown tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_log_prob[False-relu]
0.00s teardown tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[pyro-jit]
0.00s teardown tests/test_generic.py::test_trace_handler[minipyro-neals_funnel]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1153]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges934]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges685]
0.00s teardown tests/poutine/test_mapdata.py::test_replay[plate-20]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-w,w->]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-w->]
0.00s teardown tests/ops/test_stats.py::test_statistics_A_ok_with_sample_shape[sample_shape0-_quantile]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[True-A0]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[,i,i,ia,ia->,i,ia-i]
0.00s teardown tests/nn/test_module.py::test_constraints[shape0-constraint_0]
0.00s teardown tests/infer/test_valid_models.py::test_dim_allocation_ok[TraceGraph_ELBO-False]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-TraceGraph_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_and_plate_ok[sizes0-Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-TraceTMC_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_ok[False-TraceGraph_ELBO]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[26]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class10]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-10.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-0.9]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-1.1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.5-1.0]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-100]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[2-1]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[0-3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape1]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape2-sample_shape4]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-p_mask2]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[True-True]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype1-size2]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype1-size1]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype0-size1]
0.00s teardown tests/distributions/test_conjugate_update.py::test_gamma_poisson[(3, 2)-(4,)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(3, 2)-(3, 2)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(4,)-(4,)]
0.00s teardown tests/distributions/test_categorical.py::test_batch_log_dims[dim=3-probs0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[0.0-0.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.03]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[5.0-5.0]
0.00s teardown tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[minipyro-jit]
0.00s teardown tests/test_generic.py::test_rng_seed[pyro-eight_schools]
0.00s teardown tests/test_generic.py::test_rng_seed[pyro-neals_funnel]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1423]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges286]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges182]
0.00s teardown tests/poutine/test_mapdata.py::test_replay[plate-5]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-ij->ji]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size2-3]
0.00s teardown tests/ops/test_tensor_utils.py::test_block_diag_embed[block_size0-3]
0.00s teardown tests/ops/test_gaussian.py::test_add[3-()]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[i->-i]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[ia,ja,ija->,a,i,ia,j,ja,ija-ij]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[,i->,i-i]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[ab,bc->,a,b,c,ab,bc,ac,abc-]
0.00s teardown tests/nn/test_module.py::test_decorator[1-DecoratorModel]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-EnergyDistance_prior]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoDiagonalNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-0.9]
0.00s teardown tests/distributions/test_stable.py::test_shape[batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[1-3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape6]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal[()-(4,)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_gamma_poisson[()-(4,)]
0.00s teardown tests/distributions/test_conjugate_update.py::test_dirichlet_multinomial[(4,)-()]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[4-4]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.01]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_negative_binomial_vs_poisson[1.0-10.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[10.0-1.0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges847]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[idct--1]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[idct--3]
0.00s teardown tests/ops/test_contract.py::test_adjoint_marginal[a->-]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[a,b,c,di,ei,fj->,a,b,c,di,ei,fj-ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[,i->,i-i]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[i->,i-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum[abi,bcij->,i,ai,bi,abi,j,ij,aij,bij,cij,abij,bcij,abcij-ij]
0.00s teardown tests/nn/test_module.py::test_constraints[shape6-constraint_6]
0.00s teardown tests/infer/test_valid_models.py::test_dim_allocation_error[TraceEnum_ELBO-False]
0.00s teardown tests/infer/test_valid_models.py::test_plate_ok[Trace_ELBO-subsample]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-EnergyDistance_noprior]
0.00s teardown tests/infer/test_jit.py::test_svi[JitTraceGraph_ELBO-10]
0.00s teardown tests/infer/test_jit.py::test_svi[Trace_ELBO-1]
0.00s teardown tests/infer/test_autoguide.py::test_subsample_guide_2[AutoDelta-dependent]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-0.5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-100]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[2-4]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[2-2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape6]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape4]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape4]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_polya_gamma.py::test_polya_gamma[batch_shape1]
0.00s teardown tests/distributions/test_one_hot_categorical.py::test_support_dims[dim=2-probs0]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[False-False]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal[(4,)-()]
0.00s teardown tests/distributions/test_conjugate_update.py::test_gamma_poisson[()-(3, 2)]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[1.0-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[10.0-1]
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_amortized_smoke[None]
0.00s teardown tests/test_generic.py::test_model_sample[minipyro-logistic_regression]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges947]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-ij->ij]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[idct--4]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[dct-1]
0.00s teardown tests/ops/test_indexing.py::test_hmm_example[-4--5]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),:,:]-(5, 3, 4)]
0.00s teardown tests/ops/test_contract.py::test_adjoint_marginal[a,a->-]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[,i->i-i]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[bi,ia->,i,ia,ib,iab-i]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[ab,bc,cd->,a,b,c,d,ab,ac,ad,bc,bd,cd,abc,acd,bcd,abcd-]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[->-]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_vector[True-flat]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class9]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class9]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-0.1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-1.0]
0.00s teardown tests/distributions/test_stable.py::test_variance[1.7-0.5]
0.00s teardown tests/distributions/test_stable.py::test_shape[batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[1-1]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[0-1]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape6]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape1]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[True-False]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[False-True]
0.00s teardown tests/distributions/test_categorical.py::test_batch_log_dims[dim=1-probs0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[1.0--1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-1.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.03]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.01]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges311]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[idct-2]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[False-A3]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[a->,a-]
0.00s teardown tests/nn/test_module.py::test_decorator[1-AttributeModel]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-1]
0.00s teardown tests/infer/test_valid_models.py::test_plate_reuse_ok[TraceTMC_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceGraph_ELBO-auto_class9]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-AutoDiagonalNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-1.1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape3]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape6]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape3]
0.00s teardown tests/distributions/test_one_hot_categorical.py::test_sample_dims[dim=3-probs0]
0.00s teardown tests/distributions/test_one_hot_categorical.py::test_support_dims[dim=1-probs0]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[False-p_mask3]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal[(2, 3)-()]
0.00s teardown tests/distributions/test_conjugate_update.py::test_gamma_poisson[(3, 2)-()]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-2]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.01]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.03]
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_amortized_smoke[init_to_median]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1100]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[dct--3]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[1-event_shape2]
0.00s teardown tests/ops/test_stats.py::test_crps_empirical[1-event_shape0]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_model_error_enum_dim_clash[HMC-kwargs0]
0.00s teardown tests/infer/test_jit.py::test_svi[JitTrace_ELBO-10]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[3]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-AutoDelta]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.5-1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.1-0.9]
0.00s teardown tests/distributions/test_unit.py::test_expand[batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[2.0--1.0]
0.00s teardown tests/distributions/test_stable.py::test_shape[batch_shape2-sample_shape2]
0.00s teardown tests/distributions/test_stable.py::test_shape[batch_shape2-sample_shape1]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-3]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[2-3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape1]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape6]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape5-sample_shape1]
0.00s teardown tests/distributions/test_empirical.py::test_unweighted_mean_and_var[dtype0-size2]
0.00s teardown tests/distributions/test_conjugate_update.py::test_gamma_poisson[()-()]
0.00s teardown tests/distributions/test_conjugate.py::test_log_prob_support[dist1-None]
0.00s teardown tests/distributions/test_categorical.py::test_support_dims[dim=1-probs0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_bp_vs_exact[1-4]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.03]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.01]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[1.0-10]
0.00s teardown tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[minipyro-jit]
0.00s teardown tests/test_generic.py::test_rng_state[pyro-logistic_regression]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1410]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1028]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges560]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges467]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges838]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges125]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[i,wi->i]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct_dim[idct-3]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[False-A4]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[True-A3]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_add[2-()]
0.00s teardown tests/ops/test_contract.py::test_adjoint_marginal[ai,ai->i-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum_total[i->i-i]
0.00s teardown tests/ops/test_contract.py::test_ubersum[,i,i->,i-i]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[bi,ia->,i,ia,ib,iab-i]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[,i,i->,i-i]
0.00s teardown tests/infer/reparam/test_stable.py::test_subsample_smoke[SymmetricStableReparam-full]
0.00s teardown tests/infer/test_jit.py::test_infer_discrete[sample-2]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_vector[False-flat]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class12]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.99-1.0]
0.00s teardown tests/distributions/test_stable.py::test_shape[batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_reshape.py::test_reshape_reshape[0-1]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape5-sample_shape0]
0.00s teardown tests/distributions/test_one_hot_categorical.py::test_sample_dims[dim=1-probs0]
0.00s teardown tests/distributions/test_conjugate.py::test_log_prob_support[dist0-None]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-1.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-1.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_beta_binomial_vs_binomial[1.0-10.0]
0.00s teardown tests/contrib/test_minipyro.py::test_plate_ok[pyro-py]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1057]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges941]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges593]
0.00s teardown tests/poutine/test_mapdata.py::test_cond_indep_stack[iplate-20]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[marginal-->]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2),:]-(2, 4)]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[ab,bcdi,defij,fgijk->,a,b,ci,di,eij,fij,gijk-ijk]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[ab->,a,b,ab,ba-]
0.00s teardown tests/nn/test_module.py::test_constraints[shape17-constraint_17]
0.00s teardown tests/infer/test_valid_models.py::test_vectorized_num_particles[TraceEnum_ELBO]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[20]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model[10]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[4-2]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[TraceEnum_ELBO-auto_class12]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_1_gate[0.9-0.9]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zinb_0_gate[0.9-0.5]
0.00s teardown tests/distributions/test_stable.py::test_variance[1.7-1.5]
0.00s teardown tests/distributions/test_stable.py::test_shape[batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_approx_smoke[python-30]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape4-sample_shape4]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape3-sample_shape5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape4]
0.00s teardown tests/distributions/test_one_hot_categorical.py::test_support_dims[dim=3-probs0]
0.00s teardown tests/distributions/test_categorical.py::test_sample_dims[dim=2-probs0]
0.00s teardown tests/contrib/gp/test_models.py::test_mean_function_VGP
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.03]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-1.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-0.3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-1.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_binomial_vs_poisson[5.0-5]
0.00s teardown tests/test_generic.py::test_trace_handler[pyro-logistic_regression]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1163]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges531]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[True-A5]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[i->,i-i]
0.00s teardown tests/nn/test_module.py::test_pyro_serialize
0.00s teardown tests/nn/test_module.py::test_constraints[shape15-constraint_15]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gamma_poisson[True]
0.00s teardown tests/infer/test_jit.py::test_svi[JitTraceGraph_ELBO-1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape1]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.7-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.3--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.000001-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.999999-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.4--1.0]
0.00s teardown tests/distributions/test_stable.py::test_shape[batch_shape2-sample_shape0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape2-sample_shape4]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_standard_gamma_sample_shape[batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[q_mask2-False]
0.00s teardown tests/distributions/test_conjugate.py::test_log_prob_support[dist3-values3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-1.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-0.1]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.7-0.3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.1]
0.00s teardown tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[pyro-py]
0.00s teardown tests/test_generic.py::test_rng_seed[pyro-logistic_regression]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1432]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1175]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges864]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges333]
0.00s call     tests/infer/test_jit.py::test_cond_indep_equality[x0-y0]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1342]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges900]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-i->i]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[True-A4]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_add[3-()]
0.00s teardown tests/ops/test_contract.py::test_ubersum[a->,a-]
0.00s teardown tests/nn/test_module.py::test_constraints[shape20-constraint_20]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_dirichlet_categorical[JIT=True]
0.00s teardown tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-TraceGraph_ELBO]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[30]
0.00s teardown tests/infer/test_predictive.py::test_posterior_predictive_svi_one_hot
0.00s teardown tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape1-True]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_vector[True-dense]
0.00s teardown tests/infer/test_autoguide.py::test_autoguide_serialization[Trace_ELBO-auto_class11]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape2]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape3]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.7--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.999999--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.8--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.99-1.0]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-30]
0.00s teardown tests/distributions/test_one_hot_categorical.py::test_sample_dims[dim=2-probs0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.97-0.01]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.9-1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-1.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.03-0.01]
0.00s teardown tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[minipyro-py]
0.00s teardown tests/test_generic.py::test_rng_state[minipyro-neals_funnel]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1096]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges706]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges521]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges448]
0.00s teardown tests/ops/test_contract.py::test_ubersum_5[naive_ubersum]
0.00s teardown tests/nn/test_module.py::test_constraints[shape23-constraint_23]
0.00s teardown tests/nn/test_module.py::test_constraints[shape16-constraint_16]
0.00s teardown tests/infer/test_valid_models.py::test_dim_allocation_error[Trace_ELBO-False]
0.00s teardown tests/infer/test_jit.py::test_infer_discrete[map-1]
0.00s teardown tests/infer/test_jit.py::test_categorical_enumerate[shape2-False]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[3-2]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.000001--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.1-0.5]
0.00s teardown tests/distributions/test_stable.py::test_shape[batch_shape1-sample_shape0]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-10]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[1-4]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[1-3]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape3]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[q_mask3-False]
0.00s teardown tests/distributions/test_kl.py::test_kl_independent_normal[()-()]
0.00s teardown tests/distributions/test_categorical.py::test_batch_log_dims[dim=2-probs0]
0.00s teardown tests/distributions/test_categorical.py::test_sample_dims[dim=3-probs0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.99-0.03]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.3]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.3-0.01]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.01-0.03]
0.00s teardown tests/contrib/bnn/test_hidden_layer.py::test_hidden_layer_log_prob[True-relu]
0.00s teardown tests/contrib/test_util.py::test_rmv[A0-b0]
0.00s teardown tests/contrib/test_minipyro.py::test_constraints[minipyro-py]
0.00s teardown tests/contrib/test_minipyro.py::test_nonempty_model_empty_guide_ok[pyro-py]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1414]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1187]
0.00s teardown tests/poutine/test_mapdata.py::test_replay[nested_iplate-5]
0.00s teardown tests/poutine/test_mapdata.py::test_cond_indep_stack[iplate-5]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[sample-->]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-->]
0.00s teardown tests/ops/test_special.py::test_log_beta_stirling[0.05]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[False-A0]
0.00s teardown tests/ops/test_contract.py::test_einsum_linear[,i->,i-i]
0.00s teardown tests/nn/test_module.py::test_decorator[2-AttributeModel]
0.00s teardown tests/nn/test_module.py::test_constraints[shape24-constraint_24]
0.00s teardown tests/nn/test_module.py::test_constraints[shape2-constraint_2]
0.00s teardown tests/nn/test_module.py::test_constraints[shape1-constraint_1]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_scale_plate_ok[Trace_ELBO-scale2]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.000001--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.01-0.0]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[1-0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape2]
0.00s teardown tests/distributions/test_conjugate_update.py::test_gamma_poisson[(4,)-()]
0.00s teardown tests/contrib/gp/test_models.py::test_mean_function_VGP_whiten
0.00s teardown tests/test_generic.py::test_model_sample[pyro-eight_schools]
0.00s teardown tests/test_generic.py::test_model_sample[pyro-neals_funnel]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1095]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1012]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape0-sample_shape0]
0.00s teardown tests/poutine/test_mapdata.py::test_replay[iplate-5]
0.00s teardown tests/nn/test_module.py::test_decorator[2-DecoratorModel]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enumeration_in_tree[TraceTreeEvaluator]
0.00s teardown tests/infer/test_valid_models.py::test_dim_allocation_error[TraceGraph_ELBO-False]
0.00s teardown tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[True-TraceGraph_ELBO]
0.00s teardown tests/infer/test_jit.py::test_bernoulli_enumerate[shape0-False]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_reuse[3-2]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_iplate_in_plate[4-5]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[2.0-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[2.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.7-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.1--0.5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape4]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape0]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_flat_exact_1_1[0.0-1.0]
0.00s teardown tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[minipyro-py]
0.00s teardown tests/contrib/test_minipyro.py::test_nested_plate_plate_ok[pyro-jit]
0.00s teardown tests/test_generic.py::test_model_sample[minipyro-neals_funnel]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1327]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1039]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges197]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape0-sample_shape2]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_distribution[2-()]
0.00s teardown tests/nn/test_module.py::test_constraints[shape22-constraint_22]
0.00s teardown tests/nn/test_module.py::test_constraints[shape14-constraint_14]
0.00s teardown tests/infer/reparam/test_transform.py::test_log_normal[()]
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_reentrant_history[2]
0.00s teardown tests/infer/test_jit.py::test_cond_indep_equality[x1-y1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape3]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape2]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape1-gate_shape1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape1]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.7--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.000001-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.999999-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.999999-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.8-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.8--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.01--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.4-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.1--1.0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape4]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape1]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence_type[False-p_mask2]
0.00s teardown tests/distributions/test_categorical.py::test_support_dims[dim=2-probs0]
0.00s teardown tests/test_generic.py::test_model_sample[pyro-logistic_regression]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1402]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1385]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1140]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1080]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges138]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...]-(2,)]
0.00s teardown tests/ops/test_gaussian.py::test_add[1-()]
0.00s teardown tests/nn/test_module.py::test_constraints[shape19-constraint_19]
0.00s teardown tests/nn/test_module.py::test_constraints[shape4-constraint_4]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape3-gate_shape2]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape0-gate_shape2]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[2.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[2.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.99--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.7-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.7--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.01--1.0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape3-sample_shape5]
0.00s teardown tests/test_generic.py::test_trace_handler[pyro-neals_funnel]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1196]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1149]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1098]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges933]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges843]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges800]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges266]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[replay-normal_normal_model]
0.00s teardown tests/poutine/test_mapdata.py::test_replay[nested_iplate-20]
0.00s teardown tests/poutine/test_mapdata.py::test_cond_indep_stack[nested_iplate-20]
0.00s teardown tests/ops/test_gaussian.py::test_add[2-()]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[,ia,ija,ika->,i,j,k,ij,ik,ijk,ia,ija,ika,ijka-ijk]
0.00s teardown tests/nn/test_module.py::test_constraints[shape10-constraint_10]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::ExponentialGammaTests::test_elbo_reparameterized
0.00s teardown tests/infer/test_jit.py::test_one_hot_categorical_enumerate[shape2-True]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zid_shape[base_shape2-gate_shape0]
0.00s teardown tests/distributions/test_unit.py::test_expand[batch_shape0-sample_shape2]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.3-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.8-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.4-1.0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape4-sample_shape4]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape4]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1216]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges998]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges198]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[replay-bernoulli_normal_model]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[wi,wxi,xyi,yzi->i]
0.00s teardown tests/infer/test_enum.py::test_avoid_nan[None]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-2]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.01--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.000001-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.999999--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[2.0-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.7-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.3--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.01-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.0--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.8--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.4--0.5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape0]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_deepGP
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_overdispersed_bound[0.1-1.0]
0.00s teardown tests/test_generic.py::test_trace_handler[pyro-eight_schools]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1316]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1170]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1043]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges868]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges733]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges727]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges564]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges112]
0.00s teardown tests/ops/test_gamma_gaussian.py::test_add[1-()]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[2]
0.00s teardown tests/infer/test_jit.py::test_categorical_enumerate[shape0-False]
0.00s teardown tests/infer/test_jit.py::test_bernoulli_enumerate[shape1-False]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.7-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.3-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.3-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.01-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.3--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.99--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.99--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.8-1.0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape1-sample_shape1]
0.00s teardown tests/distributions/test_categorical.py::test_support_dims[dim=3-probs0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1277]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1151]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1049]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges781]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges753]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges738]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges177]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges162]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges149]
0.00s teardown tests/nn/test_module.py::test_constraints[shape11-constraint_11]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gaussian_mixture_model[JIT=True]
0.00s teardown tests/infer/test_valid_models.py::test_dim_allocation_error[TraceEnum_ELBO-True]
0.00s teardown tests/infer/test_jit.py::test_categorical_enumerate[shape2-True]
0.00s teardown tests/infer/test_jit.py::test_bernoulli_enumerate[shape1-True]
0.00s teardown tests/distributions/test_unit.py::test_expand[batch_shape0-sample_shape1]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.99-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.99--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.4--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.1--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.7--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.0-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.0--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.8--1.0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1348]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1324]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1094]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1023]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges989]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges980]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges833]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges815]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges799]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges743]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges723]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges592]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges565]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges388]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges310]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges192]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_distribution[1-(3, 2)]
0.00s teardown tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-TraceGraph_ELBO]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model[5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.3-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.01-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.01--1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.4-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.4-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.1-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.0-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.8-0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.4-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.4--1.0]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape5]
0.00s teardown tests/distributions/test_rejector.py::test_rejection_exponential_sample_shape[batch_shape0-sample_shape2]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1285]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1282]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1131]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1122]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1111]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges859]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges791]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges789]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges350]
0.00s call     tests/distributions/test_distributions.py::test_distribution_validate_args[False-Normal-args0]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges531]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[block-bernoulli_normal_model]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[wxi,xyi,yzi,zwi->i]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_all_discrete_sites_log_prob[TraceTreeEvaluator]
0.00s teardown tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[TraceEnum_ELBO]
0.00s teardown tests/infer/test_jit.py::test_categorical_enumerate[shape1-False]
0.00s teardown tests/infer/test_jit.py::test_bernoulli_enumerate[shape0-True]
0.00s teardown tests/infer/test_enum.py::test_vectorized_importance[10000]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[5]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.3-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.1-1.0]
0.00s teardown tests/distributions/test_categorical.py::test_sample_dims[dim=1-probs0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1201]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1085]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1062]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1036]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges985]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges984]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges982]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges876]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges822]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges766]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges413]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges408]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges340]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges203]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges169]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges365]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[trace-bernoulli_normal_model]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[block-normal_normal_model]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[block-trivial_model]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-ij->ij]
0.00s teardown tests/nn/test_module.py::test_constraints[shape3-constraint_3]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_bernoulli_latent_model[JIT=True]
0.00s teardown tests/infer/test_jit.py::test_bernoulli_enumerate[shape2-True]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-5]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.01-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.99-0.0]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_categorical[2]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1318]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1174]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1093]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1064]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges776]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges765]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges534]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges508]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges324]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges303]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges244]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges120]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges217]
0.00s teardown tests/poutine/test_mapdata.py::test_cond_indep_stack[nested_iplate-5]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_shape[map-i->i]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-Trace_ELBO]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[21]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model[3]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-4]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-3]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-2]
0.00s teardown tests/distributions/test_unit.py::test_expand[batch_shape1-sample_shape2]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.8-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.4-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.1-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.7-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[1.3-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.99-0.5]
0.00s teardown tests/contrib/tracking/test_em.py::test_em[True]
0.00s teardown tests/contrib/oed/test_xexpx.py::test_xexpx[argument2-output2]
0.00s teardown tests/contrib/oed/test_xexpx.py::test_xexpx[argument0-output0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1347]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1267]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1172]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1159]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1156]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1146]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1128]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1038]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges940]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges907]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges854]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges827]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges823]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges807]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges702]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges681]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges637]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges362]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges331]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges308]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges164]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges161]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges77]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges5]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape0-sample_shape0]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_all_discrete_sites_log_prob[TraceEinsumEvaluator]
0.00s teardown tests/infer/test_jit.py::test_svi[JitTrace_ELBO-1]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[1.3--0.5]
0.00s teardown tests/distributions/test_stable.py::test_sample_2[0.8-1.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[2.0--1.0]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[5]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1429]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1427]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1406]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1390]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1375]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1355]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1337]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1303]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1288]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1275]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1185]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1177]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1141]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges981]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges938]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges927]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges926]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges870]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges809]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges797]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges779]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges751]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges750]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges722]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges720]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges602]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges581]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges552]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges341]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges282]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges175]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1376]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[ab->,a,b,ab,ba-]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_subsample_param_ok[TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_not_has_rsample_ok[TraceTMC_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[False-TraceGraph_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_sample[2.0--0.5]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1336]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1331]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1322]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1319]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1283]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1192]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1183]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1123]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1105]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1097]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1056]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1037]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges979]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges971]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges925]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges902]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges897]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges836]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges788]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges782]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges769]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges740]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges730]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges692]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges629]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges547]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges522]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges442]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges434]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges353]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges292]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges283]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges234]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges213]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges208]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges193]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges187]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges146]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[replay-normal_model]
0.00s teardown tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[2-True]
0.00s teardown tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[2-False]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[True-A1]
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_reparameterized_three_layers
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_mean at 0x7f6d86b7e8c8>1]
0.00s teardown tests/distributions/test_stable.py::test_sample[2.0-0.0]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.1-0.0]
0.00s teardown tests/distributions/test_reshape.py::test_reshape[1-2]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1403]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1393]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1342]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1304]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1214]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1207]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1157]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1147]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1139]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1126]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1124]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1102]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1082]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1034]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1003]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges969]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges954]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges928]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges912]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges900]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges887]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges830]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges761]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges678]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges666]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges638]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges624]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges511]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges491]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges455]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges436]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges391]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges373]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges313]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges261]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges246]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges87]
0.00s teardown tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[1-True]
0.00s teardown tests/ops/test_gaussian.py::test_rsample_distribution[1-(4,)]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_dirichlet_categorical[JIT=False]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-3]
0.00s teardown tests/distributions/test_unit.py::test_expand[batch_shape0-sample_shape0]
0.00s teardown tests/distributions/test_reshape.py::test_idempotent[2-0]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1409]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1329]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1306]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1300]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1263]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1213]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1203]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1188]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1136]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1069]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1013]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1006]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1001]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges992]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges990]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges959]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges890]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges832]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges814]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges794]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges778]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges770]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges768]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges757]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges755]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges725]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges712]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges707]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges640]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges537]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges469]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges465]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges450]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges447]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges439]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges429]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges416]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges364]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges323]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges318]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges306]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges298]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges293]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges288]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges262]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges257]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges240]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges190]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges176]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges172]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges171]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges152]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges151]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges148]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges33]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges17]
0.00s teardown tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[1-False]
0.00s teardown tests/poutine/test_poutines.py::test_plate_preserves_has_rsample[0-True]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[29]
0.00s teardown tests/ops/test_contract.py::test_ubersum_jit[->-]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gamma_beta[JIT=False]
0.00s teardown tests/infer/test_valid_models.py::test_empty_model_empty_guide_ok[False-Trace_ELBO]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model[4]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_order[dense-1]
0.00s teardown tests/distributions/test_spanning_tree.py::test_sample_tree_mcmc_smoke[python-1]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[4]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1424]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1394]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1380]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1357]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1321]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1278]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1270]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1262]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1252]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1219]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1195]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1165]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1144]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1143]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1142]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1066]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1059]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1040]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1033]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1021]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1016]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1010]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1000]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges923]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges917]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges889]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges853]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges850]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges844]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges802]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges754]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges703]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges686]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges596]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges573]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges524]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges517]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges501]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges490]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges480]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges476]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges440]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges419]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges371]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges368]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges344]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges329]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges290]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges280]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges277]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges264]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges247]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges229]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges206]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges179]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges167]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges154]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges139]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges133]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges123]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges119]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges82]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges54]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges34]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges14]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges13]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges994]
0.00s teardown tests/poutine/test_properties.py::test_idempotent[trace-trivial_model]
0.00s teardown tests/optim/test_multi.py::test_optimizers[<lambda>2]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[3]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[False-A2]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_bernoulli_latent_model[JIT=False]
0.00s teardown tests/infer/test_valid_models.py::test_plate_no_size_ok[TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_nonempty_model_empty_guide_error[True-Trace_ELBO]
0.00s teardown tests/distributions/test_stable.py::test_sample[0.1-0.5]
0.00s teardown tests/distributions/test_reshape.py::test_idempotent[1-2]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_categorical[7]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1426]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1411]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1398]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1384]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1373]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1359]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1340]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1291]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1249]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1224]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1182]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1178]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1138]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1121]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1110]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1108]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1072]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1067]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1017]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges975]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges962]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges922]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges921]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges920]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges884]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges882]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges875]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges871]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges848]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges771]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges759]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges744]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges737]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges709]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges660]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges650]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges639]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges620]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges571]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges542]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges516]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges475]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges472]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges470]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges460]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges339]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges287]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges254]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges245]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges220]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges211]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges185]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges174]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges135]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges29]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges0]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2,3)]-(2, 3)]
0.00s teardown tests/ops/test_contract.py::test_ubersum_collide_implemented[ubersum-False]
0.00s teardown tests/distributions/test_reshape.py::test_idempotent[0-0]
0.00s teardown tests/contrib/oed/test_xexpx.py::test_xexpx[argument1-output1]
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_amortized_smoke[init_to_mean]
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_delta_smoke[init_to_mean]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1332]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1314]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1301]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1273]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1259]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1234]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1229]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1226]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1223]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1217]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1179]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1164]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1115]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1070]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1061]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1030]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1025]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1018]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges972]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges964]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges904]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges896]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges856]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges810]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges804]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges763]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges724]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges717]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges665]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges661]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges632]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges608]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges599]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges594]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges570]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges509]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges488]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges409]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges360]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges349]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges337]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges300]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges274]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges272]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges249]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges224]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges180]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges143]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges141]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges134]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges114]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges107]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges98]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges89]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Normal]
0.00s call     tests/pyroapi/test_pyroapi.py::test_optimizer[minipyro-ClippedAdam-optim_kwargs1-jit]
0.00s teardown tests/ops/test_linalg.py::test_sym_rinverse[False-A1]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4)]-(4, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[0,:]-(3,)]
0.00s teardown tests/ops/test_contract.py::test_ubersum[->-]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[6]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_categorical[5]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_categorical[4]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_sparse_grid_smoke
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1434]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1421]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1417]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1404]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1383]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1381]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1362]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1326]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1320]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1257]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1198]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1167]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1160]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1152]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1133]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1118]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1109]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1044]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1022]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1011]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges995]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges951]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges931]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges918]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges891]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges881]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges845]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges841]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges835]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges746]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges704]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges701]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges697]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges676]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges671]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges656]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges647]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges623]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges607]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges513]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges499]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges498]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges496]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges493]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges473]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges468]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges463]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges452]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges400]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges383]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges334]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges321]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges296]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges255]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges188]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges170]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges124]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges121]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges102]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges85]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges66]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(2)]-(2,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0]-(2,)]
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_reentrant_history[1]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_order[flat-1]
0.00s teardown tests/infer/test_autoguide.py::test_scores[AutoMultivariateNormal]
0.00s teardown tests/distributions/test_reshape.py::test_idempotent[0-2]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1431]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1419]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1370]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1364]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1352]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1296]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1268]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1242]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1237]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1210]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1205]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1190]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1169]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1162]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1090]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1079]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1051]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1031]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges953]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges946]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges943]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges936]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges930]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges879]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges877]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges849]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges820]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges813]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges805]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges670]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges663]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges589]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges585]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges568]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges548]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges494]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges492]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges478]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges458]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges454]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges426]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges404]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges376]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges375]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges366]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges365]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges316]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges275]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges256]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges251]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges248]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges236]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges166]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges57]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.5-0.6-0.4]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges870]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixj->ij]
0.00s teardown tests/ops/test_integrator.py::test_trajectory[QuarticOscillator]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:,z(4)]-(4, 2)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2))[:]-(2,)]
0.00s teardown tests/ops/test_contract.py::test_naive_ubersum[->-]
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_reparameterized_N_is_8
0.00s teardown tests/distributions/test_pickle.py::test_pickle[DiscreteHMM]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1422]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1416]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1401]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1391]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1349]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1293]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1286]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1272]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1254]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1246]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1231]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1228]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1206]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1150]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1129]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1106]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1103]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1087]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1007]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1005]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges996]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges977]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges910]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges899]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges874]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges866]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges818]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges787]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges767]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges742]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges696]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges691]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges653]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges645]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges642]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges634]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges616]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges614]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges580]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges563]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges555]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges536]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges535]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges464]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges449]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges406]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges385]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges346]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges326]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges305]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges278]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges269]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges238]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges205]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges202]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges157]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges136]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges128]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges116]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges113]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges100]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges92]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges76]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[27]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[22]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[12]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_categorical[6]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1437]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1369]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1360]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1354]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1343]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1339]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1334]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1313]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1311]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1247]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1154]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1135]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1046]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges991]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges976]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges974]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges961]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges955]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges944]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges894]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges869]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges863]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges749]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges728]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges719]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges699]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges684]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges683]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges628]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges618]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges612]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges606]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges601]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges598]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges566]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges553]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges527]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges519]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges514]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges507]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges485]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges445]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges444]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges390]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges347]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges332]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges159]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges144]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges80]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges62]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges43]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape0-sample_shape1]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape0-sample_shape0]
0.00s teardown tests/optim/test_multi.py::test_optimizers[<lambda>4]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:,0]-(2,)]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_feasible at 0x7f6d86b7e730>1]
0.00s teardown tests/infer/test_valid_models.py::test_no_log_prob_ok[EnergyDistance_noprior]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1430]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1412]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1280]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1265]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1251]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1250]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1236]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1232]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1211]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges994]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges915]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges792]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges658]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges648]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges644]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges622]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges611]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges584]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges540]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges532]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges530]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges529]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges481]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges432]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges427]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges424]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges421]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges411]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges392]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges387]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges382]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges377]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges370]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges352]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges342]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges285]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges270]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges265]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges228]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges216]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges140]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges130]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges115]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges103]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges94]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges90]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges88]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges59]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges28]
0.00s teardown tests/optim/test_multi.py::test_optimizers[<lambda>3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[98]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5),z(5)]-(5,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),0,:]-(5, 4)]
0.00s teardown tests/distributions/test_reshape.py::test_idempotent[1-0]
0.00s teardown tests/distributions/test_reshape.py::test_idempotent[0-1]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1244]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1161]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges958]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges935]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges886]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges774]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges748]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges710]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges694]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges679]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges655]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges652]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges635]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges597]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges558]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges457]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges418]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges395]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges386]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges372]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges359]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges314]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges218]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges118]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges110]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges108]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges71]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges22]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2),0]-(2,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:,:]-(2, 3)]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[19]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_4[10]
0.00s teardown tests/distributions/test_reshape.py::test_idempotent[2-2]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Categorical]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1408]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1372]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1345]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1295]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1255]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1227]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1083]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges948]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges824]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges673]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges619]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges583]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges562]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges506]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges503]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges355]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges319]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges242]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges219]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges215]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges79]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges38]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges21]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape0-sample_shape1]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1300]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[wx,xy,yz,zw->]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5),:]-(5, 2, 4)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),0,0]-(5,)]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_enumeration_in_tree[TraceEinsumEvaluator]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_sample at 0x7f6d86b7e7b8>1]
0.00s teardown tests/infer/test_valid_models.py::test_plate_no_size_ok[TraceTMC_ELBO]
0.00s teardown tests/distributions/test_conjugate.py::test_mean[dist4]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1397]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1389]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1378]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1368]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1238]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1074]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1055]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1050]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1019]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1014]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges775]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges693]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges680]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges643]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges550]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges512]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges486]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges398]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges295]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges259]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges191]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges72]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges8]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(6),:,z(5,1)]-(5, 6, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(6),0,z(5,1)]-(5, 6)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),0,z(5)]-(5,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(6),z(5,1)]-(5, 6, 2)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,0,z(5)]-(5, 2)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[0,0]-()]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1435]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1399]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1193]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges949]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges905]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges752]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges731]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges729]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges715]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges674]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges627]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges575]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges557]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges345]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges221]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges74]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges69]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges67]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges63]
0.00s teardown tests/ops/test_indexing.py::test_index[0-0]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),:,z(5)]-(5, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5),z(5)]-(5, 2)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5,1),z(6)]-(5, 6)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(5,1),z(4)]-(5, 4)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[:,0]-(2,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),0]-(4,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...,0]-()]
0.00s teardown tests/ops/test_contract.py::test_ubersum_5[ubersum]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[20]
0.00s teardown tests/infer/test_enum.py::test_avoid_nan[parallel]
0.00s teardown tests/distributions/test_reshape.py::test_idempotent[1-1]
0.00s teardown tests/distributions/test_lkj.py::test_log_prob_eta1[2]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1439]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1374]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1356]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1194]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1191]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1145]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1130]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1120]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges929]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges858]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges801]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges689]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges675]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges586]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges431]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges307]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges250]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges186]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges173]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges95]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges86]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes0]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-2-batch_shape0-sample_shape2]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4-1.0]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5),:]-(5, 4)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),:,:]-(5, 3, 4)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5,1),:]-(5, 2, 4)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(2,1)]-(2, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),0]-(4,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(()))[...]-()]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_3[10]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1400]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1189]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1171]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1155]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1092]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1091]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1008]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges970]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges965]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges880]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges834]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges777]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges735]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges609]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges479]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges459]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges446]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges356]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges223]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges111]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes7]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5,1),0,z(6)]-(5, 6)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5),:,0]-(5, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5,1),0]-(5, 2)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(5,1),z(4)]-(5, 4)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4,1)]-(4, 2)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,:]-(2, 3)]
0.00s teardown tests/infer/test_autoguide.py::test_discrete_parallel[AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Multinomial]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1113]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1112]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1077]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1073]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1029]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges911]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges839]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges785]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges695]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges456]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges438]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges399]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges83]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges64]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_error[shapes1]
0.00s teardown tests/optim/test_multi.py::test_optimizers[<lambda>1]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[95]
0.00s teardown tests/ops/test_special.py::test_log_binomial_stirling[1.0]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5),0]-(5, 2)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,:,z(5)]-(5, 2, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,z(5),0]-(5,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,:,z(5)]-(5, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(3)]-(2, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),z(4)]-(4,)]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[1]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_aps_hash[1]
0.00s teardown tests/contrib/tracking/test_em.py::test_em[False]
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_delta_smoke[init_to_median]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1405]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1371]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1361]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1240]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1199]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1173]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1114]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1089]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1086]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1047]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1035]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges963]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges783]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges739]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges718]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges572]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges428]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges378]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges210]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges160]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges27]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes8]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixyj,iyzj->ji]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[:,z(5,1),z(6)]-(5, 6, 2)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),:]-(4, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[:,z(4)]-(4, 2)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),:]-(4, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2))[0]-()]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...,z(3)]-(3,)]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_gamma_beta[JIT=True]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[10]
0.00s teardown tests/infer/test_autoguide.py::test_discrete_parallel[AutoMultivariateNormal]
0.00s teardown tests/distributions/test_conjugate.py::test_variance[dist5]
0.00s teardown tests/distributions/test_conjugate.py::test_mean[dist5]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1341]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1333]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1325]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1186]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1176]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1137]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1104]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1071]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1058]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1032]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges966]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges942]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges837]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges803]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges798]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges795]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges780]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges734]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges713]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges526]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges520]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges502]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges407]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges397]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges338]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges271]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges258]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges207]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges189]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges184]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges168]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges147]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges126]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges37]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes3]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape0-sample_shape2]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwxj,ixyj,iyzj,izwj->ji]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5,1),:,z(6)]-(5, 6, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),0,:]-(5, 4)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...,z(5,1,1)]-(5, 2, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[0,z(4)]-(4,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,z(4),z(5,1)]-(5, 4)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0,z(4)]-(4,)]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plates_dependency_ok[sequential]
0.00s teardown tests/infer/test_autoguide.py::test_callable[AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_mvt.py::test_mean_var[()]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[ShapeAugmentedGamma]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1366]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1305]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1299]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1276]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1158]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1125]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1107]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1099]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1068]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1048]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges885]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges878]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges867]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges826]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges736]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges659]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges567]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges546]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges336]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges330]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges252]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges209]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges194]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges163]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges158]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges65]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges40]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges12]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges3]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-3-2-7]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),:,0]-(5, 3)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),z(4)]-(4,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2))[...,:]-(2,)]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[24]
0.00s teardown tests/infer/test_autoguide.py::test_init_loc_fn[AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_bernoulli_shapes[probs1]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1433]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1396]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1376]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1317]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1312]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1307]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1269]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1261]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1260]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1256]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1132]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1078]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1042]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1009]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges973]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges960]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges950]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges862]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges857]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges852]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges831]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges649]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges641]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges523]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges482]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges466]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges384]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges284]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges281]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges263]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges178]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges145]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges132]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges122]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges105]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges99]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges96]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges53]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges35]
0.00s call     tests/contrib/gp/test_models.py::test_inference[VariationalSparseGP_y2D]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwxj,ixyj,iyzj->ij]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[z(5),0,0]-(5,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...,0,0]-()]
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_grid[True]
0.00s teardown tests/infer/test_autoguide.py::test_discrete_parallel[AutoDelta]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1420]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1395]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1377]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1315]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1289]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1274]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1258]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1218]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1209]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1181]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1134]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1119]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges999]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges978]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges901]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges888]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges821]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges816]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges793]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges741]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges711]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges690]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges615]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges605]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges556]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges462]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges443]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges348]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges196]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges106]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges42]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges39]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges31]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape0-sample_shape0]
0.00s teardown tests/ops/test_indexing.py::test_index[args9-expected9]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[0,0,z(5)]-(5,)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[z(4),z(5,1)]-(5, 4)]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2))[z(3)]-(3,)]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_plate_ok[TraceGraph_ELBO]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[14]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[11]
0.00s teardown tests/distributions/test_polya_gamma.py::test_polya_gamma[batch_shape2]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[ZeroInflatedPoisson]
0.00s teardown tests/distributions/test_hmm.py::test_gaussian_hmm_high_obs_dim
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[7]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1415]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1413]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1379]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1353]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1302]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1292]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1279]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1248]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1117]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1060]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges993]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges987]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges914]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges840]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges829]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges772]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges762]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges603]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges600]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges551]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges549]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges510]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges483]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges451]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges405]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges394]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges389]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges351]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges343]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges325]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges297]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges279]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges231]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges183]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges155]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges127]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges73]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges61]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges48]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges19]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges4]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-1-batch_shape0-sample_shape1]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape0-sample_shape0]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_dirichlet_categorical[JIT=True]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_ok[Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plates_dependency_ok[None]
0.00s teardown tests/infer/test_valid_models.py::test_plate_reuse_ok[Trace_ELBO]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1425]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1382]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1367]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1344]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1338]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1323]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1310]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1281]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1253]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1245]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1225]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1220]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1168]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1081]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1063]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1045]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1024]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1004]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges986]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges952]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges842]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges806]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges790]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges747]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges721]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges714]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges708]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges705]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges569]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges489]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges412]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges410]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges312]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges204]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges181]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges153]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges142]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges55]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges16]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges15]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape0-sample_shape0]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[MultivariateStudentT]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwxj,ixyj,iyzj,izwj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[wi->i]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[:,:]-(2, 3)]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_plates_dependency_ok[parallel]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_hmm[4]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1438]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1351]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1335]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1294]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1284]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1266]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1235]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1184]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1065]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1053]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges939]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges937]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges893]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges872]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges855]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges819]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges646]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges626]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges515]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges487]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges461]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges433]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges425]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges374]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges358]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges327]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges304]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges291]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges289]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges268]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges253]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges235]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges201]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges137]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges75]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges51]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges45]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges36]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges32]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges30]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges24]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges10]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges6]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges2]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_non_enumerated_plate_ok[parallel]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_non_enumerated_plate_ok[None]
0.00s teardown tests/infer/test_autoguide.py::test_callable[AutoNormal]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1407]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1392]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1387]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1386]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1330]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1328]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1297]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1271]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1233]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1202]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1197]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1076]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges909]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges698]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges677]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges654]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges630]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges613]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges587]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges561]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges559]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges554]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges544]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges538]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges533]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges500]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges495]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges415]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges299]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges276]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges225]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges117]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges81]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges56]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges49]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges44]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges41]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges18]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges7]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes6]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes1]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape0-sample_shape0]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0-0.5]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges876]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwj,ixj->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[w,wx,x->]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct[shape2]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct[shape0]
0.00s teardown tests/ops/test_special.py::test_log_binomial_stirling[0.02]
0.00s teardown tests/ops/test_special.py::test_log_beta_stirling[0.11]
0.00s teardown tests/ops/test_indexing.py::test_index[args7-expected7]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3))[...]-(2, 3)]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::RaoBlackwellizationTests::test_nested_iplate_in_elbo
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[23]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[17]
0.00s teardown tests/infer/test_csis.py::test_csis_sampling
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Gamma]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1428]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1346]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1309]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1230]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1215]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges932]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges916]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges808]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges688]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges662]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges631]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges595]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges590]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges582]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges579]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges574]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges504]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges477]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges453]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges402]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges317]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges302]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges243]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges230]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges227]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges217]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges93]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges84]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges58]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges46]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges25]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges20]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes4]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape0-sample_shape0]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-3-batch_shape0-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-1-batch_shape0-sample_shape1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges70]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[wi,xi->i]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[18]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[4]
0.00s teardown tests/contrib/autoguide/test_inference.py::test_auto_transform[AutoDiagonalNormal]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1365]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges919]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges898]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges873]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges811]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges716]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges682]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges664]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges657]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges588]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges474]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges379]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges363]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges322]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges101]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges50]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges11]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes5]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape0-sample_shape1]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[FoldedNormal]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[wi,wxi,xi->i]
0.00s teardown tests/ops/test_indexing.py::test_index[args5-expected5]
0.00s teardown tests/ops/test_indexing.py::test_shape[Vindex(z(2,3,4))[...]-(2, 3, 4)]
0.00s teardown tests/infer/mcmc/test_adaptation.py::test_arrowhead_mass_matrix[False]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[10]
0.00s teardown tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTrace_ELBO]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[7]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1287]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1204]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges957]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges945]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges687]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges621]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges577]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges541]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges420]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges417]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges381]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges70]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges68]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes2]
0.00s teardown tests/ops/test_tensor_utils.py::test_idct[shape0]
0.00s teardown tests/infer/test_valid_models.py::test_plate_enum_discrete_no_discrete_vars_warning[True]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[13]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_hmm[2]
0.00s teardown tests/infer/test_autoguide.py::test_nested_autoguide[TraceEnum_ELBO]
0.00s teardown tests/distributions/test_lkj.py::test_log_prob_d2[5.0]
0.00s teardown tests/distributions/test_conjugate.py::test_variance[dist4]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges543]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges441]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges361]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges357]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges237]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges232]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges222]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges214]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges212]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges109]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges9]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-3-batch_shape0-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape0-sample_shape1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges806]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwj->ji]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[46]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[23]
0.00s teardown tests/ops/test_special.py::test_log_binomial_stirling[0.05]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[badc]
0.00s teardown tests/infer/test_valid_models.py::test_vectorized_num_particles[TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_callable_return_dict[AutoNormal]
0.00s teardown tests/infer/test_autoguide.py::test_guide_list[AutoNormal]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_bernoulli_shapes[probs0]
0.00s teardown tests/distributions/test_lkj.py::test_log_prob_d2[2.0]
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_delta_smoke[None]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges1222]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges883]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges865]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges633]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges525]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges309]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges129]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-2-batch_shape0-sample_shape2]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape0-sample_shape0]
0.00s call     tests/contrib/tracking/test_hashing.py::test_aps_init[-1.0]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[w,wx,xy,yz->]
0.00s teardown tests/ops/test_special.py::test_log_binomial_stirling[0.11]
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_mutual_recursion
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[9]
0.00s teardown tests/distributions/test_lkj.py::test_log_prob_d2[1.0]
0.00s teardown tests/distributions/test_lkj.py::test_log_prob_d2[0.5]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ExtendedBinomial]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges669]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges528]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges52]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges23]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape0-sample_shape2]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Cauchy]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-1-1-1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges193]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwj,ixj->ij]
0.00s teardown tests/ops/test_tensor_utils.py::test_idct[shape2]
0.00s teardown tests/ops/test_special.py::test_log_binomial_stirling[0.2]
0.00s teardown tests/ops/test_special.py::test_log_binomial_stirling[0.01]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[2]
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_generalized_channel_permute
0.00s teardown tests/distributions/test_lkj.py::test_log_prob_eta1[10]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_categorical[1]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[MultivariateNormal]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape0-sample_shape1]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[ji,iwj->ji]
0.00s teardown tests/ops/test_special.py::test_log_beta_stirling[1.0]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[bcda]
0.00s teardown tests/ops/test_indexing.py::test_index[args3-expected3]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<pyro.infer.autoguide.initialization._InitToGenerated object at 0x7f6d757450f0>]
0.00s teardown tests/infer/test_enum.py::test_svi_step_guide_uses_grad[sequential]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs0]
0.00s teardown tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[1]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_categorical[8]
0.00s call     tests/poutine/test_trace_struct.py::test_topological_sort[edges315]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape0-sample_shape1]
0.00s call     tests/distributions/test_independent.py::test_independent[normal-2-batch_shape0-sample_shape2]
0.00s teardown tests/ops/test_indexing.py::test_index[args6-expected6]
0.00s teardown tests/ops/test_indexing.py::test_index[None-None]
0.00s teardown tests/infer/test_valid_models.py::test_variable_clash_in_model_error[Trace_ELBO]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[16]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[5]
0.00s teardown tests/infer/test_sampling.py::ImportanceTest::test_importance_prior
0.00s teardown tests/distributions/test_spanning_tree.py::test_log_prob[5]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs2]
0.00s teardown tests/distributions/test_lkj.py::test_log_prob_eta1[4]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[Bernoulli]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[LowRankMultivariateNormal]
0.00s call     tests/contrib/tracking/test_hashing.py::test_lsh_init[0.0]
0.00s teardown tests/ops/test_indexing.py::test_index[args8-expected8]
0.00s teardown tests/ops/test_indexing.py::test_index[args4-expected4]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[8]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[6]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[3]
0.00s teardown tests/infer/test_autoguide.py::test_guide_list[AutoDiagonalNormal]
0.00s teardown tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape4]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_categorical[9]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_log_prob[3]
0.00s teardown tests/contrib/timeseries/test_gp.py::test_dependent_matern_gp[3]
0.00s call     tests/distributions/test_independent.py::test_independent[mvn-3-batch_shape0-sample_shape1]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[55]
0.00s teardown tests/ops/test_special.py::test_log_binomial_stirling[1e-08]
0.00s teardown tests/ops/test_indexing.py::test_index[1-1]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model_and_guide[2]
0.00s teardown tests/distributions/test_von_mises.py::test_von_mises_3d[10.0]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[3]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_missing_config_warning[False]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[15]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[7]
0.00s teardown tests/infer/test_jit.py::test_cond_indep_equality[x0-y0]
0.00s teardown tests/infer/test_autoguide.py::test_guide_list[AutoMultivariateNormal]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs3]
0.00s teardown tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape1]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[8]
0.00s call     tests/distributions/test_independent.py::test_independent[exponential-1-batch_shape0-sample_shape2]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_model_with_subsample[10]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[inf]
0.00s teardown tests/infer/test_smcfilter.py::test_systematic_sample[4]
0.00s teardown tests/infer/test_jit.py::test_traceenum_elbo[2]
0.00s teardown tests/distributions/test_von_mises.py::test_von_mises_3d[0.1]
0.00s teardown tests/distributions/test_spanning_tree.py::test_log_prob[4]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[IndependentHMM]
0.00s teardown tests/distributions/test_lkj.py::test_log_prob_d2[0.1]
0.00s teardown tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape3]
0.00s call     tests/contrib/tracking/test_hashing.py::test_aps_init[0.0]
0.00s teardown tests/ops/test_special.py::test_log_binomial_stirling[1e-06]
0.00s teardown tests/ops/test_special.py::test_log_beta_stirling[0.2]
0.00s teardown tests/infer/test_valid_models.py::test_markov_history[0]
0.00s teardown tests/infer/test_valid_models.py::test_plate_enum_discrete_no_discrete_vars_warning[False]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_6[sequential]
0.00s teardown tests/infer/test_enum.py::test_svi_step_guide_uses_grad[None]
0.00s teardown tests/infer/test_autoguide.py::test_callable_return_dict[AutoLaplaceApproximation]
0.00s teardown tests/infer/test_autoguide.py::test_callable_return_dict[AutoLowRankMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_callable_return_dict[AutoDiagonalNormal]
0.00s teardown tests/distributions/test_von_mises.py::test_log_prob_normalized[1.0]
0.00s teardown tests/distributions/test_spanning_tree.py::test_log_prob[2]
0.00s teardown tests/distributions/test_spanning_tree.py::test_partition_function[4]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3--0.5]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1304]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[bca]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_missing_config_warning[True]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_7[1]
0.00s teardown tests/distributions/test_von_mises.py::test_von_mises_3d[1.0]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[OneHotCategorical]
0.00s teardown tests/ops/test_contract.py::test_contract_to_tensor[example6]
0.00s teardown tests/ops/test_contract.py::test_contract_to_tensor[example4]
0.00s teardown tests/ops/test_arrowhead.py::test_utilities[2]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_median at 0x7f6d86b7e840>]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_1[10]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[abc]
0.00s teardown tests/ops/test_contract.py::test_ubersum_3[naive_ubersum]
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_grid[False]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[0]
0.00s teardown tests/infer/test_autoguide.py::test_guide_list[AutoLaplaceApproximation]
0.00s teardown tests/infer/test_autoguide.py::test_guide_list[AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_von_mises.py::test_log_prob_normalized[30.0]
0.00s teardown tests/distributions/test_von_mises.py::test_log_prob_normalized[0.3]
0.00s teardown tests/distributions/test_unit.py::test_shapes[batch_shape0]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[9]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[MultivariateNormal]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_svi_custom_smoke[False]
0.00s teardown tests/ops/test_special.py::test_log_binomial_stirling[0.0001]
0.00s teardown tests/infer/test_valid_models.py::test_not_has_rsample_ok[TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_callable[AutoLaplaceApproximation]
0.00s teardown tests/infer/test_autoguide.py::test_callable[AutoDiagonalNormal]
0.00s teardown tests/distributions/test_von_mises.py::test_log_prob_normalized[3.0]
0.00s teardown tests/distributions/test_spanning_tree.py::test_log_prob[3]
0.00s teardown tests/distributions/test_lkj.py::test_log_prob_eta1[3]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_diag_normal[2]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[OneHotCategorical]
0.00s teardown tests/distributions/test_categorical.py::TestCategorical::test_mean_and_var
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_smoke[bp]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7--0.5]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Stable]
0.00s teardown tests/ops/test_contract.py::test_ubersum_2[naive_ubersum]
0.00s teardown tests/infer/test_autoguide.py::test_callable_return_dict[AutoMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_callable_return_dict[AutoDelta]
0.00s teardown tests/distributions/test_von_mises.py::test_log_prob_normalized[10.0]
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_conditional_generalized_channel_permute
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[OneHotCategorical]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[DirichletMultinomial]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1030]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[abdc]
0.00s teardown tests/ops/test_contract.py::test_ubersum_1[naive_ubersum]
0.00s teardown tests/ops/test_contract.py::test_contract_to_tensor[example7]
0.00s teardown tests/ops/test_contract.py::test_contract_to_tensor[example0]
0.00s teardown tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data1]
0.00s teardown tests/infer/test_enum.py::test_svi_step_guide_uses_grad[parallel]
0.00s teardown tests/infer/test_autoguide.py::test_callable[AutoMultivariateNormal]
0.00s teardown tests/distributions/test_von_mises.py::test_von_mises_3d[1.1]
0.00s teardown tests/distributions/test_von_mises.py::test_von_mises_3d[0.9]
0.00s teardown tests/distributions/test_von_mises.py::test_log_prob_normalized[100.0]
0.00s teardown tests/distributions/test_von_mises.py::test_log_prob_normalized[0.1]
0.00s teardown tests/distributions/test_hmm.py::test_discrete_hmm_categorical[3]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_cat2_scalar[(2, 3)]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[cba]
0.00s teardown tests/ops/test_contract.py::test_ubersum_2[ubersum]
0.00s teardown tests/ops/test_contract.py::test_contract_to_tensor[example3]
0.00s teardown tests/ops/test_contract.py::test_contract_to_tensor[example2]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_1[1]
0.00s teardown tests/infer/test_autoguide.py::test_init_loc_fn[AutoMultivariateNormal]
0.00s teardown tests/infer/test_autoguide.py::test_guide_list[AutoDelta]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_shared_cov_log_prob[3]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[FoldedNormal]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[acbd]
0.00s teardown tests/ops/test_contract.py::test_ubersum_3[ubersum]
0.00s teardown tests/ops/test_contract.py::test_contract_tensor_tree[example4]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_feasible at 0x7f6d86b7e730>0]
0.00s teardown tests/infer/test_valid_models.py::test_markov_history[3]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_4[1]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_3[1]
0.00s teardown tests/infer/test_autoguide.py::test_callable[AutoDelta]
0.00s teardown tests/distributions/test_von_mises.py::test_log_prob_normalized[0.01]
0.00s teardown tests/distributions/test_lkj.py::test_constraint[value_shape0]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[Categorical]
0.00s teardown tests/contrib/forecast/test_forecaster.py::test_svi_custom_smoke[True]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges997]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[abcd]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[cab]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[acb]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[ab]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[a]
0.00s teardown tests/ops/test_contract.py::test_ubersum_4[naive_ubersum]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_mean at 0x7f6d86b7e8c8>0]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_7[1]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_6[10]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_5[10]
0.00s teardown tests/distributions/test_spanning_tree.py::test_partition_function[3]
0.00s teardown tests/distributions/test_spanning_tree.py::test_partition_function[1]
0.00s teardown tests/distributions/test_extended.py::test_extended_beta_binomial[0.1]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Categorical]
0.00s teardown tests/distributions/test_delta.py::test_expand[batch_shape0]
0.00s setup    tests/test_primitives.py::test_sample_ok
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0--1.0]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[LatentStableReparam-2.0-1.0]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[NaiveDirichlet]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[ba]
0.00s teardown tests/ops/test_contract.py::test_ubersum_4[ubersum]
0.00s teardown tests/ops/test_contract.py::test_contract_tensor_tree[example6]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_6[1]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Beta]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Categorical]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4-0.5]
0.00s teardown tests/ops/test_contract.py::test_contract_to_tensor[example1]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_3[10]
0.00s teardown tests/infer/test_enum.py::test_hmm_enumerate_model[2]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Categorical]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[MultivariateNormal]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[MultivariateNormal]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[bac]
0.00s teardown tests/ops/test_contract.py::test_ubersum_collide_ok_2[ubersum]
0.00s teardown tests/ops/test_contract.py::test_ubersum_collide_ok_2[naive_ubersum]
0.00s teardown tests/ops/test_contract.py::test_ubersum_collide_ok_1[ubersum]
0.00s teardown tests/ops/test_contract.py::test_contract_tensor_tree[example5]
0.00s teardown tests/ops/test_contract.py::test_contract_tensor_tree[example1]
0.00s teardown tests/ops/test_contract.py::test_contract_tensor_tree[example0]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_potential_fn_pickling[False]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_parallel_nested_ok[1]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_hmm[30]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_5[1]
0.00s teardown tests/distributions/test_spanning_tree.py::test_log_prob[1]
0.00s teardown tests/distributions/test_spanning_tree.py::test_partition_function[5]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[LowRankMultivariateNormal]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[MultivariateNormal]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99-1.0]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1-0.5]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[ShapeAugmentedDirichlet]
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_potential_fn_pickling[True]
0.00s teardown tests/infer/test_valid_models.py::test_not_has_rsample_ok[Trace_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_2[10]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_3[1]
0.00s teardown tests/infer/test_enum.py::test_avoid_nan[sequential]
0.00s teardown tests/infer/test_autoguide.py::test_nested_autoguide[TraceGraph_ELBO]
0.00s teardown tests/distributions/test_lkj.py::test_constraint[value_shape2]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[OneHotCategorical]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[LowRankMultivariateNormal]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4--0.5]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.1--0.5]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Multinomial]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-2-1-1]
0.00s teardown tests/optim/test_optim.py::test_dynamic_lr[scheduler1]
0.00s teardown tests/ops/test_contract.py::test_ubersum_1[ubersum]
0.00s teardown tests/ops/test_contract.py::test_contract_tensor_tree[example2]
0.00s teardown tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[TraceTMC_ELBO]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_hmm[1]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_6[parallel]
0.00s teardown tests/distributions/test_spanning_tree.py::test_partition_function[2]
0.00s teardown tests/distributions/test_lkj.py::test_constraint[value_shape1]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99--0.5]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-1-1-7]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges410]
0.00s teardown tests/ops/test_stats.py::test_waic
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[functools.partial(<function init_to_uniform at 0x7f6d86b7e950>, radius=0.1)]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_1[10]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[10]
0.00s teardown tests/contrib/autoguide/test_inference.py::AutoGaussianChain::test_multivariatate_normal_auto
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[LogNormal]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[RejectionExponential]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-2-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-2-1-7]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges677]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges523]
0.00s teardown tests/ops/test_integrator.py::test_trajectory[HarmonicOscillator]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_value at 0x7f6d86b7e9d8>]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_non_enumerated_plate_ok[sequential]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_hmm[20]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.99--1.0]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8-1.0]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8-0.5]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges981]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[61]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[functools.partial(<function init_to_median at 0x7f6d86b7e840>, num_samples=4)]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_sample at 0x7f6d86b7e7b8>0]
0.00s teardown tests/infer/test_valid_models.py::test_mean_field_mask_ok[False]
0.00s teardown tests/infer/test_valid_models.py::test_markov_history[1]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[1]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[0]
0.00s teardown tests/infer/test_predictive.py::test_shapes[False]
0.00s teardown tests/contrib/oed/test_ewma.py::test_ewma[0.5]
0.00s call     tests/contrib/epidemiology/test_util.py::test_cat2_scalar[()]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[HalfCauchy]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[Categorical]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[ExtendedBetaBinomial]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[LKJCorrCholesky]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1408]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges65]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges817]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges664]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[26]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[]
0.00s teardown tests/infer/test_valid_models.py::test_mean_field_mask_ok[True]
0.00s teardown tests/infer/test_autoguide.py::test_scores[AutoDiagonalNormal]
0.00s teardown tests/contrib/tracking/test_assignment.py::test_persistent_smoke[enum]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0--0.5]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[BetaBinomial]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-2-1-7]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges588]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges92]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges549]
0.00s teardown tests/infer/test_valid_models.py::test_markov_history[2]
0.00s teardown tests/infer/test_valid_models.py::test_enum_in_model_diamond_error[True]
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_parallel_ok[2]
0.00s teardown tests/infer/test_valid_models.py::test_plate_no_size_ok[Trace_ELBO]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[LowRankMultivariateNormal]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Categorical]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[LowRankMultivariateNormal]
0.00s teardown tests/contrib/gp/test_models.py::test_mean_function_SGPR_FITC
0.00s teardown tests/contrib/test_minipyro.py::test_generate_data_plate[pyro]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.4--1.0]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[ShapeAugmentedBeta]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Beta]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[ShapeAugmentedGamma]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-1-1-7]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges897]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges781]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges186]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges711]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges423]
0.00s teardown tests/optim/test_optim.py::test_clippedadam_pass[1.0]
0.00s teardown tests/ops/test_integrator.py::test_trajectory[CircularPlanetaryMotion]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_subsample_param_ok[Trace_ELBO]
0.00s teardown tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data0]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_hmm[10]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Gamma]
0.00s teardown tests/distributions/test_delta.py::test_expand[batch_shape1]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[2]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.0--1.0]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Geometric]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[Poisson]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-2-1-1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges536]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges477]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges326]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges866]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges843]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges476]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges406]
0.00s teardown tests/ops/test_integrator.py::test_energy_conservation[CircularPlanetaryMotion]
0.00s teardown tests/ops/test_integrator.py::test_energy_conservation[HarmonicOscillator]
0.00s teardown tests/ops/test_contract.py::test_contract_tensor_tree[example3]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[functools.partial(<function init_to_value at 0x7f6d86b7e9d8>, values={'x': tensor(3.)})]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_7[10]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_2[1]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_in_iplate[9]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[CoalescentTimes]
0.00s teardown tests/distributions/test_lkj.py::test_unconstrained_to_corr_cholesky_transform[y_shape2]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[LKJCorrCholesky]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[NaiveBeta]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[Multinomial]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[DirichletMultinomial]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-2-1-1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1351]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges593]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges97]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges28]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1226]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1093]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges393]
0.00s teardown tests/poutine/test_poutines.py::test_pickling[<lambda>2]
0.00s teardown tests/optim/test_optim.py::test_clippedadam_pass[5.0]
0.00s teardown tests/optim/test_optim.py::test_clippedadam_pass[3.0]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[wi,wi->i]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_model_with_subsample[5]
0.00s teardown tests/infer/test_valid_models.py::test_enum_in_model_diamond_error[False]
0.00s teardown tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data3]
0.00s teardown tests/infer/test_enum.py::test_backwardsample_posterior_smoke[data2]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_7[10]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_in_iplate[5]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Uniform]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[GammaPoisson]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[MultivariateStudentT]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-1-1-7]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-1-1-1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1070]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges944]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1046]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[74]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[59]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[56]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[50]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[44]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[40]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[33]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[10]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[6]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[4]
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_strategy_smoke[<function init_to_uniform at 0x7f6d86b7e950>]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0-0.5]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3-0.5]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[GammaPoisson]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[ExtendedBinomial]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[Binomial]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[Categorical]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-1-1-1]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape0-3-2-1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1225]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges892]
0.00s teardown tests/optim/test_optim.py::test_clippedadam_clip[1.0]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[w,x->]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[90]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[81]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[73]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[68]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[63]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[57]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[32]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[27]
0.00s teardown tests/infer/test_enum.py::test_compute_marginals_hmm[3]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Poisson]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Dirichlet]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-3-2-1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1309]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges630]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges242]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges726]
0.00s teardown tests/optim/test_optim.py::test_dynamic_lr[scheduler2]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[w,w->]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[89]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[70]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[69]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[65]
0.00s teardown tests/ops/test_stats.py::test_resample[True]
0.00s teardown tests/infer/test_autoguide.py::test_guide_list[AutoIAFNormal]
0.00s teardown tests/infer/test_autoguide.py::test_discrete_parallel[AutoDiagonalNormal]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.2]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[StableReparam-2.0-1.0]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01-1.0]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Stable]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Gamma]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[OneHotCategorical]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[ExtendedBetaBinomial]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-3-2-7]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1228]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1075]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges960]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges687]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges134]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges609]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges359]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges52]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[77]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[76]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[67]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[58]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[54]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[47]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[35]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[34]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[18]
0.00s teardown tests/ops/test_stats.py::test_resample[False]
0.00s teardown tests/ops/test_integrator.py::test_time_reversibility[QuarticOscillator]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_ok[Trace_ELBO]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_plate_in_iplate[2]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_0_gate[0.9]
0.00s teardown tests/distributions/test_mask.py::test_mask_type[mask2]
0.00s teardown tests/distributions/test_haar.py::test_haar_ortho[9]
0.00s teardown tests/distributions/test_extended.py::test_extended_binomial[0.1]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[FoldedNormal]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[27]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[1.0]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.1]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[DirichletMultinomial]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[DirichletMultinomial]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape1-3-2-7]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1176]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges776]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges724]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges625]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges394]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges331]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges279]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges183]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges60]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges737]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges583]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges489]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges278]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges143]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges0]
0.00s teardown tests/optim/test_optim.py::test_dynamic_lr[scheduler0]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[ij->ji]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[ij,iwj->ij]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[w->]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[64]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[51]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[37]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[31]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[20]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[17]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[9]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[8]
0.00s teardown tests/infer/test_valid_models.py::test_nonnested_plate_plate_ok[Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_iplate_ok[TraceGraph_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_variable_clash_in_model_error[TraceEnum_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plates_1[1]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_2[1]
0.00s teardown tests/infer/test_autoguide.py::test_scores[AutoNormal]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_onehot_shapes[probs1]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[GaussianMRF]
0.00s teardown tests/distributions/test_extended.py::test_extended_beta_binomial[0.0]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0-0.0]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.3-1.0]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[VonMises]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges976]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges274]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1428]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1158]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges750]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges484]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges338]
0.00s teardown tests/optim/test_optim.py::test_clippedadam_clip[3.0]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[,w->]
0.00s teardown tests/ops/test_tensor_utils.py::test_dct[shape1]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[99]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[92]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[52]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[41]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[28]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[5]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_3[TraceGraph_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_iplate_ok[Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[TraceTMC_ELBO]
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_2[10]
0.00s teardown tests/infer/test_autoguide.py::test_discrete_parallel[AutoIAFNormal]
0.00s teardown tests/infer/test_autoguide.py::test_scores[AutoIAFNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_0_gate[10.0]
0.00s teardown tests/distributions/test_extended.py::test_extended_binomial[0.0]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01--0.5]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-0.8--0.5]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support[Geometric]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges902]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges472]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges405]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1368]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges939]
0.00s teardown tests/ops/test_tensor_utils.py::test_idct[shape1]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[96]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[88]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[87]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[85]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[75]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[72]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[62]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[53]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[43]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[38]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[22]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[12]
0.00s teardown tests/infer/mcmc/test_adaptation.py::test_arrowhead_mass_matrix[True]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_3[TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_nested_autoguide[Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_scores[AutoLowRankMultivariateNormal]
0.00s teardown tests/distributions/test_extended.py::test_extended_binomial[0.02]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Geometric]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.5]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.2]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7--1.0]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1415]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1321]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1080]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1033]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges887]
0.00s teardown tests/poutine/test_poutines.py::test_pickling[<lambda>0]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[iwj,iwj->ji]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[93]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[84]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[60]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[49]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[45]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[42]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[30]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[25]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[24]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[16]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[11]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[7]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[acdb]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_subsample_param_ok[TraceGraph_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[TraceGraph_ELBO]
0.00s teardown tests/infer/test_predictive.py::test_shapes[True]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[ShapeAugmentedBeta]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[0.05]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0-1.0]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[Exponential]
0.00s call     tests/distributions/test_distributions.py::test_batch_log_prob[GammaPoisson]
0.00s call     tests/contrib/forecast/test_util.py::test_prefix_condition[LinearHMM-batch_shape2-3-2-1]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges856]
0.00s teardown tests/optim/test_optim.py::test_clippedadam_clip[5.0]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[97]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[94]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[91]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[80]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[71]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[39]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[19]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[15]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[13]
0.00s teardown tests/ops/test_arrowhead.py::test_utilities[5]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_ok[TraceGraph_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_1_gate[1.1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_0_gate[0.5]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Multinomial]
0.00s teardown tests/distributions/test_extended.py::test_extended_beta_binomial[0.05]
0.00s teardown tests/distributions/test_extended.py::test_extended_binomial[0.05]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ShapeAugmentedGamma]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[NaiveDirichlet]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[14]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[12]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.7-1.0]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1272]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1262]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1168]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges860]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges467]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges178]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges906]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges596]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges440]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges325]
0.00s teardown tests/optim/test_optim.py::test_clippedadam_lrd[5.0]
0.00s teardown tests/optim/test_optim.py::test_clippedadam_lrd[1.0]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[86]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[83]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[36]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[21]
0.00s teardown tests/ops/test_integrator.py::test_time_reversibility[CircularPlanetaryMotion]
0.00s teardown tests/infer/test_valid_models.py::test_reparam_ok[TraceMeanField_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_3[Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_plate_ok[Trace_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_discrete_parallel[AutoNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_0_gate[2.0]
0.00s teardown tests/distributions/test_haar.py::test_haar_ortho[3]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[ExtendedBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Beta]
0.00s teardown tests/contrib/oed/test_ewma.py::test_ewma[0.99]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1355]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges244]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges166]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[82]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[78]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[adcb]
0.00s teardown tests/infer/test_valid_models.py::test_not_has_rsample_ok[TraceGraph_ELBO]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[VonMises]
0.00s teardown tests/distributions/test_mask.py::test_mask_type[False]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[NaiveBeta]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[NaiveBeta]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[22]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[16]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[15]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[11]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[8]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1207]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges321]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges973]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges643]
0.00s teardown tests/params/test_module.py::test_random_module[outer]
0.00s teardown tests/optim/test_optim.py::test_dynamic_lr[scheduler3]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[79]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[66]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[14]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[TraceTMC_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_1[Trace_ELBO]
0.00s teardown tests/infer/test_jit.py::test_traceenum_elbo[1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_1_gate[10.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_0_gate[1.0]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Binomial]
0.00s teardown tests/distributions/test_mask.py::test_mask_type[mask3]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Bernoulli]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Multinomial]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[21]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[20]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[9]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_binomial[1.0]
0.00s setup    tests/test_util.py::test_warn_if_nan
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01-0.5]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1028]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1140]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges372]
0.00s teardown tests/ops/test_tensor_utils.py::test_periodic_features[48]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[bcad]
0.00s teardown tests/ops/test_arrowhead.py::test_utilities[0]
0.00s teardown tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTraceEnum_ELBO]
0.00s teardown tests/infer/test_enum.py::test_unsafe_bernoulli[sample_shape1]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[RelaxedOneHotCategorical]
0.00s teardown tests/distributions/test_haar.py::test_haar_ortho[1]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_log_prob[1]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ExtendedBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[LKJCorrCholesky]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ShapeAugmentedDirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ShapeAugmentedDirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ShapeAugmentedBeta]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Poisson]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Stable]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[VonMises]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Dirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Uniform]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[29]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[25]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[13]
0.00s teardown tests/contrib/oed/test_ewma.py::test_ewma[0.9]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1267]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges833]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges771]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1295]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1171]
0.00s teardown tests/poutine/test_poutines.py::test_pickling[<lambda>3]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_2[TraceGraph_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[TraceEnum_ELBO]
0.00s teardown tests/infer/test_jit.py::test_svi_irregular_batch_size[TraceEnum_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_init_loc_fn[AutoDiagonalNormal]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_1_gate[0.1]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_0_gate[0.1]
0.00s teardown tests/distributions/test_polya_gamma.py::test_polya_gamma[batch_shape0]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[GaussianScaleMixture]
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_mix_of_diag_normals_shared_cov_log_prob[1]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Binomial]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[HalfCauchy]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[NaiveDirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[ShapeAugmentedGamma]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[28]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[18]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[4]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-1.01--1.0]
0.00s call     tests/distributions/test_distributions.py::test_enumerate_support_shape[FoldedNormal]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges246]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1248]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1119]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges304]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[dcba]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[cdab]
0.00s teardown tests/infer/test_valid_models.py::test_nonnested_plate_plate_ok[TraceGraph_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_1_gate[0.5]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Categorical]
0.00s teardown tests/distributions/test_mask.py::test_mask_noop[(3, 2)]
0.00s teardown tests/distributions/test_haar.py::test_haar_ortho[7]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[ShapeAugmentedDirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[LogNormal]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[ShapeAugmentedBeta]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[MultivariateStudentT]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Geometric]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Normal]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[LogNormal]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.05]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0--1.0]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges531]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges231]
0.00s teardown tests/optim/test_optim.py::test_clippedadam_lrd[3.0]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[i->i]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[dacb]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[cbda]
0.00s teardown tests/ops/test_integrator.py::test_energy_conservation[QuarticOscillator]
0.00s teardown tests/ops/test_contract.py::test_ubersum_collide_ok_1[naive_ubersum]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_4[TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_sizes[sizes0]
0.00s teardown tests/infer/test_valid_models.py::test_variable_clash_in_model_error[TraceTMC_ELBO]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[11]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_1_gate[2.0]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_error[shapes1]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[RelaxedOneHotCategoricalStraightThrough]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Binomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[LogNormal]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[ShapeAugmentedBeta]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Binomial]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[LKJCorrCholesky]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[26]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[23]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[19]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_cat2_scalar[(2, 3, 4)]
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_overdispersed_beta_binomial[0.5]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes4]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges939]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges400]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[dbac]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[bdca]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_variable_clash_error[TraceEnum_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_0_gate[1.1]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[RelaxedBernoulliStraightThrough]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[MixtureOfDiagNormals]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[MaskedDistribution]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Gamma]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Beta]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[ShapeAugmentedGamma]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[RejectionExponential]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[24]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes12]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1081]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1106]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_reparameterized
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianPyramidTests::test_elbo_nonreparameterized_two_layers_model_permuted
0.00s teardown tests/infer/test_enum.py::test_unsafe_bernoulli[sample_shape0]
0.00s teardown tests/infer/test_autoguide.py::test_discrete_parallel[AutoLaplaceApproximation]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Rejector]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[MixtureOfDiagNormalsSharedCovariance]
0.00s teardown tests/distributions/test_haar.py::test_haar_ortho[8]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Cauchy]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Cauchy]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Binomial]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[ShapeAugmentedDirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Uniform]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[17]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[5]
0.00s teardown tests/test_generic.py::test_not_implemented[minipyro]
0.00s call     tests/infer/reparam/test_stable.py::test_distribution[SymmetricStableReparam-2.0--0.5]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1163]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[cdba]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[bdac]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::NormalNormalNormalTests::test_elbo_nonreparameterized_nn_baseline
0.00s teardown tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTraceMeanField_ELBO]
0.00s teardown tests/infer/test_jit.py::test_svi_irregular_batch_size[TraceGraph_ELBO]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_1_gate[1.0]
0.00s teardown tests/distributions/test_zero_inflated.py::test_zip_1_gate[0.9]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[SpanningTree]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[ContinuousBernoulli]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_normal_shape[()]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[BetaBinomial]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[6]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges83]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[cbad]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[cadb]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[adbc]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_2[TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_2[Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_variable_clash_error[Trace_ELBO]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[3]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_normal_shape[(2, 3)]
0.00s teardown tests/distributions/test_haar.py::test_haar_ortho[4]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Stable]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Normal]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[Multinomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ShapeAugmentedBeta]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[RejectionExponential]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[NaiveBeta]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Beta]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[ShapeAugmentedDirichlet]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_aps_hash[100]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1065]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges271]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[->]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[dbca]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[dabc]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[bacd]
0.00s teardown tests/infer/test_valid_models.py::test_variable_clash_in_guide_error[Trace_ELBO]
0.00s teardown tests/infer/test_sampling.py::ImportanceTest::test_importance_guide
0.00s teardown tests/infer/test_enum.py::test_unsafe_bernoulli[sample_shape2]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[FoldedDistribution]
0.00s teardown tests/distributions/test_kl.py::test_kl_delta_normal_shape[(4,)]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Multinomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ShapeAugmentedGamma]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[VonMises]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_aps_try_add[0.1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1427]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges224]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[dcab]
0.00s teardown tests/ops/test_packed.py::test_unpack_pack[cabd]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_sizes[sizes1]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[7]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[RelaxedBernoulli]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[NegativeBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[HalfCauchy]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[NaiveDirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Uniform]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[HalfCauchy]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Exponential]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_aps_try_add[100]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1220]
0.00s teardown tests/infer/test_valid_models.py::test_plate_reuse_ok[TraceGraph_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_stack_sizes[sizes2]
0.00s teardown tests/infer/test_valid_models.py::test_plate_no_size_ok[TraceGraph_ELBO]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[ZeroInflatedNegativeBinomial]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Exponential]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Empirical]
0.00s teardown tests/distributions/test_extended.py::test_extended_beta_binomial[0.02]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Normal]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Gamma]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Cauchy]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_aps_hash[0.1]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges109]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges96]
0.00s teardown tests/ops/test_integrator.py::test_time_reversibility[HarmonicOscillator]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_4[Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_broadcast_error[Trace_ELBO]
0.00s teardown tests/infer/test_jit.py::test_svi_irregular_batch_size[JitTraceGraph_ELBO]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[MultivariateStudentT]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[LKJCorrCholesky]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[BetaBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[OneHotCategorical]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[NaiveDirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[RejectionExponential]
0.00s teardown tests/params/test_module.py::test_random_module[outest]
0.00s teardown tests/ops/einsum/test_adjoint.py::test_marginal[ij->ij]
0.00s teardown tests/ops/test_contract.py::test_ubersum_collide_ok_3[naive_ubersum]
0.00s teardown tests/infer/test_valid_models.py::test_nested_plate_plate_dim_error_4[TraceGraph_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_plate_broadcast_error[TraceGraph_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_variable_clash_in_guide_error[TraceEnum_ELBO]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[MixtureSameFamily]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[ExtendedBinomial]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[DirichletMultinomial]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[MultivariateStudentT]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Bernoulli]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Gamma]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Poisson]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[ExtendedBinomial]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_aps_hash[10]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1439]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges564]
0.00s teardown tests/poutine/test_mapdata.py::test_custom_subsample[iplate]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Stable]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[OneHotCategorical]
0.00s teardown tests/distributions/test_mask.py::test_mask_type[True]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Multinomial]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[ExtendedBetaBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Bernoulli]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[Categorical]
0.00s teardown tests/contrib/gp/test_models.py::test_mean_function_SGPR_DTC
0.00s teardown tests/contrib/test_minipyro.py::test_generate_data_plate[minipyro]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1027]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges404]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges397]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges204]
0.00s teardown tests/params/test_module.py::test_param_no_grad[outer]
0.00s teardown tests/ops/test_packed.py::test_broadcast_all[shapes3]
0.00s teardown tests/ops/test_contract.py::test_ubersum_collide_ok_3[ubersum]
0.00s teardown tests/infer/test_valid_models.py::test_variable_clash_in_model_error[TraceGraph_ELBO]
0.00s teardown tests/infer/test_jit.py::test_svi_irregular_batch_size[TraceMeanField_ELBO]
0.00s teardown tests/infer/test_jit.py::test_svi_irregular_batch_size[Trace_ELBO]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[GammaPoisson]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Dirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Normal]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Dirichlet]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_aps_try_add[1]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes0]
0.00s teardown tests/infer/test_compute_downstream_costs.py::test_compute_downstream_costs_duplicates[2]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[ZeroInflatedDistribution]
0.00s teardown tests/distributions/test_mask.py::test_mask_noop[None]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Beta]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[Geometric]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[RejectionExponential]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[GammaPoisson]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Cauchy]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[LogNormal]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob_shape[Exponential]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_aps_try_add[10]
0.00s call     tests/contrib/gp/test_models.py::test_inference[VariationalGP_y2D]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::BernoulliBetaTests::test_elbo_nonreparameterized
0.00s teardown tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[TraceGraph_ELBO]
0.00s teardown tests/infer/test_autoguide.py::test_init_loc_fn[AutoDelta]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Unit]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[StudentT]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[InverseGamma]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Geometric]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[GammaPoisson]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[FisherSnedecor]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[ShapeAugmentedGamma]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Bernoulli]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[BetaBinomial]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[0.1]
0.00s teardown tests/poutine/test_poutines.py::test_pickling[<lambda>1]
0.00s teardown tests/infer/test_valid_models.py::test_variable_clash_in_guide_error[TraceGraph_ELBO]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[MaskedMixture]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[ImproperUniform]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[HalfNormal]
0.00s teardown tests/distributions/test_mask.py::test_mask_noop[(4,)]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[BetaBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[VonMises]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges89]
0.00s teardown tests/params/test_module.py::test_module_nn[outest]
0.00s teardown tests/infer/test_valid_models.py::test_plate_broadcast_error[TraceEnum_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[TraceGraph_ELBO]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Uniform]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Gumbel]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[GaussianHMM]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[CoalescentTimesWithRate]
0.00s teardown tests/distributions/test_independent.py::test_to_event[normal]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[ExtendedBetaBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[MultivariateStudentT]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Exponential]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes14]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges60]
0.00s teardown tests/ops/test_packed.py::test_broadcast_all[shapes2]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_variable_clash_error[TraceGraph_ELBO]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_scalar[dense]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Bernoulli]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[100]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1361]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1273]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges75]
0.00s teardown tests/ops/test_packed.py::test_broadcast_all[shapes4]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[VonMises3D]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[TransformedDistribution]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[ExtendedBetaBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[DirichletMultinomial]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Bernoulli]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[VonMises]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[NaiveBeta]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[FoldedNormal]
0.00s teardown tests/contrib/tracking/test_em.py::test_svi_multi
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges163]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1020]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[HalfCauchy]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[ShapeAugmentedDirichlet]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_hash_nearby[10]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges895]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges33]
0.00s teardown tests/infer/test_valid_models.py::test_iplate_variable_clash_error[TraceTMC_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_model_guide_shape_mismatch_error[Trace_ELBO]
0.00s teardown tests/infer/test_valid_models.py::test_model_guide_dim_mismatch_error[Trace_ELBO]
0.00s teardown tests/distributions/test_relaxed_straight_through.py::test_onehot_svi_usage
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Weibull]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Pareto]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[LogisticNormal]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Cauchy]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Beta]
0.00s teardown tests/distributions/test_mask.py::test_mask_noop[()]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[ExtendedBetaBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[LKJCorrCholesky]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[GammaPoisson]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[BetaBinomial]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges859]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges482]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges507]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges26]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Normal]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[GammaGaussianHMM]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Delta]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Geometric]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[FoldedNormal]
0.00s teardown tests/distributions/test_distributions.py::test_batch_entropy_shape[Cauchy]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[ExtendedBetaBinomial]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[reparam-JitTrace_ELBO]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges830]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges511]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1116]
0.00s teardown tests/params/test_module.py::test_param_no_grad[outest]
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_mmd_vectorized
0.00s teardown tests/distributions/test_spanning_tree.py::test_enumerate_support[5]
0.00s teardown tests/distributions/test_spanning_tree.py::test_enumerate_support[1]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[LogNormal]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Chi2]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[MultivariateNormal]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[BetaBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Bernoulli]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_aps_init[scale2]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1149]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1033]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges959]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges907]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges685]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges627]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges570]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges165]
0.00s teardown tests/poutine/test_mapdata.py::test_custom_subsample[plate]
0.00s teardown tests/params/test_module.py::test_module_nn[outer]
0.00s teardown tests/infer/test_enum.py::test_iter_discrete_traces_scalar[flat]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Poisson]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Laplace]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[NaiveDirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Poisson]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[HalfCauchy]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Dirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[GammaPoisson]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[DirichletMultinomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[MultivariateStudentT]
0.00s teardown tests/contrib/gp/test_models.py::test_mean_function_VSGP
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1366]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges949]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges853]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges697]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges484]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges255]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges130]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::BernoulliBetaTests::test_elbo_reparameterized
0.00s teardown tests/infer/test_enum.py::test_multi_dependence_enumeration
0.00s teardown tests/distributions/test_spanning_tree.py::test_enumerate_support[4]
0.00s teardown tests/distributions/test_spanning_tree.py::test_enumerate_support[3]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Dirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[ShapeAugmentedBeta]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Normal]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Gamma]
0.00s teardown tests/distributions/test_coalescent.py::test_sample_is_valid[3]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes15]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1277]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1181]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges858]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges640]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges373]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges367]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1269]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1243]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1148]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges828]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges471]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges187]
0.00s teardown tests/poutine/test_poutines.py::QueueHandlerDiscreteTest::test_queue_max_tries
0.00s teardown tests/distributions/test_unit.py::test_shapes[batch_shape1]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[HalfCauchy]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[NaiveBeta]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[RejectionExponential]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[Poisson]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[ExtendedBetaBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Poisson]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Stable]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1421]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges954]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges714]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges603]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges420]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges368]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges336]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges220]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1127]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges988]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges860]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges765]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges218]
0.00s teardown tests/params/test_module.py::test_module_sequential[nn_module0]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[NaiveBeta]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Uniform]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[LogNormal]
0.00s teardown tests/contrib/test_minipyro.py::test_generate_data[minipyro]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1314]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1107]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges848]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges713]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges598]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges415]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges247]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges102]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1192]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1007]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges437]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges335]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges312]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Poisson]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[LogNormal]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_add[1]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes14]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1413]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1293]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1230]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1186]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1154]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1137]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges821]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges551]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges544]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges424]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges318]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges47]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges38]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges956]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges926]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges796]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges539]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges505]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges427]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges267]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges265]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges252]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges161]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges13]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::NormalNormalTests::test_elbo_nonreparameterized
0.00s teardown tests/integration_tests/test_conjugate_gaussian_models.py::GaussianChainTests::test_elbo_reparameterized_N_is_3
0.00s teardown tests/infer/mcmc/test_nuts.py::test_arrowhead_mass
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Gamma]
0.00s teardown tests/distributions/test_independent.py::test_to_event[mvn]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[GammaPoisson]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[VonMises]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Dirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Exponential]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[Binomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[GammaPoisson]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Cauchy]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[NaiveDirichlet]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Beta]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Exponential]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_cat2_scalar[(2,)]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=4 --cuda]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1356]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1229]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1178]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1177]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1159]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1084]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1073]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1049]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges991]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges917]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges796]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges692]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges645]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges618]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges617]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges581]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges569]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges546]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges443]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges366]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges308]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges289]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges250]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges243]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges144]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges48]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1389]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1329]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1153]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1114]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1067]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges764]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges759]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges557]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges301]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[Geometric]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[DirichletMultinomial]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[ExtendedBinomial]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_non_broadcastable_data_shape[Stable]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Geometric]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[MultivariateStudentT]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_init[scale2]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1433]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1419]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1381]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1321]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1294]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1253]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1250]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1164]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1119]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1113]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1112]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1108]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1091]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1062]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges975]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges946]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges889]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges843]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges801]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges786]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges773]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges743]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges656]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges635]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges597]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges556]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges547]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges500]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges463]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges423]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges411]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges337]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges316]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges284]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges240]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges160]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges50]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges42]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1213]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges981]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges767]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges700]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges682]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges630]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges353]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges306]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges286]
0.00s teardown tests/integration_tests/test_tracegraph_elbo.py::NormalNormalTests::test_elbo_reparameterized
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support[OneHotCategorical]
0.00s teardown tests/distributions/test_distributions.py::test_batch_log_prob[DirichletMultinomial]
0.00s teardown tests/contrib/gp/test_models.py::test_mean_function_SGPR
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1410]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1406]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1342]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1320]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1319]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1299]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1221]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1180]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1170]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1106]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1060]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1059]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1001]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges929]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges887]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges875]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges791]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges772]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges731]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges715]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges661]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges654]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges609]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges602]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges583]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges526]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges520]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges437]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges395]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges383]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges378]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges375]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges363]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges327]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges280]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges267]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges221]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges219]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges192]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges156]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges139]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges105]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges77]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges64]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1263]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges960]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges924]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges874]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges847]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges738]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges719]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges636]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges544]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges493]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges346]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges203]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges174]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges97]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges39]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[Independent]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[LKJCorrCholesky]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Normal]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_add[100]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1438]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1418]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1405]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1352]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1348]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1346]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1288]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1265]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1248]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1247]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1211]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1206]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1189]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1182]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1175]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1167]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1130]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1120]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1118]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1080]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1077]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges986]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges973]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges961]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges953]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges916]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges899]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges857]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges816]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges793]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges784]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges768]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges742]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges729]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges725]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges688]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges681]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges673]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges666]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges659]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges636]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges624]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges621]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges601]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges579]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges573]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges532]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges530]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges442]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges328]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges307]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges304]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges273]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges236]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges216]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges190]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges39]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges21]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1256]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1240]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1201]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1180]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1052]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1044]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1027]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1012]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges983]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges892]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges844]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges830]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges804]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges778]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges704]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges698]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges480]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges458]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges443]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges363]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges362]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges330]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges320]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges299]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges293]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges260]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges239]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges131]
0.00s teardown tests/distributions/test_unit.py::test_shapes[batch_shape2]
0.00s teardown tests/distributions/test_spanning_tree.py::test_enumerate_support[2]
0.00s teardown tests/distributions/test_independent.py::test_to_event[exponential]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[HalfCauchy]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Dirichlet]
0.00s teardown tests/distributions/test_delta.py::test_expand[batch_shape5]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes10]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1412]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1397]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1373]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1371]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1354]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1353]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1350]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1341]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1325]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1323]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1296]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1285]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1251]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1223]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1214]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1165]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1099]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1097]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1096]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1004]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1000]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges998]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges989]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges982]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges965]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges955]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges952]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges933]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges922]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges921]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges914]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges912]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges901]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges891]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges890]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges888]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges881]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges847]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges832]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges811]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges774]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges760]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges748]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges744]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges741]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges740]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges721]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges718]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges709]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges699]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges672]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges669]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges657]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges651]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges623]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges616]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges576]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges574]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges570]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges568]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges537]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges519]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges510]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges499]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges483]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges453]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges449]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges427]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges416]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges413]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges406]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges399]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges392]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges371]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges313]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges312]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges311]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges268]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges256]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges255]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges244]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges232]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges229]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges226]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges196]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges188]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges147]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges128]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges123]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges100]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges45]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1311]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1265]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1264]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1251]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1234]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1195]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1113]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1111]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1095]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges911]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges864]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges825]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges782]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges775]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges717]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges696]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges641]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges536]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges510]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges508]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges465]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges461]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges403]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges392]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges354]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges289]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges282]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges277]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges253]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges246]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges221]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges205]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges186]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges146]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges4]
0.00s teardown tests/ops/test_packed.py::test_broadcast_all[shapes1]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Uniform]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_add[0.1]
0.00s teardown tests/test_generic.py::test_not_implemented[pyro]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1434]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1432]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1414]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1379]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1365]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1359]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1357]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1347]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1322]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1312]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1306]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1301]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1297]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1295]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1287]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1279]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1278]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1264]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1261]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1255]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1213]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1210]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1205]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1166]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1146]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1141]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1098]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1094]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1085]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1068]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1046]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1025]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1021]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1017]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1014]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1010]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges996]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges964]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges962]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges957]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges905]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges880]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges865]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges861]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges856]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges849]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges841]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges837]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges829]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges792]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges783]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges780]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges777]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges749]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges739]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges734]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges712]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges708]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges703]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges701]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges689]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges664]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges655]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges642]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges629]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges613]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges596]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges592]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges575]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges572]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges558]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges545]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges496]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges493]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges486]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges471]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges455]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges429]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges412]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges402]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges396]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges387]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges376]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges374]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges364]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges360]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges352]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges342]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges340]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges333]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges323]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges322]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges299]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges294]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges292]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges263]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges245]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges239]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges231]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges222]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges215]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges214]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges210]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges150]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges137]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges71]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges56]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges46]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges34]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges4]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1433]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1420]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1402]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1395]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1361]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1359]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1299]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1287]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1275]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1214]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1212]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1210]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1204]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1193]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1183]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1156]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1135]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1132]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1124]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1013]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1005]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges996]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges826]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges732]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges697]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges666]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges652]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges650]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges606]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges579]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges558]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges551]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges515]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges467]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges436]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges420]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges388]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges365]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges331]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges322]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges315]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges305]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges298]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges296]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges261]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges254]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges216]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges117]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges77]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges55]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges11]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges3]
0.00s teardown tests/distributions/test_distributions.py::test_score_errors_event_dim_mismatch[Exponential]
0.00s teardown tests/distributions/test_delta.py::test_expand[batch_shape3]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_add[10]
0.00s call     tests/pyroapi/test_pyroapi.py::test_mean_field_ok[minipyro]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1429]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1425]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1420]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1417]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1399]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1392]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1378]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1349]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1338]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1334]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1332]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1330]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1326]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1324]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1289]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1257]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1252]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1249]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1236]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1232]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1231]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1208]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1204]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1203]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1201]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1191]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1185]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1173]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1145]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1129]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1128]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1116]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1105]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1104]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1093]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1092]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1089]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1072]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1056]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1047]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1039]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1019]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1011]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1009]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges997]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges984]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges979]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges967]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges966]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges958]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges947]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges942]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges936]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges934]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges924]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges918]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges903]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges893]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges878]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges873]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges869]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges834]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges824]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges814]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges807]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges790]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges785]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges782]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges761]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges738]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges737]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges735]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges717]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges710]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges702]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges694]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges693]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges691]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges676]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges675]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges649]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges648]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges643]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges612]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges608]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges604]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges594]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges586]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges585]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges582]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges561]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges560]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges549]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges506]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges505]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges501]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges497]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges494]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges475]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges469]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges468]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges466]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges460]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges458]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges457]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges448]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges447]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges422]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges391]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges359]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges351]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges324]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges315]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges298]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges290]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges282]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges278]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges264]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges249]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges211]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges208]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges181]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges158]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges148]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges119]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges99]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges96]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges87]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges85]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges72]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges61]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges59]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges52]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges41]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges26]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges18]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges12]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1431]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1312]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1289]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1285]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1281]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1258]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1208]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1173]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1172]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1157]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1144]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1121]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1096]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1071]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1034]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges977]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges946]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges941]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges854]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges807]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges797]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges791]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges771]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges766]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges760]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges758]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges754]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges703]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges644]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges640]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges632]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges628]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges604]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges569]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges538]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges526]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges519]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges506]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges499]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges498]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges490]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges485]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges459]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges435]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges434]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges369]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges357]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges337]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges332]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges311]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges290]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges281]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges274]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges271]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges262]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges234]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges199]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges191]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges189]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges178]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges170]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges127]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges70]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_dirichlet_categorical_grad_adapt
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes5]
0.00s teardown tests/distributions/test_pickle.py::test_pickle[TruncatedPolyaGamma]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes13]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1404]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1400]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1395]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1388]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1358]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1329]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1290]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1258]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1256]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1235]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1226]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1218]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1217]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1216]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1190]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1188]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1172]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1169]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1158]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1152]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1144]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1132]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1122]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1117]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1103]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1100]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1078]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1076]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1066]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1048]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1043]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1006]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges999]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges988]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges978]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges972]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges963]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges945]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges940]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges931]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges904]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges894]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges866]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges854]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges836]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges835]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges820]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges817]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges812]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges805]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges803]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges802]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges795]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges788]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges775]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges751]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges747]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges736]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges733]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges726]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges700]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges695]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges690]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges682]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges680]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges677]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges667]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges653]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges652]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges647]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges638]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges637]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges614]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges600]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges555]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges550]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges524]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges521]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges514]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges513]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges507]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges487]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges479]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges476]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges474]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges462]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges459]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges446]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges445]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges440]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges436]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges419]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges418]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges417]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges401]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges398]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges393]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges386]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges339]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges330]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges325]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges306]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges266]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges233]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges230]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges225]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges209]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges199]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges187]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges184]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges177]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges174]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges168]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges164]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges162]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges157]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges155]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges153]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges146]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges132]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges121]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges111]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges90]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges78]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges76]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges68]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges63]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges40]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges37]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges36]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges24]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges15]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges5]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges0]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1434]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1432]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1398]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1388]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1385]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1371]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1348]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1343]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1276]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1268]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1267]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1242]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1222]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1207]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1194]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1188]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1186]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1178]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1177]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1169]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1149]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1141]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1133]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1110]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1101]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1100]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1099]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1097]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1092]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1084]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1078]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1060]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1021]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1011]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges969]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges968]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges948]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges935]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges901]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges895]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges885]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges855]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges840]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges822]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges790]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges777]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges773]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges761]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges746]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges744]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges723]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges667]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges655]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges653]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges614]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges603]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges591]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges586]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges581]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges580]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges565]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges554]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges545]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges516]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges514]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges500]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges495]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges494]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges492]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges482]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges475]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges474]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges442]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges425]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges391]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges384]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges378]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges370]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges368]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges356]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges351]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges334]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges333]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges309]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges295]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges292]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges279]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges273]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges268]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges256]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges238]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges233]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges207]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges201]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges168]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges159]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges37]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1437]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1428]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1422]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1411]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1409]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1402]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1390]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1383]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1382]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1377]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1369]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1344]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1343]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1336]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1310]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1307]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1305]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1303]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1302]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1300]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1269]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1266]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1238]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1237]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1233]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1209]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1199]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1155]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1153]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1147]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1133]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1123]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1110]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1088]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1083]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1082]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1069]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1063]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1024]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1015]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1013]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1012]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1005]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1002]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges995]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges977]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges969]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges920]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges911]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges909]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges885]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges884]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges876]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges874]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges868]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges862]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges855]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges851]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges846]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges845]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges844]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges840]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges822]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges818]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges764]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges762]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges753]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges720]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges719]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges706]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges678]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges670]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges668]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges658]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges633]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges626]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges619]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges610]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges605]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges590]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges589]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges571]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges562]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges541]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges540]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges528]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges527]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges509]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges508]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges490]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges481]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges478]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges464]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges461]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges441]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges435]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges431]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges428]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges408]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges390]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges381]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges380]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges361]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges343]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges338]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges319]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges317]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges297]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges291]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges287]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges277]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges272]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges261]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges254]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges251]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges248]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges237]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges228]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges218]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges195]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges179]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges167]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges159]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges152]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges141]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges127]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges86]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges83]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges54]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges51]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges49]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges30]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges23]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges20]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges19]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges16]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges14]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1417]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1404]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1400]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1394]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1384]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1379]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1374]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1366]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1338]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1334]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1318]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1305]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1301]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1271]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1261]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1260]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1244]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1238]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1228]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1205]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1199]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1187]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1182]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1155]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1137]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1130]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1126]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1098]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1088]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1086]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1085]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1082]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1081]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1057]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1050]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1049]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1041]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1028]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1017]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges979]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges959]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges951]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges950]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges929]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges925]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges917]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges902]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges890]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges888]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges880]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges879]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges867]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges850]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges848]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges835]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges831]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges818]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges799]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges792]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges787]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges783]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges780]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges762]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges740]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges735]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges716]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges714]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges709]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges691]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges685]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges661]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges657]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges638]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges637]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges635]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges594]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges588]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges582]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges578]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges574]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges572]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges537]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges486]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges454]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges452]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges451]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges438]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges433]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges429]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges414]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges412]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges410]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges386]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges380]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges340]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges336]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges314]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges307]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges297]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges294]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges285]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges283]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges275]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges249]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges248]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges245]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges214]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges208]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges195]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges192]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges163]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges153]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges151]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges79]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges61]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges44]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges27]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges24]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges7]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges5]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes0]
0.00s teardown tests/distributions/test_distributions.py::test_enumerate_support_shape[FoldedNormal]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_init[-1.0]
0.00s teardown tests/contrib/randomvariable/test_random_variable.py::test_multiply_divide
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1423]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1407]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1401]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1389]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1375]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1374]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1370]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1360]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1337]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1333]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1331]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1316]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1311]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1308]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1292]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1291]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1283]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1281]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1271]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1254]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1219]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1212]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1198]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1197]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1195]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1193]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1183]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1162]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1161]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1160]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1151]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1127]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1126]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1124]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1109]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1064]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1061]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1053]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1052]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1050]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1040]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1022]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1020]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1018]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges990]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges985]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges950]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges948]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges943]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges935]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges930]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges928]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges925]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges915]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges879]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges831]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges827]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges826]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges815]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges809]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges797]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges789]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges787]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges778]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges770]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges769]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges766]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges758]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges754]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges750]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges745]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges722]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges711]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges705]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges698]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges684]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges674]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges663]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges662]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges650]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges641]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges632]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges631]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges628]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges622]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges591]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges584]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges577]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges565]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges559]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges543]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges539]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges538]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges529]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges517]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges498]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges488]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges444]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges434]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges430]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges421]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges385]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges357]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges356]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges355]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges354]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges350]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges349]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges348]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges335]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges320]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges310]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges293]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges283]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges281]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges276]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges257]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges253]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges213]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges212]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges207]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges206]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges205]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges198]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges191]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges185]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges182]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges161]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges140]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges133]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges130]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges129]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges120]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges117]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges113]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges106]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges101]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges93]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges80]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges74]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges73]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges43]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges9]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges6]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1425]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1424]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1422]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1419]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1411]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1407]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1396]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1382]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1365]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1350]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1345]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1336]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1324]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1322]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1314]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1304]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1298]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1294]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1293]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1262]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1253]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1252]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1249]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1245]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1241]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1233]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1227]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1198]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1197]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1196]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1184]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1179]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1175]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1170]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1161]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1134]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1131]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1129]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1112]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1083]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1077]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1040]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1036]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1030]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1014]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1009]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1008]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1006]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges985]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges974]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges970]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges962]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges958]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges954]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges934]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges933]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges927]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges918]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges915]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges903]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges897]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges896]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges893]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges889]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges882]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges881]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges859]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges853]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges849]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges821]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges812]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges809]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges801]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges786]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges784]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges763]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges734]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges728]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges725]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges712]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges687]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges684]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges681]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges675]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges673]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges668]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges649]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges642]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges621]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges608]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges573]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges567]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges542]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges530]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges521]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges502]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges478]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges477]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges473]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges464]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges457]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges450]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges426]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges407]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges399]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges371]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges366]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges361]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges358]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges350]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges348]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges342]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges339]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges324]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges323]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges310]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges288]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges276]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges266]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges263]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges237]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges236]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges206]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges200]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges193]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges172]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges162]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges156]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges126]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges114]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges111]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges56]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges43]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges17]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges14]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges12]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges10]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes8]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes1]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1430]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1424]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1394]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1384]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1376]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1362]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1345]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1328]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1317]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1298]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1270]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1260]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1245]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1242]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1241]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1239]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1234]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1202]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1200]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1192]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1163]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1150]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1140]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1135]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1121]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1114]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1111]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1101]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1087]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1079]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1071]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1058]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1057]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1054]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1045]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1041]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges994]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges993]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges983]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges980]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges974]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges971]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges968]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges951]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges941]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges937]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges932]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges913]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges886]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges872]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges867]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges852]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges828]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges823]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges819]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges808]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges804]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges799]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges798]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges779]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges757]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges756]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges755]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges732]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges727]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges696]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges671]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges665]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges660]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges644]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges599]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges595]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges587]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges557]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges553]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges525]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges523]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges503]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges485]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges454]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges452]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges450]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges433]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges414]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges407]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges388]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges370]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges369]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges358]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges353]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges334]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges329]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges309]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges295]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges286]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges269]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges265]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges262]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges259]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges258]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges252]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges238]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges227]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges223]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges202]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges201]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges197]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges180]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges176]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges169]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges165]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges154]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges151]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges143]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges125]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges115]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges110]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges107]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges89]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges88]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges67]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges62]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges31]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges29]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges11]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges7]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1429]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1427]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1421]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1416]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1414]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1412]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1410]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1408]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1392]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1369]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1363]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1354]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1346]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1288]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1283]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1280]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1274]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1272]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1254]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1236]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1190]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1164]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1154]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1152]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1139]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1138]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1136]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1118]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1107]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1090]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1072]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1039]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1032]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1031]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1026]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1025]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1023]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges991]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges989]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges986]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges984]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges967]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges945]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges922]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges921]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges920]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges914]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges908]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges905]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges898]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges884]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges868]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges861]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges846]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges833]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges820]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges802]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges795]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges794]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges781]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges779]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges769]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges748]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges727]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges695]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges671]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges670]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges645]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges633]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges625]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges623]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges613]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges607]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges593]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges576]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges562]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges559]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges552]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges546]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges534]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges533]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges529]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges527]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges524]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges522]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges518]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges501]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges497]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges488]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges479]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges466]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges456]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges447]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges446]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges445]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges444]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges441]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges439]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges431]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges419]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges408]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges405]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges404]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges402]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges397]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges394]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges385]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges382]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges381]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges376]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges375]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges355]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges349]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges327]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges321]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges318]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges308]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges302]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges300]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges291]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges280]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges272]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges270]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges258]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges250]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges247]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges235]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges220]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges211]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges209]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges196]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges176]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges154]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges136]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges120]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges93]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges54]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges34]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges29]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges23]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges21]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges20]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges8]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_aps_init[0.0]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1436]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1396]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1393]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1386]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1367]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1364]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1355]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1282]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1240]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1227]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1156]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1136]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1131]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1115]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1074]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1055]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1044]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1042]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1037]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1029]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1023]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1008]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1003]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges992]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges987]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges970]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges938]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges927]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges926]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges910]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges896]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges877]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges871]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges850]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges839]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges794]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges767]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges763]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges752]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges746]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges730]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges728]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges723]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges704]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges646]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges620]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges615]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges606]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges580]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges578]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges563]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges554]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges548]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges533]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges518]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges516]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges515]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges495]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges492]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges491]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges489]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges473]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges439]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges438]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges426]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges409]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges389]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges384]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges372]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges362]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges347]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges332]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges303]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges285]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges275]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges241]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges235]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges234]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges189]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges171]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges166]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges149]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges135]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges126]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges112]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges104]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges84]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges57]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges55]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges25]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges13]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges3]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges2]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1426]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1405]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1403]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1401]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1397]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1383]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1381]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1372]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1370]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1358]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1357]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1353]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1349]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1347]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1337]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1335]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1332]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1325]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1323]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1317]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1316]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1303]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1297]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1296]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1278]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1277]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1266]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1250]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1225]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1221]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1211]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1189]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1176]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1143]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1125]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1109]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1105]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1104]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1091]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1076]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1074]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1070]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1064]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1055]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1048]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1042]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1038]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1018]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1016]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1015]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1003]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1000]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges998]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges978]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges975]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges965]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges963]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges961]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges955]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges953]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges947]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges944]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges932]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges919]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges907]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges904]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges899]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges891]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges883]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges872]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges870]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges869]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges845]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges842]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges841]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges839]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges819]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges814]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges810]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges753]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges741]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges730]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges722]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges713]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges710]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges707]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges705]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges694]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges690]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges679]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges678]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges674]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges656]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges651]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges647]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges627]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges626]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges622]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges605]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges602]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges600]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges597]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges556]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges555]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges491]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges481]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges448]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges422]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges413]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges401]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges396]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges383]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges379]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges374]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges367]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges364]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges319]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges317]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges284]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges243]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges241]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges240]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges230]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges228]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges226]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges217]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges197]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges194]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges180]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges177]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges160]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges157]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges152]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges144]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges141]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges129]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges123]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges119]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges72]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges53]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges47]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges42]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges40]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges16]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_aps_init[-1.0]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=7 --cuda]
0.00s call     tests/ops/test_packed.py::test_broadcast_all[shapes0]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes13]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1416]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1415]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1403]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1387]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1372]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1335]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1318]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1313]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1275]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1268]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1244]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1243]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1224]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1215]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1194]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1187]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1171]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1148]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1143]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1139]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1095]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1090]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1067]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1051]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1038]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1034]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1031]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1026]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1016]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges956]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges900]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges898]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges883]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges864]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges863]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges842]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges825]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges765]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges686]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges683]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges634]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges567]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges566]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges552]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges480]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges465]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges451]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges425]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges403]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges346]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges345]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges344]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges341]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges301]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges270]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges260]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges203]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges142]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges138]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges131]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges118]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges116]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges103]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges95]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges79]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges69]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges58]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges32]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1439]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1423]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1393]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1377]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1373]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1360]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1351]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1341]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1340]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1326]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1315]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1309]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1259]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1247]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1237]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1235]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1229]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1217]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1216]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1206]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1200]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1185]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1166]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1165]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1160]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1151]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1142]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1122]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1117]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1115]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1087]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1075]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1073]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1066]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1059]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1058]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1056]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1051]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1047]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1037]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1019]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1002]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges995]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges990]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges957]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges952]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges943]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges938]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges928]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges913]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges910]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges873]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges871]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges865]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges852]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges851]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges834]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges823]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges816]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges805]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges800]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges755]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges736]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges718]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges706]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges701]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges693]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges688]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges680]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges672]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges669]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges658]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges654]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges631]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges629]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges620]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges619]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges617]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges610]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges595]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges585]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges566]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges561]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges560]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges543]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges540]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges535]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges517]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges509]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges504]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges472]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges470]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges469]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges463]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges455]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges432]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges424]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges418]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges416]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges400]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges389]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges344]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges313]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges259]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges251]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges242]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges232]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges219]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges198]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges190]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges185]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges164]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges149]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges138]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges135]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges124]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges121]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges106]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges105]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges103]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges95]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges94]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges90]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges57]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges48]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges30]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges28]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges18]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges9]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges6]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes2]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes15]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes12]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes0]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes2]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1391]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1368]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1315]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1286]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1280]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1263]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1259]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1246]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1196]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1179]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1174]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1157]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1086]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1007]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges923]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges906]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges882]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges813]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges800]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges679]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges535]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges512]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges504]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges456]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges432]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges382]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges305]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges300]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges136]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges122]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges109]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges98]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges81]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges66]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges35]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges27]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges17]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1438]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1418]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1387]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1380]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1364]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1352]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1344]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1339]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1328]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1308]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1284]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1270]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1231]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1230]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1224]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1220]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1219]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1215]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1167]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1159]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1147]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1146]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1145]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1102]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1069]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1068]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1061]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1054]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1053]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1001]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges992]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges987]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges982]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges966]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges937]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges930]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges909]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges838]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges827]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges824]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges806]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges803]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges789]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges788]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges785]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges770]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges768]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges743]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges739]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges733]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges731]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges715]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges665]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges659]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges648]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges634]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges612]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges611]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges598]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges589]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges563]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges550]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges548]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges547]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges541]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges503]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges460]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges453]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges415]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges411]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges360]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges352]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges345]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges341]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges225]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges223]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges213]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges184]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges175]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges171]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges155]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges145]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges140]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges139]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges137]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges128]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges122]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges110]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges108]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges100]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges87]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges84]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges81]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges78]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges66]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges65]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges63]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges58]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges35]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges31]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges22]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges15]
0.00s teardown tests/optim/test_optim.py::test_autowrap[ClippedAdam]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes3]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes5]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes0]
0.00s teardown tests/distributions/test_delta.py::test_expand[batch_shape4]
0.00s setup    tests/test_util.py::test_warn_if_inf
0.00s teardown tests/test_examples.py::test_cuda[baseball.py --num-samples=200 --warmup-steps=100 --num-chains=2 --cuda]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes11]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1431]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1398]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1385]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1363]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1340]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1339]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1284]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1276]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1274]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1222]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1184]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1138]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1125]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges919]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges908]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges607]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges564]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges542]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges522]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges502]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges470]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges377]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges296]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges200]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges173]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges172]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges124]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges91]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges82]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges10]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1436]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1430]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1406]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1386]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1378]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1356]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1333]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1330]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1327]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1319]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1310]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1307]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1246]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1218]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1181]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1174]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1168]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1162]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1108]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1103]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1062]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1045]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1029]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1010]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges980]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges923]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges912]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges878]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges875]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges863]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges857]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges815]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges811]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges808]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges774]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges752]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges749]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges689]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges676]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges663]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges624]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges618]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges616]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges601]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges599]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges592]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges584]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges571]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges568]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges528]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges525]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges487]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges483]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges462]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges430]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges428]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges421]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges398]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges395]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges387]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges377]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges373]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges329]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges326]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges316]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges287]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges264]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges215]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges182]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges181]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges148]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges142]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges132]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges118]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges115]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges102]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges98]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges88]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges85]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges82]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges45]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges2]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes6]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes2]
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_init[0.0]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1435]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1426]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1327]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1142]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1036]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges716]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges639]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges611]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges379]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges314]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges194]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges170]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges145]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges114]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges94]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges22]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges8]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1435]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1390]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1367]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1362]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1331]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1320]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1292]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1291]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1290]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1273]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1255]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1209]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1202]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1191]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1150]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1128]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1123]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1063]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1004]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges999]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges949]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges942]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges940]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges916]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges877]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges858]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges836]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges832]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges776]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges772]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges751]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges745]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges742]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges729]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges721]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges708]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges702]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges692]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges686]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges662]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges646]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges639]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges590]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges575]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges513]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges468]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges390]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges343]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges328]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges229]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges224]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges212]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges204]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges188]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges173]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges167]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges158]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges147]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges134]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges104]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges92]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges86]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges76]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges74]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges59]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges51]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges50]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges46]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges33]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges32]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges19]
0.00s teardown tests/optim/test_optim.py::test_autowrap[Adam]
0.00s teardown tests/infer/test_inference.py::NormalNormalTests::test_rws_nonreparameterized
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes7]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes6]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes8]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes7]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes14]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes13]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes4]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes1]
0.00s teardown tests/contrib/epidemiology/test_util.py::test_cat2_scalar[()]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes4]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1380]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1134]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges810]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges759]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges707]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges534]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges302]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges288]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges53]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges44]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1413]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1409]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1399]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1313]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1302]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1279]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1257]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1223]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1120]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1089]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1079]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1024]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges993]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges972]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges971]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges964]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges936]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges886]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges862]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges793]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges699]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges683]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges587]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges511]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges449]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges417]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges409]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges347]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges303]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges269]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges257]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges227]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges169]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges133]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges116]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges99]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges80]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges75]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges62]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges41]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges25]
0.00s teardown tests/optim/test_optim.py::test_autowrap[DCTAdam]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_structured_mass
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes4]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes14]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes13]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_error[shapes0]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes7]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes6]
0.00s teardown tests/distributions/test_delta.py::test_expand[batch_shape2]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes7]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1035]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges175]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1391]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1306]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1286]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1232]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1094]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1065]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1043]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1022]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges931]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges813]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges798]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges757]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges747]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges724]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges660]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges577]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges532]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges520]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges496]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges222]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges202]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges179]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges91]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges73]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges67]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges64]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges49]
0.00s teardown tests/optim/test_optim.py::test_autowrap[SGD]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes10]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes2]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes11]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes9]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes11]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1437]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1375]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1203]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges1035]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges829]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges512]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges183]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges113]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges112]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges107]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges71]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges69]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges38]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges36]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes11]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes9]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes3]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes10]
0.00s teardown tests/contrib/forecast/test_evaluate.py::test_custom_warm_start
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes12]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes8]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[rsvi-JitTrace_ELBO]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges108]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges976]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges894]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges837]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges720]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges615]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges101]
0.00s teardown tests/infer/test_inference.py::NormalNormalTests::test_renyi_nonreparameterized
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict_error[shapes5]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes4]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes3]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes9]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes6]
0.00s teardown tests/poutine/test_trace_struct.py::test_connectivity_on_removal[edges1032]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges68]
0.00s teardown tests/optim/test_optim.py::test_autowrap[RMSprop]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes1]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes10]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes5]
0.00s teardown tests/poutine/test_trace_struct.py::test_topological_sort[edges553]
0.00s teardown tests/ops/test_packed.py::test_broadcast_all[shapes0]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape_strict[shapes12]
0.00s teardown tests/distributions/test_util.py::test_broadcast_shape[shapes8]
0.00s call     tests/optim/test_multi.py::test_multi_optimizer_disjoint_ok
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes8]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes3]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes2]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[nonreparam-JitTrace_ELBO]
0.00s call     tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[SparseGPRegression_y1D]
0.00s teardown tests/contrib/gp/test_models.py::test_inference_sgpr
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes5]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes9]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes1]
0.00s teardown tests/infer/test_enum.py::test_backwardsample_posterior_2
0.00s setup    tests/test_primitives.py::test_deterministic_ok
0.00s call     tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[GPRegression_y2D]
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_nonreparameterized
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes7]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes1]
0.00s teardown tests/infer/test_inference.py::NormalNormalTests::test_rws_reparameterized
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes3]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape_strict[shapes0]
0.00s call     tests/distributions/test_util.py::test_broadcast_shape[shapes6]
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_reparameterized_vectorized
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=3 --tmc --tmc-num-samples=2 --cuda]
0.00s setup    tests/infer/mcmc/test_nuts.py::test_nuts_conjugate_gaussian[dim=10_chain-len=3_num_obs=1]
0.00s teardown tests/infer/test_valid_models.py::test_enum_in_model_multi_scale_error
0.00s setup    tests/test_primitives.py::test_observe_warn
0.00s call     tests/ops/test_indexing.py::test_index[args7-expected7]
0.00s setup    tests/test_examples.py::test_cuda[dmm/dmm.py --num-epochs=1 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --cuda]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_gamma_normal
0.00s setup    tests/test_primitives.py::test_param_ok
0.00s teardown tests/infer/test_inference.py::PoissonGammaTests::test_rws_reparameterized
0.00s teardown tests/infer/test_inference.py::NormalNormalTests::test_elbo_tail_adaptive
0.00s teardown tests/infer/test_inference.py::PoissonGammaTests::test_mmd_vectorized
0.00s setup    tests/test_examples.py::test_cuda[air/main.py --num-steps=1 --cuda]
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_rws_reparameterized_vectorized
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_reparameterized
0.00s setup    tests/distributions/test_cuda.py::test_sample[Uniform]
0.00s teardown tests/infer/test_inference.py::PoissonGammaTests::test_rws_nonreparameterized
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_rws_nonreparameterized
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_rws_reparameterized
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_reparameterized
0.00s teardown tests/infer/mcmc/test_valid_models.py::test_log_prob_eval_iterates_in_correct_order
0.00s teardown tests/infer/test_enum.py::test_elbo_enumerate_plate_5
0.00s setup    tests/infer/mcmc/test_hmc.py::test_hmc_conjugate_gaussian[dim=10_chain-len=3_num_obs=1]
0.00s teardown tests/params/test_param.py::ParamStoreDictTests::test_save_and_load
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_reentrant
0.00s call     tests/ops/test_indexing.py::test_index[args6-expected6]
0.00s teardown tests/infer/test_valid_models.py::test_plate_enum_discrete_batch_ok
0.00s teardown tests/infer/test_inference.py::PoissonGammaTests::test_renyi_nonreparameterized
0.00s call     tests/optim/test_optim.py::test_autowrap[DCTAdam]
0.00s call     tests/optim/test_optim.py::test_autowrap[Adam]
0.00s teardown tests/infer/test_inference.py::PoissonGammaTests::test_elbo_nonreparameterized
0.00s call     tests/contrib/gp/test_models.py::test_inference_with_whiten[SparseGPRegression_y1D]
0.00s teardown tests/infer/test_inference.py::NormalNormalTests::test_mmd_vectorized
0.00s call     tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[SparseGPRegression_y2D]
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_nonreparameterized_vectorized
0.00s teardown tests/infer/test_inference.py::NormalNormalTests::test_mmd_nonvectorized
0.00s teardown tests/infer/test_inference.py::NormalNormalTests::test_elbo_nonreparameterized
0.00s teardown tests/contrib/gp/test_parameterized.py::test_inference
0.00s call     tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[GPRegression_y1D]
0.00s call     tests/ops/test_indexing.py::test_index[args5-expected5]
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_elbo_reparameterized_vectorized
0.00s teardown tests/infer/test_inference.py::NormalNormalTests::test_elbo_analytic_kl
0.00s call     tests/optim/test_optim.py::test_autowrap[ClippedAdam]
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_rws_nonreparameterized_vectorized
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_nonreparameterized
0.00s call     tests/ops/test_indexing.py::test_index[args8-expected8]
0.00s call     tests/ops/test_indexing.py::test_index[None-None]
0.00s teardown tests/infer/test_enum.py::test_backwardsample_posterior_3
0.00s call     tests/optim/test_optim.py::test_autowrap[SGD]
0.00s call     tests/ops/test_indexing.py::test_index[args9-expected9]
0.00s teardown tests/ops/test_stats.py::test_effective_sample_size
0.00s teardown tests/infer/test_enum.py::test_elbo_dbn_growth
0.00s call     tests/contrib/gp/test_conditional.py::test_conditional_whiten[6]
0.00s teardown tests/infer/test_inference.py::BernoulliBetaTests::test_renyi_nonreparameterized_vectorized
0.00s call     tests/ops/test_indexing.py::test_index[0-0]
0.00s teardown tests/poutine/test_poutines.py::QueueHandlerMixedTest::test_queue_single
0.00s call     tests/contrib/gp/test_conditional.py::test_conditional_whiten[1]
0.00s teardown tests/infer/test_inference.py::NormalNormalTests::test_renyi_reparameterized
0.00s call     tests/optim/test_optim.py::test_autowrap[RMSprop]
0.00s call     tests/contrib/gp/test_conditional.py::test_conditional_whiten[4]
0.00s teardown tests/infer/test_inference.py::PoissonGammaTests::test_renyi_reparameterized
0.00s teardown tests/infer/test_jit.py::test_multi_output
0.00s call     tests/ops/test_indexing.py::test_index[1-1]
0.00s call     tests/contrib/gp/test_models.py::test_inference_with_whiten[GPRegression_y1D]
0.00s teardown tests/nn/test_module.py::test_to_pyro_module_
0.00s teardown tests/infer/test_inference.py::PoissonGammaTests::test_elbo_reparameterized
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_block_autoregressive_jacobians
0.00s call     tests/ops/test_indexing.py::test_index[args4-expected4]
0.00s call     tests/contrib/gp/test_models.py::test_inference_with_whiten[SparseGPRegression_y2D]
0.00s teardown tests/infer/test_inference.py::test_reparam_stable
0.00s teardown tests/infer/test_inference.py::NormalNormalTests::test_elbo_reparameterized
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_sylvester
0.00s teardown tests/contrib/tracking/test_ekf.py::test_EKFState_with_NcvContinuous
0.00s call     tests/contrib/gp/test_models.py::test_inference[VariationalSparseGP_y1D]
0.00s call     tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[GPRegression_y1D]
0.00s teardown tests/infer/test_enum.py::test_elbo_hmm_growth
0.00s teardown tests/contrib/tracking/test_dynamic_models.py::test_NcpContinuous
0.00s teardown tests/nn/test_module.py::test_mixin_factory
0.00s call     tests/contrib/gp/test_models.py::test_inference_with_whiten[GPRegression_y2D]
0.00s call     tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[SparseGPRegression_y1D]
0.00s call     tests/contrib/gp/test_models.py::test_forward_with_empty_latent_shape[GPRegression_y2D]
0.00s teardown tests/distributions/test_one_hot_categorical.py::TestOneHotCategorical::test_support_non_vectorized
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_conditional_spline
0.00s call     tests/ops/test_indexing.py::test_index[args3-expected3]
0.00s teardown tests/poutine/test_poutines.py::LiftHandlerTests::test_prior_dict
0.00s teardown tests/infer/mcmc/test_mcmc_util.py::test_init_to_value
0.00s teardown tests/contrib/randomvariable/test_random_variable.py::test_add
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_serialize
0.00s teardown tests/poutine/test_poutines.py::LiftHandlerTests::test_random_module_warn
0.00s teardown tests/infer/test_abstract_infer.py::test_nesting
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_affine_autoregressive
0.00s teardown tests/distributions/test_lowrank_mvn.py::test_variance
0.00s call     tests/contrib/gp/test_models.py::test_inference_with_empty_latent_shape[SparseGPRegression_y2D]
0.00s teardown tests/poutine/test_poutines.py::LiftHandlerTests::test_unlifted_param
0.00s teardown tests/ops/test_stats.py::test_weighted_waic
0.00s teardown tests/infer/reparam/test_conjugate.py::test_beta_binomial_hmc
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_nested
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_single_single_ok
0.00s teardown tests/contrib/gp/test_kernels.py::test_active_dims_disjoint_ok
0.00s teardown tests/poutine/test_poutines.py::LiftHandlerTests::test_random_module
0.00s teardown tests/perf/test_benchmark.py::test_benchmark[PoissonGamma::reparam=True_TraceGraph]
0.00s teardown tests/infer/test_valid_models.py::test_plate_shape_broadcasting
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_single_ok
0.00s teardown tests/distributions/test_categorical.py::TestCategorical::test_support_non_vectorized
0.00s teardown tests/contrib/tracking/test_ekf.py::test_EKFState_with_NcpContinuous
0.00s teardown tests/test_examples.py::test_cuda[einsum.py --cuda]
0.00s teardown tests/poutine/test_poutines.py::test_trace_log_prob_err_msg
0.00s teardown tests/optim/test_optim.py::OptimTests::test_per_param_optim
0.00s teardown tests/infer/test_smcfilter.py::test_likelihood_ratio
0.00s teardown tests/distributions/test_one_hot_categorical.py::TestOneHotCategorical::test_support
0.00s setup    tests/distributions/test_torch_patch.py::test_dirichlet_grad_cuda
0.00s teardown tests/poutine/test_poutines.py::LiftHandlerTests::test_random_module_prior_dict
0.00s teardown tests/distributions/test_lowrank_mvn.py::test_log_prob
0.00s teardown tests/contrib/tracking/test_dynamic_models.py::test_NcvContinuous
0.00s teardown tests/ops/test_stats.py::test_cummin
0.00s teardown tests/optim/test_optim.py::test_dctadam_param_subsample
0.00s teardown tests/distributions/test_lowrank_mvn.py::test_scale_tril
0.00s teardown tests/distributions/test_categorical.py::TestCategorical::test_log_prob_sum
0.00s teardown tests/poutine/test_poutines.py::test_plate_error_on_enter
0.00s teardown tests/params/test_param.py::test_dict_interface
0.00s teardown tests/infer/test_valid_models.py::test_mean_field_warn
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[reparam-JitTraceGraph_ELBO]
0.00s teardown tests/infer/reparam/test_conjugate.py::test_beta_binomial_dependent_sample
0.00s teardown tests/contrib/tracking/test_dynamic_models.py::test_NcpDiscrete
0.00s teardown tests/distributions/test_categorical.py::test_view_reshape_bug
0.00s teardown tests/poutine/test_poutines.py::LiftHandlerTests::test_splice
0.00s teardown tests/infer/reparam/test_conjugate.py::test_beta_binomial_elbo
0.00s teardown tests/infer/test_smcfilter.py::test_gaussian_filter
0.00s teardown tests/contrib/tracking/test_assignment.py::test_sparse_smoke
0.00s teardown tests/infer/test_valid_models.py::test_tail_adaptive_error
0.00s teardown tests/infer/test_valid_models.py::test_tail_adaptive_ok
0.00s teardown tests/infer/test_valid_models.py::test_enum_discrete_iplate_single_ok
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_neural_autoregressive
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_chain
0.00s teardown tests/infer/test_inference.py::test_sequential_plating_sum
0.00s teardown tests/infer/test_jit.py::test_scatter_workaround
0.00s teardown tests/infer/test_enum.py::test_elbo_scale
0.00s teardown tests/infer/test_autoguide.py::test_unpack_latent
0.00s teardown tests/nn/test_autoregressive.py::AutoRegressiveNNTests::test_jacobians
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_conditional_neural_autoregressive
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_conditional_affine_autoregressive
0.00s teardown tests/distributions/test_delta.py::TestDelta::test_log_prob_sum
0.00s teardown tests/contrib/test_util.py::test_rtril
0.00s teardown tests/poutine/test_nesting.py::test_nested_reset
0.00s teardown tests/infer/test_valid_models.py::test_reparam_stable
0.00s teardown tests/distributions/test_shapes.py::test_one_hot_categorical_log_prob_shape
0.00s teardown tests/nn/test_module.py::test_svi_smoke
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_spline_autoregressive
0.00s teardown tests/infer/test_valid_models.py::test_tail_adaptive_warning
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_plate
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_polynomial
0.00s teardown tests/distributions/test_delta.py::TestDelta::test_batch_log_prob
0.00s teardown tests/contrib/gp/test_kernels.py::test_transforming
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[rsvi-JitTraceEnum_ELBO]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[rsvi-JitTraceGraph_ELBO]
0.00s teardown tests/nn/test_module.py::test_bayesian_gru
0.00s teardown tests/test_examples.py::test_coverage
0.00s teardown tests/nn/test_module.py::test_sample
0.00s teardown tests/distributions/test_categorical.py::TestCategorical::test_support
0.00s setup    tests/distributions/test_cuda.py::test_sample[LKJCorrCholesky]
0.00s teardown tests/infer/test_csis.py::test_csis_validation_batch
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_overlapping_plates_ok
0.00s teardown tests/ops/test_stats.py::test_hpdi
0.00s teardown tests/contrib/tracking/test_dynamic_models.py::test_NcvDiscrete
0.00s teardown tests/poutine/test_poutines.py::LiftHandlerTests::test_memoize
0.00s teardown tests/infer/reparam/test_conjugate.py::test_beta_binomial_static_sample
0.00s teardown tests/infer/test_valid_models.py::test_mean_field_ok
0.00s teardown tests/contrib/tracking/test_measurements.py::test_PositionMeasurement
0.00s teardown tests/contrib/test_util.py::test_rdiag
0.00s teardown tests/ops/test_stats.py::test_quantile
0.00s teardown tests/infer/test_valid_models.py::test_enum_recycling_interleave
0.00s teardown tests/distributions/test_empirical.py::test_mean_var_non_nan
0.00s teardown tests/infer/test_jit.py::test_simple
0.00s teardown tests/infer/test_inference.py::SafetyTests::test_duplicate_obs_name
0.00s teardown tests/infer/test_inference.py::TestFixedModelGuide::test_model_fixed
0.00s teardown tests/contrib/autoname/test_scoping.py::test_simple_recur
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.5-0.95-1.1]
0.00s teardown tests/ops/test_stats.py::test_pi
0.00s teardown tests/infer/test_valid_models.py::test_enum_in_model_plate_ok
0.00s teardown tests/infer/test_inference.py::SafetyTests::test_extra_samples
0.00s teardown tests/distributions/test_delta.py::TestDelta::test_mean_and_var
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Multinomial]
0.00s teardown tests/poutine/test_poutines.py::TraceHandlerTests::test_trace_return
0.00s teardown tests/infer/test_jit.py::test_backward
0.00s teardown tests/contrib/easyguide/test_easyguide.py::test_overlapping_plates_error
0.00s teardown tests/infer/test_valid_models.py::test_enum_in_model_ok
0.00s teardown tests/contrib/oed/test_ewma.py::test_ewma_log_with_s
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.9-0.95-0.4]
0.00s teardown tests/ops/test_stats.py::test_split_gelman_rubin_agree_with_gelman_rubin
0.00s teardown tests/ops/test_stats.py::test_autocovariance
0.00s teardown tests/infer/test_valid_models.py::test_three_indep_plate_at_different_depths_ok
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.9-0.6-0.4]
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[nonreparam-JitTraceGraph_ELBO]
0.00s teardown tests/poutine/test_poutines.py::UnconditionHandlerTests::test_undo_uncondition
0.00s teardown tests/poutine/test_poutines.py::ReplayHandlerTests::test_replay_full_repeat
0.00s teardown tests/ops/test_special.py::test_safe_log
0.00s teardown tests/infer/test_csis.py::test_csis_parameter_update
0.00s teardown tests/contrib/tracking/test_assignment.py::test_dense_smoke
0.00s teardown tests/ops/test_stats.py::test_autocorrelation
0.00s teardown tests/distributions/test_reshape.py::test_independent_entropy
0.00s teardown tests/contrib/oed/test_ewma.py::test_ewma_log
0.00s teardown tests/infer/test_valid_models.py::test_enum_in_model_plate_reuse_ok
0.00s teardown tests/distributions/test_shapes.py::test_diag_normal_log_prob_shape
0.00s teardown tests/distributions/test_delta.py::TestDelta::test_batch_log_prob_shape
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_binomial
0.00s teardown tests/contrib/test_hessian.py::test_hessian_multi_variables
0.00s teardown tests/poutine/test_poutines.py::ConditionHandlerTests::test_trace_data
0.00s teardown tests/nn/test_module.py::test_torch_serialize_decorators
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_conditional_planar
0.00s teardown tests/distributions/test_mask.py::test_kl_divergence
0.00s teardown tests/contrib/gp/test_parameterized.py::test_parameterized
0.00s teardown tests/ops/test_stats.py::test_gelman_rubin
0.00s teardown tests/contrib/tracking/test_hashing.py::test_merge_points_small
0.00s teardown tests/contrib/gp/test_kernels.py::test_combination
0.00s teardown tests/contrib/epidemiology/test_distributions.py::test_relaxed_beta_binomial
0.00s teardown tests/poutine/test_poutines.py::BlockHandlerTests::test_block_tutorial_case
0.00s teardown tests/poutine/test_mapdata.py::test_nested_iplate
0.00s teardown tests/nn/test_module.py::test_torch_serialize_attributes
0.00s teardown tests/infer/test_valid_models.py::test_no_plate_enum_discrete_batch_error
0.00s teardown tests/poutine/test_poutines.py::IndirectLambdaHandlerTests::test_graph_structure
0.00s teardown tests/poutine/test_poutines.py::QueueHandlerMixedTest::test_queue_enumerate
0.00s teardown tests/infer/test_util.py::test_multi_frame_tensor
0.00s teardown tests/infer/test_jit.py::test_scale_and_mask
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[reparam-JitTraceEnum_ELBO]
0.00s teardown tests/poutine/test_poutines.py::test_arg_kwarg_error
0.00s teardown tests/nn/test_module.py::test_names
0.00s teardown tests/nn/test_autoregressive.py::AutoRegressiveNNTests::test_masks
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_leaky_relu
0.00s setup    tests/infer/test_inference.py::test_exponential_gamma[nonreparam-JitTraceEnum_ELBO]
0.00s teardown tests/poutine/test_poutines.py::test_decorator_interface_queue
0.00s teardown tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full_expose
0.00s teardown tests/infer/test_inference.py::TestFixedModelGuide::test_guide_fixed
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_spline_coupling
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_conditional_affine_coupling
0.00s teardown tests/contrib/randomvariable/test_random_variable.py::test_chaining
0.00s setup    tests/distributions/test_cuda.py::test_sample[Exponential]
0.00s teardown tests/poutine/test_poutines.py::test_trace_score_parts_err_msg
0.00s teardown tests/poutine/test_poutines.py::UnconditionHandlerTests::test_uncondition
0.00s teardown tests/poutine/test_poutines.py::TraceHandlerTests::test_trace_full
0.00s teardown tests/nn/test_module.py::test_module_cache
0.00s teardown tests/distributions/test_gaussian_mixtures.py::test_gsm_log_prob
0.00s setup    tests/poutine/test_mapdata.py::test_cuda[plate-5]
0.00s teardown tests/poutine/test_poutines.py::ReplayHandlerTests::test_replay_full
0.00s teardown tests/nn/test_module.py::test_cache
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Gamma]
0.00s teardown tests/optim/test_multi.py::test_multi_optimizer_overlap_error
0.00s teardown tests/infer/test_inference.py::TestFixedModelGuide::test_guide_and_model_both_fixed
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_planar
0.00s teardown tests/distributions/test_shapes.py::test_one_hot_categorical_shape
0.00s teardown tests/poutine/test_poutines.py::test_method_decorator_interface_condition
0.00s teardown tests/poutine/test_poutines.py::EscapeHandlerTests::test_trace_compose
0.00s teardown tests/poutine/test_poutines.py::ConditionHandlerTests::test_condition
0.00s teardown tests/poutine/test_poutines.py::QueueHandlerDiscreteTest::test_queue_enumerate
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_discrete_cosine
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_affine_coupling
0.00s teardown tests/distributions/test_shapes.py::test_normal_log_prob_shape
0.00s teardown tests/contrib/randomvariable/test_random_variable.py::test_subtract
0.00s teardown tests/contrib/gp/test_kernels.py::test_active_dims_overlap_ok
0.00s teardown tests/contrib/test_util.py::test_tensor_to_dict
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.9-0.6-1.1]
0.00s teardown tests/poutine/test_poutines.py::test_decorator_interface_primitives
0.00s teardown tests/poutine/test_poutines.py::IndirectLambdaHandlerTests::test_scale_factors
0.00s teardown tests/perf/test_benchmark.py::test_benchmark[PoissonGamma::reparam=False_TraceGraph]
0.00s teardown tests/ops/test_jit.py::test_varying_kwargs
0.00s teardown tests/infer/test_valid_models.py::test_enum_sequential_in_model_error
0.00s teardown tests/infer/test_jit.py::test_scatter
0.00s teardown tests/infer/test_inference.py::TestFixedModelGuide::test_guide_and_model_free
0.00s teardown tests/contrib/randomvariable/test_random_variable.py::test_tensor_ops
0.00s teardown tests/contrib/autoname/test_scoping.py::test_recur_multi
0.00s teardown tests/contrib/autoname/test_scoping.py::test_multi_nested
0.00s teardown tests/contrib/test_hessian.py::test_hessian_mvn
0.00s setup    tests/poutine/test_mapdata.py::test_cuda[plate-20]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[ShapeAugmentedDirichlet]
0.00s teardown tests/poutine/test_poutines.py::ConditionHandlerTests::test_stack_overwrite_behavior
0.00s teardown tests/poutine/test_poutines.py::TraceHandlerTests::test_trace_param_only
0.00s teardown tests/ops/test_jit.py::test_varying_unhashable_kwargs
0.00s teardown tests/nn/test_module.py::test_nested
0.00s teardown tests/distributions/test_shapes.py::test_categorical_log_prob_shape
0.00s call     tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full_hide_expose
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[ShapeAugmentedDirichlet]
0.00s teardown tests/distributions/test_shapes.py::test_bernoulli_log_prob_shape
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.5-0.6-1.1]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Bernoulli]
0.00s teardown tests/poutine/test_poutines.py::test_trace_log_prob_sum_err_msg
0.00s teardown tests/poutine/test_poutines.py::InferConfigHandlerTests::test_infer_config_sample
0.00s teardown tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full
0.00s teardown tests/perf/test_benchmark.py::test_benchmark[PoissonGamma::reparam=False_Trace]
0.00s teardown tests/infer/test_valid_models.py::test_plate_wrong_size_error
0.00s teardown tests/infer/test_jit.py::test_jit_arange_workaround
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_spline
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_haar_transform
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_batchnorm
0.00s teardown tests/distributions/test_distributions.py::test_expand_enumerate_support
0.00s teardown tests/contrib/randomvariable/test_random_variable.py::test_neg
0.00s teardown tests/poutine/test_poutines.py::BlockHandlerTests::test_block_partial_expose
0.00s teardown tests/poutine/test_poutines.py::BlockHandlerTests::test_block_hide_fn
0.00s teardown tests/nn/test_module.py::test_clear
0.00s teardown tests/infer/test_jit.py::test_masked_fill
0.00s teardown tests/distributions/test_shapes.py::test_categorical_shape
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[ShapeAugmentedGamma]
0.00s teardown tests/perf/test_benchmark.py::test_benchmark[PoissonGamma::reparam=True_Trace]
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_conditional_radial
0.00s teardown tests/test_util.py::test_warn_if_inf
0.00s setup    tests/distributions/test_cuda.py::test_rsample[NaiveDirichlet]
0.00s teardown tests/poutine/test_poutines.py::BlockHandlerTests::test_block_partial_hide
0.00s teardown tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full_hide
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_conditional_householder
0.00s teardown tests/distributions/test_shapes.py::test_normal_shape
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.5-0.95-0.4]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Binomial]
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_radial
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_permute
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_lower_cholesky_affine
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_householder
0.00s teardown tests/distributions/test_transforms.py::TransformTests::test_elu
0.00s teardown tests/contrib/randomvariable/test_random_variable.py::test_abs
0.00s setup    tests/infer/mcmc/test_nuts.py::test_nuts_conjugate_gaussian[dim=10_chain-len=4_num_obs=1]
0.00s teardown tests/nn/test_module.py::test_hierarchy_prior_cached
0.00s teardown tests/contrib/gp/test_parameterized.py::test_nested_parameterized
0.00s setup    tests/distributions/test_cuda.py::test_rsample[BetaBinomial]
0.00s teardown tests/contrib/randomvariable/test_random_variable.py::test_pow
0.00s setup    tests/infer/mcmc/test_nuts.py::test_nuts_conjugate_gaussian[dim=5_chain-len=2_num_obs=10000]
0.00s teardown tests/poutine/test_poutines.py::BlockHandlerTests::test_block_expose_fn
0.00s teardown tests/ops/test_jit.py::test_varying_len_args
0.00s teardown tests/distributions/test_util.py::test_sum_leftmost
0.00s teardown tests/distributions/test_reshape.py::test_sample_shape_order
0.00s setup    tests/infer/mcmc/test_hmc.py::test_hmc_conjugate_gaussian[dim=5_chain-len=9_num_obs=1]
0.00s teardown tests/distributions/test_shapes.py::test_zip_shape
0.00s teardown tests/contrib/autoname/test_scoping.py::test_nested_traces
0.00s teardown tests/test_util.py::test_warn_if_nan
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.5-0.6-0.4]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Categorical]
0.00s teardown tests/infer/test_initialization.py::test_init_to_generated
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_overwrite
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.9-0.95-1.1]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.5-0.95-0.4]
0.00s teardown tests/poutine/test_poutines.py::EscapeHandlerTests::test_discrete_escape
0.00s teardown tests/ops/test_tensor_utils.py::test_next_fast_len
0.00s teardown tests/distributions/test_util.py::test_sum_rightmost
0.00s teardown tests/contrib/autoname/test_named.py::test_eval_str
0.00s teardown tests/contrib/test_util.py::test_rexpand
0.00s teardown tests/test_primitives.py::test_deterministic_ok
0.00s teardown tests/contrib/autoname/test_scoping.py::test_no_param
0.00s teardown tests/contrib/test_util.py::test_get_indices_sizes
0.00s setup    tests/distributions/test_torch_patch.py::test_linspace
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Gamma]
0.00s teardown tests/nn/test_module.py::test_delete
0.00s teardown tests/distributions/test_shapes.py::test_dirichlet_shape
0.00s teardown tests/contrib/autoname/test_named.py::test_named_list
0.00s teardown tests/contrib/autoname/test_named.py::test_named_object
0.00s setup    tests/distributions/test_cuda.py::test_sample[ShapeAugmentedBeta]
0.00s teardown tests/contrib/autoname/test_named.py::test_named_dict
0.00s teardown tests/contrib/test_util.py::test_lexpand
0.00s setup    tests/infer/mcmc/test_hmc.py::test_hmc_conjugate_gaussian[dim=5_chain-len=2_num_obs=100]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.5-0.95-1.1]
0.00s setup    tests/test_examples.py::test_cuda[contrib/epidemiology/sir.py --nojit -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 --haar --cuda]
0.00s teardown tests/poutine/test_poutines.py::ConditionHandlerTests::test_stack_success
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.9-0.95-0.4]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.9-0.6-1.1]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Uniform]
0.00s teardown tests/perf/test_benchmark.py::test_benchmark[VSGP::MultiClass_whiten=False]
0.00s teardown tests/perf/test_benchmark.py::test_benchmark[BernoulliBeta::NUTS]
0.00s teardown tests/infer/test_autoguide.py::test_empty_model_error
0.00s teardown tests/contrib/autoname/test_scoping.py::test_only_withs
0.00s teardown tests/contrib/autoname/test_named.py::test_nested
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.5-0.6-1.1]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[NaiveDirichlet]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[BetaBinomial]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[NaiveDirichlet]
0.00s setup    tests/poutine/test_mapdata.py::test_cuda[iplate-5]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[OneHotCategorical]
0.00s teardown tests/distributions/test_cuda.py::test_sample[FoldedNormal]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.9-0.6-0.4]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Normal]
0.00s teardown tests/distributions/test_util.py::test_weakmethod
0.00s teardown tests/distributions/test_cuda.py::test_rsample[VonMises]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[LogNormal]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Poisson]
0.00s teardown tests/distributions/test_cuda.py::test_sample[RejectionExponential]
0.00s teardown tests/contrib/autoname/test_scoping.py::test_basic_scope
0.00s teardown tests/contrib/autoname/test_scoping.py::test_mutual_recur
0.00s setup    tests/infer/mcmc/test_nuts.py::test_nuts_conjugate_gaussian[dim=5_chain-len=9_num_obs=1]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[ShapeAugmentedGamma]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=0 --cuda]
0.00s teardown tests/infer/test_inference.py::SafetyTests::test_duplicate_names
0.00s teardown tests/contrib/tracking/test_hashing.py::test_lsh_remove
0.00s setup    tests/poutine/test_mapdata.py::test_cuda[iplate-20]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[LKJCorrCholesky]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Cauchy]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[FoldedNormal]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[ShapeAugmentedBeta]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=5 --cuda]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[VonMises]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[HalfCauchy]
0.00s setup    tests/distributions/test_cuda.py::test_sample[FoldedNormal]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Beta]
0.00s teardown tests/nn/test_module.py::test_submodule_contains_torch_module
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Categorical]
0.00s setup    tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.9-0.95-1.1]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[BetaBinomial]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[MultivariateNormal]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Gamma]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[DirichletMultinomial]
0.00s setup    tests/distributions/test_cuda.py::test_sample[RejectionExponential]
0.00s setup    tests/test_examples.py::test_cuda[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -p=10000 --sequential --cuda]
0.00s teardown tests/optim/test_multi.py::test_multi_optimizer_disjoint_ok
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Bernoulli]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[LowRankMultivariateNormal]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[NaiveBeta]
0.00s setup    tests/distributions/test_cuda.py::test_sample[HalfCauchy]
0.00s setup    tests/distributions/test_cuda.py::test_sample[MultivariateNormal]
0.00s setup    tests/distributions/test_cuda.py::test_sample[LogNormal]
0.00s setup    tests/distributions/test_cuda.py::test_sample[NaiveBeta]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=0 --tmc --tmc-num-samples=2 --cuda]
0.00s teardown tests/poutine/test_poutines.py::EscapeHandlerTests::test_all_escape
0.00s setup    tests/distributions/test_cuda.py::test_rsample[ExtendedBinomial]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Multinomial]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Poisson]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=7 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --haar --cuda]
0.00s setup    tests/test_examples.py::test_cuda[contrib/cevae/synthetic.py --num-epochs=1 --cuda]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Geometric]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[LKJCorrCholesky]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Cauchy]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Dirichlet]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[MultivariateStudentT]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Beta]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[ShapeAugmentedGamma]
0.00s setup    tests/distributions/test_cuda.py::test_sample[GammaPoisson]
0.00s setup    tests/distributions/test_cuda.py::test_sample[DirichletMultinomial]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Cauchy]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Dirichlet]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Normal]
0.00s setup    tests/test_examples.py::test_cuda[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -m=1 --enum --cuda]
0.00s setup    tests/test_examples.py::test_cuda[contrib/epidemiology/sir.py --nojit -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -nb=16 --cuda]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.5-0.95-1.1]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Stable]
0.00s setup    tests/distributions/test_cuda.py::test_sample[VonMises]
0.00s setup    tests/test_examples.py::test_cuda[sir_hmc.py -t=2 -w=2 -n=4 -d=100 -p=10000 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[einsum.py --cuda]
0.00s teardown tests/test_primitives.py::test_param_ok
0.00s teardown tests/test_primitives.py::test_sample_ok
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[GammaPoisson]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Stable]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Categorical]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[LogNormal]
0.00s setup    tests/distributions/test_cuda.py::test_sample[ExtendedBinomial]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Multinomial]
0.00s setup    tests/distributions/test_cuda.py::test_sample[NaiveDirichlet]
0.00s setup    tests/distributions/test_cuda.py::test_sample[LowRankMultivariateNormal]
0.00s setup    tests/distributions/test_cuda.py::test_sample[ShapeAugmentedGamma]
0.00s setup    tests/test_examples.py::test_cuda[vae/vae.py --num-epochs=1 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=2 --tmc --tmc-num-samples=2 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[dmm/dmm.py --num-epochs=1 --num-iafs=1 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[contrib/gp/sv-dkl.py --epochs=1 --num-inducing=4 --cuda]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[HalfCauchy]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[VonMises]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[FoldedNormal]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[ShapeAugmentedBeta]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Beta]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[GammaPoisson]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[ExtendedBetaBinomial]
0.00s setup    tests/distributions/test_cuda.py::test_sample[MultivariateStudentT]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=6 --tmc --tmc-num-samples=2 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=1 --tmc --tmc-num-samples=2 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=2 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[dmm/dmm.py --num-epochs=1 --tmc --tmc-num-samples=2 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[contrib/epidemiology/sir.py --nojit -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[baseball.py --num-samples=200 --warmup-steps=100 --num-chains=2 --cuda]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[ShapeAugmentedBeta]
0.00s teardown tests/test_primitives.py::test_observe_warn
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Binomial]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[MultivariateStudentT]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[RejectionExponential]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Dirichlet]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Normal]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[RejectionExponential]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Geometric]
0.00s setup    tests/distributions/test_cuda.py::test_sample[OneHotCategorical]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Bernoulli]
0.00s setup    tests/test_examples.py::test_cuda[vae/ss_vae_M2.py --num-epochs=1 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=6 --cuda --raftery-parameterization]
0.00s teardown tests/test_examples.py::test_cuda[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=parallel --cuda]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Poisson]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[DirichletMultinomial]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Exponential]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[LowRankMultivariateNormal]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Uniform]
0.00s setup    tests/distributions/test_cuda.py::test_sample[ExtendedBetaBinomial]
0.00s setup    tests/distributions/test_cuda.py::test_sample[ShapeAugmentedDirichlet]
0.00s setup    tests/test_examples.py::test_cuda[vae/ss_vae_M2.py --num-epochs=1 --aux-loss --cuda]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=1 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[lkj.py --n=50 --num-chains=1 --warmup-steps=100 --num-samples=200 --cuda]
0.00s teardown tests/poutine/test_poutines.py::BlockHandlerTests::test_block_full_hide_expose
0.00s teardown tests/distributions/test_cuda.py::test_sample[LowRankMultivariateNormal]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[ExtendedBetaBinomial]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Geometric]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Exponential]
0.00s setup    tests/distributions/test_cuda.py::test_sample[Binomial]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=3 --cuda]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[LogNormal]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Poisson]
0.00s setup    tests/test_examples.py::test_cuda[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential --cuda]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=5 --tmc --tmc-num-samples=2 --cuda]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=6 --cuda]
0.00s teardown tests/perf/test_benchmark.py::test_benchmark[VSGP::MultiClass_whiten=True]
0.00s teardown tests/perf/test_benchmark.py::test_benchmark[BernoulliBeta::HMC]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[Categorical]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[ExtendedBinomial]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[NaiveBeta]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[Stable]
0.00s setup    tests/distributions/test_cuda.py::test_log_prob[HalfCauchy]
0.00s teardown tests/distributions/test_cuda.py::test_sample[NaiveDirichlet]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[OneHotCategorical]
0.00s setup    tests/distributions/test_cuda.py::test_sample[BetaBinomial]
0.00s setup    tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=4 --tmc --tmc-num-samples=2 --cuda]
0.00s setup    tests/distributions/test_cuda.py::test_rsample[MultivariateNormal]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.9-0.6-0.4]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=2 --cuda]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Multinomial]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_nuts_conjugate_gaussian[dim=10_chain-len=3_num_obs=1]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_nuts_conjugate_gaussian[dim=5_chain-len=9_num_obs=1]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[NaiveDirichlet]
0.00s teardown tests/poutine/test_mapdata.py::test_cuda[plate-20]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_nuts_conjugate_gaussian[dim=5_chain-len=2_num_obs=10000]
0.00s teardown tests/distributions/test_cuda.py::test_sample[LogNormal]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[ShapeAugmentedDirichlet]
0.00s teardown tests/distributions/test_torch_patch.py::test_linspace
0.00s teardown tests/distributions/test_cuda.py::test_rsample[BetaBinomial]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Bernoulli]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_hmc_conjugate_gaussian[dim=10_chain-len=4_num_obs=1]
0.00s teardown tests/infer/mcmc/test_nuts.py::test_nuts_conjugate_gaussian[dim=10_chain-len=4_num_obs=1]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_hmc_conjugate_gaussian[dim=10_chain-len=3_num_obs=1]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.9-0.95-0.4]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[ShapeAugmentedDirichlet]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Cauchy]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Gamma]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Uniform]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Geometric]
0.00s teardown tests/distributions/test_torch_patch.py::test_dirichlet_grad_cuda
0.00s teardown tests/test_examples.py::test_cuda[air/main.py --num-steps=1 --cuda]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Dirichlet]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Bernoulli]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.9-0.95-0.4]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.5-0.6-0.4]
0.00s teardown tests/test_examples.py::test_cuda[contrib/cevae/synthetic.py --num-epochs=1 --cuda]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.5-0.6-1.1]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.9-0.6-1.1]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[HalfCauchy]
0.00s teardown tests/test_examples.py::test_cuda[vae/ss_vae_M2.py --num-epochs=1 --aux-loss --cuda]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=5 --tmc --tmc-num-samples=2 --cuda]
0.00s teardown tests/poutine/test_mapdata.py::test_cuda[iplate-5]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.5-0.95-0.4]
0.00s teardown tests/distributions/test_cuda.py::test_sample[OneHotCategorical]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=0 --tmc --tmc-num-samples=2 --cuda]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.9-0.6-1.1]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[DirichletMultinomial]
0.00s teardown tests/test_examples.py::test_cuda[vae/vae.py --num-epochs=1 --cuda]
0.00s teardown tests/test_examples.py::test_cuda[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -p=10000 --sequential --cuda]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_hmc_conjugate_gaussian[dim=5_chain-len=2_num_obs=100]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.5-0.95-1.1]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.5-0.6-1.1]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[ExtendedBetaBinomial]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[ExtendedBinomial]
0.00s teardown tests/distributions/test_cuda.py::test_sample[GammaPoisson]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Bernoulli]
0.00s teardown tests/distributions/test_cuda.py::test_sample[ShapeAugmentedBeta]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=2 --tmc --tmc-num-samples=2 --cuda]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=1 --tmc --tmc-num-samples=2 --cuda]
0.00s teardown tests/test_examples.py::test_cuda[dmm/dmm.py --num-epochs=1 --tmcelbo --tmc-num-samples=2 --cuda]
0.00s teardown tests/poutine/test_mapdata.py::test_cuda[iplate-20]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.5-0.95-0.4]
0.00s teardown tests/test_examples.py::test_cuda[dmm/dmm.py --num-epochs=1 --tmc --tmc-num-samples=2 --cuda]
0.00s teardown tests/test_examples.py::test_cuda[contrib/epidemiology/sir.py --nojit -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 -nb=16 --cuda]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.9-0.95-1.1]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[OMTMultivariateNormal-3-0.9-0.6-0.4]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Poisson]
0.00s teardown tests/test_examples.py::test_cuda[vae/ss_vae_M2.py --num-epochs=1 --enum-discrete=sequential --cuda]
0.00s teardown tests/poutine/test_mapdata.py::test_cuda[plate-5]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Categorical]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[RejectionExponential]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[ExtendedBetaBinomial]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Binomial]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[LowRankMultivariateNormal]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[ShapeAugmentedGamma]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Geometric]
0.00s teardown tests/distributions/test_cuda.py::test_sample[ExtendedBinomial]
0.00s teardown tests/distributions/test_cuda.py::test_sample[LKJCorrCholesky]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Beta]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Exponential]
0.00s teardown tests/test_examples.py::test_cuda[dmm/dmm.py --num-epochs=1 --cuda]
0.00s teardown tests/test_examples.py::test_cuda[lkj.py --n=50 --num-chains=1 --warmup-steps=100 --num-samples=200 --cuda]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Beta]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[FoldedNormal]
0.00s teardown tests/distributions/test_cuda.py::test_sample[DirichletMultinomial]
0.00s teardown tests/distributions/test_cuda.py::test_sample[MultivariateStudentT]
0.00s teardown tests/distributions/test_cuda.py::test_sample[VonMises]
0.00s teardown tests/distributions/test_cuda.py::test_sample[MultivariateNormal]
0.00s teardown tests/test_examples.py::test_cuda[sir_hmc.py -t=2 -w=2 -n=4 -d=100 -p=10000 --cuda]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=4 --tmc --tmc-num-samples=2 --cuda]
0.00s teardown tests/infer/mcmc/test_hmc.py::test_hmc_conjugate_gaussian[dim=5_chain-len=9_num_obs=1]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[OneHotCategorical]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Multinomial]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Uniform]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[LKJCorrCholesky]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[RejectionExponential]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Normal]
0.00s teardown tests/test_examples.py::test_cuda[sir_hmc.py -t=2 -w=2 -n=4 -d=2 -m=1 --enum --cuda]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[LKJCorrCholesky]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Poisson]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Multinomial]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Categorical]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Binomial]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Cauchy]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=6 --tmc --tmc-num-samples=2 --cuda]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=3 --cuda]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=1 --cuda]
0.00s teardown tests/test_examples.py::test_cuda[contrib/epidemiology/regional.py --nojit -t=2 -w=2 -n=4 -r=3 -d=20 -p=1000 -f 2 --cuda]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[FoldedNormal]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[NaiveBeta]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Geometric]
0.00s teardown tests/distributions/test_cuda.py::test_sample[ExtendedBetaBinomial]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Stable]
0.00s teardown tests/distributions/test_cuda.py::test_sample[HalfCauchy]
0.00s teardown tests/distributions/test_cuda.py::test_sample[NaiveBeta]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=4 --cuda]
0.00s teardown tests/test_examples.py::test_cuda[contrib/epidemiology/sir.py --nojit -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 --cuda]
0.00s teardown tests/distributions/test_omt_mvn.py::test_mean_single_gradient[AVFMultivariateNormal-3-0.5-0.6-0.4]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Binomial]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Stable]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[LowRankMultivariateNormal]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[OneHotCategorical]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Stable]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[MultivariateNormal]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[NaiveBeta]
0.00s teardown tests/distributions/test_cuda.py::test_sample[ShapeAugmentedGamma]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=0 --cuda]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[GammaPoisson]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Normal]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[ShapeAugmentedBeta]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Exponential]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Exponential]
0.00s teardown tests/distributions/test_cuda.py::test_sample[Dirichlet]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=6 --cuda --raftery-parameterization]
0.00s teardown tests/test_examples.py::test_cuda[contrib/gp/sv-dkl.py --epochs=1 --num-inducing=4 --cuda]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[DirichletMultinomial]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[MultivariateStudentT]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[VonMises]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Dirichlet]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Beta]
0.00s teardown tests/distributions/test_cuda.py::test_sample[BetaBinomial]
0.00s teardown tests/distributions/test_cuda.py::test_sample[ShapeAugmentedDirichlet]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[ExtendedBinomial]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[MultivariateNormal]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[LogNormal]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[GammaPoisson]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[MultivariateStudentT]
0.00s teardown tests/test_examples.py::test_cuda[dmm/dmm.py --num-epochs=1 --num-iafs=1 --cuda]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Normal]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Gamma]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=5 --cuda]
0.00s teardown tests/distributions/test_cuda.py::test_log_prob[Cauchy]
0.00s teardown tests/distributions/test_cuda.py::test_rsample[Uniform]
0.00s teardown tests/test_examples.py::test_cuda[contrib/epidemiology/sir.py --nojit -t=2 -w=2 -n=4 -d=20 -p=1000 -f 2 --haar --cuda]
0.00s teardown tests/test_examples.py::test_cuda[vae/ss_vae_M2.py --num-epochs=1 --cuda]
0.00s teardown tests/test_examples.py::test_cuda[hmm.py --num-steps=1 --truncate=10 --model=6 --cuda]
=============================== warnings summary ===============================
/home/aryaman4/pyro/pyro/generic.py:9: DeprecationWarning: pyro.generic has moved to the pyroapi package
  warnings.warn("pyro.generic has moved to the pyroapi package", DeprecationWarning)

/home/aryaman4/pyro/pyro/infer/autoguide/guides.py:223: DeprecationWarning: The method `.add` has been deprecated in favor of `.append`.
  warnings.warn("The method `.add` has been deprecated in favor of `.append`.", DeprecationWarning)

/home/aryaman4/pyro/pyro/infer/autoguide/guides.py:223: DeprecationWarning: The method `.add` has been deprecated in favor of `.append`.
  warnings.warn("The method `.add` has been deprecated in favor of `.append`.", DeprecationWarning)

/home/aryaman4/pyro/pyro/infer/autoguide/guides.py:223: DeprecationWarning: The method `.add` has been deprecated in favor of `.append`.
  warnings.warn("The method `.add` has been deprecated in favor of `.append`.", DeprecationWarning)

/home/aryaman4/pyro/pyro/infer/predictive.py:205: DeprecationWarning: The method `.get_samples` has been deprecated in favor of `.forward`.
  DeprecationWarning)

/home/aryaman4/pyro/pyro/infer/predictive.py:205: DeprecationWarning: The method `.get_samples` has been deprecated in favor of `.forward`.
  DeprecationWarning)

/home/aryaman4/pyro/pyro/infer/predictive.py:205: DeprecationWarning: The method `.get_samples` has been deprecated in favor of `.forward`.
  DeprecationWarning)

/home/aryaman4/pyro/pyro/infer/predictive.py:205: DeprecationWarning: The method `.get_samples` has been deprecated in favor of `.forward`.
  DeprecationWarning)

/home/aryaman4/pyro/pyro/infer/predictive.py:205: DeprecationWarning: The method `.get_samples` has been deprecated in favor of `.forward`.
  DeprecationWarning)

/home/aryaman4/pyro/pyro/infer/predictive.py:205: DeprecationWarning: The method `.get_samples` has been deprecated in favor of `.forward`.
  DeprecationWarning)

/home/aryaman4/pyro/pyro/infer/predictive.py:205: DeprecationWarning: The method `.get_samples` has been deprecated in favor of `.forward`.
  DeprecationWarning)

/home/aryaman4/pyro/pyro/infer/predictive.py:205: DeprecationWarning: The method `.get_samples` has been deprecated in favor of `.forward`.
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

/home/aryaman4/pyro/pyro/ops/contract.py:411: DeprecationWarning: 'ubersum' is deprecated, use 'pyro.ops.contract.einsum' instead
  DeprecationWarning)

-- Docs: https://docs.pytest.org/en/latest/warnings.html
= 66 failed, 19696 passed, 763 skipped, 159 xfailed, 5 xpassed, 177 warnings, 8 error in 3856.65 seconds =
