Path	Lines of Code
examples/contrib/oed/ab_test.py	74
tests/__init__.py	4
tests/common.py	204
tests/conftest.py	72
tests/contrib/__init__.py	1
tests/contrib/autoguide/test_inference.py	144
tests/contrib/autoguide/test_mean_field_entropy.py	20
tests/contrib/autoname/test_named.py	77
tests/contrib/autoname/test_scoping.py	137
tests/contrib/bnn/test_hidden_layer.py	35
tests/contrib/cevae/test_cevae.py	54
tests/contrib/conftest.py	8
tests/contrib/easyguide/test_easyguide.py	186
tests/contrib/epidemiology/__init__.py	1
tests/contrib/epidemiology/test_distributions.py	183
tests/contrib/epidemiology/test_models.py	427
tests/contrib/epidemiology/test_quant.py	19
tests/contrib/epidemiology/test_util.py	33
tests/contrib/forecast/__init__.py	1
tests/contrib/forecast/test_evaluate.py	103
tests/contrib/forecast/test_forecaster.py	179
tests/contrib/forecast/test_util.py	103
tests/contrib/funsor/conftest.py	10
tests/contrib/funsor/test_enum_funsor.py	1139
tests/contrib/funsor/test_named_handlers.py	100
tests/contrib/funsor/test_pyroapi_funsor.py	13
tests/contrib/funsor/test_tmc.py	110
tests/contrib/funsor/test_valid_models_enum.py	315
tests/contrib/funsor/test_valid_models_plate.py	138
tests/contrib/funsor/test_valid_models_sequential_plate.py	85
tests/contrib/gp/__init__.py	1
tests/contrib/gp/test_conditional.py	70
tests/contrib/gp/test_kernels.py	143
tests/contrib/gp/test_likelihoods.py	125
tests/contrib/gp/test_models.py	326
tests/contrib/gp/test_parameterized.py	108
tests/contrib/oed/test_ewma.py	38
tests/contrib/oed/test_finite_spaces_eig.py	121
tests/contrib/oed/test_glmm.py	117
tests/contrib/oed/test_linear_models_eig.py	137
tests/contrib/oed/test_xexpx.py	10
tests/contrib/randomvariable/test_random_variable.py	61
tests/contrib/test_hessian.py	21
tests/contrib/test_minipyro.py	166
tests/contrib/test_util.py	62
tests/contrib/timeseries/test_gp.py	99
tests/contrib/timeseries/test_lgssm.py	54
tests/contrib/tracking/__init__.py	1
tests/contrib/tracking/test_assignment.py	248
tests/contrib/tracking/test_distributions.py	19
tests/contrib/tracking/test_dynamic_models.py	137
tests/contrib/tracking/test_ekf.py	64
tests/contrib/tracking/test_em.py	149
tests/contrib/tracking/test_hashing.py	123
tests/contrib/tracking/test_measurements.py	20
tests/distributions/__init__.py	1
tests/distributions/conftest.py	482
tests/distributions/dist_fixture.py	129
tests/distributions/test_binomial.py	38
tests/distributions/test_categorical.py	71
tests/distributions/test_coalescent.py	141
tests/distributions/test_conjugate.py	80
tests/distributions/test_conjugate_update.py	40
tests/distributions/test_cuda.py	67
tests/distributions/test_delta.py	55
tests/distributions/test_distributions.py	224
tests/distributions/test_empirical.py	134
tests/distributions/test_extended.py	64
tests/distributions/test_gaussian_mixtures.py	159
tests/distributions/test_haar.py	10
tests/distributions/test_hmm.py	700
tests/distributions/test_ig.py	22
tests/distributions/test_improper_uniform.py	20
tests/distributions/test_independent.py	94
tests/distributions/test_kl.py	69
tests/distributions/test_lkj.py	79
tests/distributions/test_lowrank_mvn.py	28
tests/distributions/test_mask.py	106
tests/distributions/test_mixture.py	93
tests/distributions/test_mvn.py	27
tests/distributions/test_mvt.py	111
tests/distributions/test_omt_mvn.py	87
tests/distributions/test_one_hot_categorical.py	76
tests/distributions/test_pickle.py	68
tests/distributions/test_polya_gamma.py	13
tests/distributions/test_rejector.py	125
tests/distributions/test_relaxed_straight_through.py	81
tests/distributions/test_reshape.py	101
tests/distributions/test_shapes.py	71
tests/distributions/test_spanning_tree.py	136
tests/distributions/test_stable.py	104
tests/distributions/test_tensor_type.py	44
tests/distributions/test_torch_patch.py	25
tests/distributions/test_transforms.py	212
tests/distributions/test_unit.py	22
tests/distributions/test_util.py	165
tests/distributions/test_von_mises.py	84
tests/distributions/test_zero_inflated.py	83
tests/doctest_fixtures.py	27
tests/infer/__init__.py	1
tests/infer/conftest.py	8
tests/infer/mcmc/__init__.py	1
tests/infer/mcmc/test_adaptation.py	50
tests/infer/mcmc/test_hmc.py	265
tests/infer/mcmc/test_mcmc_api.py	195
tests/infer/mcmc/test_mcmc_util.py	96
tests/infer/mcmc/test_nuts.py	390
tests/infer/mcmc/test_valid_models.py	321
tests/infer/reparam/__init__.py	1
tests/infer/reparam/test_conjugate.py	158
tests/infer/reparam/test_discrete_cosine.py	67
tests/infer/reparam/test_haar.py	67
tests/infer/reparam/test_hmm.py	182
tests/infer/reparam/test_loc_scale.py	48
tests/infer/reparam/test_neutra.py	59
tests/infer/reparam/test_split.py	45
tests/infer/reparam/test_stable.py	102
tests/infer/reparam/test_studentt.py	49
tests/infer/reparam/test_transform.py	43
tests/infer/reparam/test_unit_jacobian.py	42
tests/infer/test_abstract_infer.py	50
tests/infer/test_autoguide.py	628
tests/infer/test_compute_downstream_costs.py	347
tests/infer/test_conjugate_gradients.py	23
tests/infer/test_csis.py	62
tests/infer/test_discrete.py	247
tests/infer/test_elbo_mapdata.py	90
tests/infer/test_enum.py	2709
tests/infer/test_gradient.py	201
tests/infer/test_inference.py	638
tests/infer/test_initialization.py	25
tests/infer/test_jit.py	396
tests/infer/test_multi_sample_elbos.py	43
tests/infer/test_predictive.py	119
tests/infer/test_sampling.py	57
tests/infer/test_smcfilter.py	187
tests/infer/test_svgd.py	97
tests/infer/test_tmc.py	158
tests/infer/test_util.py	54
tests/infer/test_valid_models.py	1621
tests/integration_tests/__init__.py	1
tests/integration_tests/conftest.py	8
tests/integration_tests/test_conjugate_gaussian_models.py	398
tests/integration_tests/test_tracegraph_elbo.py	396
tests/nn/__init__.py	1
tests/nn/conftest.py	8
tests/nn/test_autoregressive.py	77
tests/nn/test_module.py	486
tests/ops/__init__.py	1
tests/ops/conftest.py	8
tests/ops/einsum/conftest.py	8
tests/ops/einsum/test_adjoint.py	98
tests/ops/einsum/test_torch_log.py	43
tests/ops/gamma_gaussian.py	35
tests/ops/gaussian.py	29
tests/ops/test_arrowhead.py	31
tests/ops/test_contract.py	559
tests/ops/test_gamma_gaussian.py	247
tests/ops/test_gaussian.py	304
tests/ops/test_indexing.py	135
tests/ops/test_integrator.py	126
tests/ops/test_jit.py	29
tests/ops/test_linalg.py	20
tests/ops/test_newton.py	92
tests/ops/test_packed.py	58
tests/ops/test_special.py	34
tests/ops/test_ssm_gp.py	18
tests/ops/test_stats.py	177
tests/ops/test_tensor_utils.py	134
tests/ops/test_welford.py	69
tests/optim/__init__.py	1
tests/optim/conftest.py	8
tests/optim/test_multi.py	53
tests/optim/test_optim.py	192
tests/params/__init__.py	1
tests/params/conftest.py	8
tests/params/test_module.py	73
tests/params/test_param.py	108
tests/perf/__init__.py	1
tests/perf/conftest.py	8
tests/perf/test_benchmark.py	137
tests/poutine/__init__.py	1
tests/poutine/conftest.py	8
tests/poutine/test_counterfactual.py	59
tests/poutine/test_mapdata.py	124
tests/poutine/test_nesting.py	23
tests/poutine/test_poutines.py	695
tests/poutine/test_properties.py	92
tests/poutine/test_trace_struct.py	38
tests/pyroapi/conftest.py	6
tests/pyroapi/test_pyroapi.py	8
tests/test_examples.py	244
tests/test_generic.py	67
tests/test_primitives.py	21
tests/test_util.py	81