pymc-devs/pymc

Speed up test times :rocket:

Open

#7,686 opened on Feb 20, 2025

 (5 comments) (0 reactions) (0 assignees)Python (1,902 forks)batch import
help wanted

Repository metrics

Stars
 (7,926 stars)
PR merge metrics
 (Avg merge 11d) (7 merged PRs in 30d)

Description

If you are motivated to help speed up some tests, we would appreciate it!

Here are some of the slowest test times:

============================= slowest 50 durations =============================
14.37s call     tests/sampling/test_mcmc_external.py::TestNutpieAutoSelection::test_falls_back_to_pymc_for_disqualifying_kwargs[init-advi]
8.79s call     tests/sampling/test_jax.py::test_get_log_likelihood
7.04s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-None-sampler0]
6.07s call     tests/sampling/test_mcmc_external.py::test_sample_var_names[pymc]
6.03s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-None-sampler0]
6.03s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-None-sampler1]
5.73s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[nutpie]
5.71s call     tests/distributions/test_random_alternative_backends.py::test_mvstudentt[FAST_RUN]
5.34s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-cpu-sampler0]
5.29s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-None-sampler0]
5.12s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-cpu-sampler0]
5.12s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-cpu-sampler0]
5.06s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-None-sampler1]
5.01s call     tests/sampling/test_jax.py::test_get_batched_jittered_initial_points
4.81s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[blackjax]
4.65s call     tests/sampling/test_mcmc_external.py::test_sample_var_names[blackjax]
4.61s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-None-sampler0]
4.59s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-cpu-sampler1]
4.54s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-cpu-sampler0]
4.37s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[pymc]
4.33s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-None-sampler1]
4.21s call     tests/sampling/test_mcmc_external.py::test_sample_var_names[numpyro]
4.03s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-cpu-sampler1]
3.99s call     tests/sampling/test_mcmc_external.py::test_sample_var_names[nutpie]
3.95s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[numpyro]
3.90s call     tests/distributions/test_random_alternative_backends.py::test_dirichlet_multinomial[FAST_RUN]
3.69s call     tests/sampling/test_jax.py::test_seeding[2-None-sampler0]
3.55s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-cpu-sampler1]
3.48s call     tests/sampling/test_jax.py::test_seeding[2-123-sampler0]
3.38s call     tests/sampling/test_jax.py::test_initvals_without_jitter[sampler0]
3.31s call     tests/sampling/test_jax.py::test_initvals_without_jitter[sampler1]
3.25s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-None-sampler1]
3.20s call     tests/sampling/test_jax.py::test_seeding[1-None-sampler0]
3.20s call     tests/sampling/test_jax.py::test_seeding[2-123-sampler1]
3.19s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-cpu-sampler1]
3.10s call     tests/sampling/test_jax.py::test_seeding[1-123-sampler0]
2.66s call     tests/distributions/test_random_alternative_backends.py::test_dirichlet_multinomial_dims[JAX]
2.53s call     tests/sampling/test_jax.py::test_seeding[2-None-sampler1]
2.26s call     tests/sampling/test_jax.py::test_seeding[1-None-sampler1]
2.18s call     tests/sampling/test_mcmc_external.py::test_initvals[nutpie-unconstrained_partial]
2.17s call     tests/sampling/test_jax.py::test_idata_kwargs[None-idata_kwargs0-sampler1]
2.12s call     tests/sampling/test_jax.py::test_deterministic_samples[sampler0]
2.12s call     tests/sampling/test_jax.py::test_idata_contains_stats[sample_blackjax_nuts]
2.09s call     tests/sampling/test_mcmc_external.py::test_initvals[blackjax-full_constrained]
2.05s call     tests/sampling/test_jax.py::test_idata_kwargs[None-idata_kwargs0-sampler0]
1.97s call     tests/sampling/test_jax.py::test_seeding[1-123-sampler1]
1.94s call     tests/sampling/test_mcmc_external.py::TestNutpieAutoSelection::test_falls_back_to_pymc_for_disqualifying_kwargs[backend-c]
1.93s call     tests/sampling/test_jax.py::test_idata_kwargs[cpu-idata_kwargs1-sampler0]
1.90s call     tests/sampling/test_jax.py::test_idata_kwargs[None-idata_kwargs3-sampler1]
1.89s call     tests/sampling/test_jax.py::test_idata_kwargs[None-idata_kwargs1-sampler0]
================= 131 passed, 27 warnings in 285.32s (0:04:45) =================
============================ slowest 50 durations =============================
22.08s call     tests/sampling/test_mcmc.py::test_step_args
21.16s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_log_jac_det
21.07s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
11.76s call     tests/distributions/test_transform.py::test_ordered
11.59s call     tests/sampling/test_mcmc.py::test_sample_var_names
10.95s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_half_normal_ordered[sigma1-size1]
9.89s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_backward
9.82s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-advi]
9.78s call     tests/distributions/test_transform.py::test_simplex_bounds
9.71s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-advi]
9.70s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a1-None]
9.69s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-advi]
9.61s call     tests/distributions/test_transform.py::TestCholeskyCovTransform::test_log_jac_det
8.94s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_1_param
8.91s call     tests/sampling/test_mcmc.py::test_blas_cores
8.34s call     tests/sampling/test_mcmc.py::test_sample_find_MAP_does_not_modify_start
7.99s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_mvnormal_transform[transform0-mu0-cov0-None-shape0]
7.91s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_2_param
7.78s call     tests/distributions/test_transform.py::test_interval
7.78s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-map]
6.97s call     tests/sampling/test_mcmc.py::TestSample::test_sample_does_not_rely_on_external_global_seeding
6.95s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-map]
6.90s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
6.74s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_beta_ordered[a1-b1-size1]
6.59s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-map]
6.38s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_backward_produces_valid_cholesky_corr[3]
6.35s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-jitter+adapt_diag]
6.28s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-jitter+adapt_diag]
6.23s call     tests/distributions/test_transform.py::test_simplex
6.15s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-jitter+adapt_diag]
6.09s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[map]
5.74s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
5.63s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_other[0.5-5.5-size1-transform1]
5.57s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_ordered[lower1-upper1-size1]
5.28s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_vonmises_ordered[mu1-kappa1-size1]
5.02s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_mvnormal_transform[transform0-mu1-cov1-size1-shape1]
4.83s call     tests/distributions/test_transform.py::test_chain_jacob_det
4.62s call     tests/ode/test_ode.py::TestDiffEqModel::test_scalar_ode_2_param
4.58s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a2-size2]
4.39s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-advi]
4.37s call     tests/distributions/test_transform.py::test_logodds
4.20s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-2-advi]
4.16s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-2-advi]
4.08s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-2-advi]
4.05s call     tests/distributions/test_transform.py::TestCholeskyCovTransform::test_backward_produces_valid_spd_matrix[3]
3.98s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[False]
3.97s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-1-advi]
3.94s call     tests/distributions/test_transform.py::test_sum_to_1
3.93s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-1-advi]
3.89s call     tests/sampling/test_mcmc.py::TestSample::test_default_sample_does_not_set_global_seed
========== 210 passed, 7 skipped, 212 warnings in 565.31s (0:09:25) ===========
============================ slowest 50 durations =============================
19.82s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_log_jac_det
18.98s call     tests/sampling/test_mcmc.py::test_step_args
17.50s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
10.81s call     tests/distributions/test_transform.py::test_ordered
10.15s call     tests/sampling/test_mcmc.py::test_sample_var_names
9.26s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_backward
9.09s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_half_normal_ordered[sigma1-size1]
8.73s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a1-None]
8.71s call     tests/distributions/test_transform.py::test_simplex_bounds
8.17s call     tests/sampling/test_mcmc.py::test_blas_cores
8.10s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-advi]
8.00s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-advi]
7.98s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_1_param
7.92s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-advi]
7.55s call     tests/distributions/test_transform.py::TestCholeskyCovTransform::test_log_jac_det
7.32s call     tests/sampling/test_mcmc.py::test_sample_find_MAP_does_not_modify_start
6.88s call     tests/distributions/test_transform.py::test_interval
6.59s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-map]
6.38s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
6.29s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_beta_ordered[a1-b1-size1]
6.14s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_mvnormal_transform[transform0-mu0-cov0-None-shape0]
6.13s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-map]
6.01s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-map]
5.95s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_backward_produces_valid_cholesky_corr[3]
5.90s call     tests/sampling/test_mcmc.py::TestSample::test_sample_does_not_rely_on_external_global_seeding
5.54s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[map]
5.42s call     tests/distributions/test_transform.py::test_simplex
5.30s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_other[0.5-5.5-size1-transform1]
5.21s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_ordered[lower1-upper1-size1]
5.17s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-jitter+adapt_diag]
5.17s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
5.17s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-jitter+adapt_diag]
5.17s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-jitter+adapt_diag]
4.87s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_vonmises_ordered[mu1-kappa1-size1]
4.80s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_2_param
4.69s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_mvnormal_transform[transform0-mu1-cov1-size1-shape1]
4.26s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a2-size2]
4.20s call     tests/distributions/test_transform.py::test_chain_jacob_det
3.85s call     tests/distributions/test_transform.py::test_logodds
3.81s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_beta_ordered[1.0-1.0-size0]
3.75s call     tests/distributions/test_transform.py::TestCholeskyCovTransform::test_backward_produces_valid_spd_matrix[3]
3.53s call     tests/distributions/test_transform.py::test_sum_to_1
3.49s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[False]
3.48s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-advi]
3.48s call     tests/sampling/test_mcmc.py::TestSample::test_default_sample_does_not_set_global_seed
3.41s call     tests/sampling/test_mcmc.py::test_no_init_nuts_compound
3.37s call     tests/sampling/test_mcmc.py::TestSampleReturn::test_logs_sampler_warnings[2]
3.33s call     tests/distributions/test_transform.py::test_log
3.33s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-2-advi]
3.33s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-2-advi]
========== 210 passed, 7 skipped, 212 warnings in 497.40s (0:08:17) ===========
============================= slowest 50 durations =============================
34.70s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
12.08s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
9.54s call     tests/sampling/test_mcmc.py::test_sample_var_names
8.57s call     tests/sampling/test_mcmc.py::TestType::test_float32
8.51s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi_map]
7.68s setup    tests/sampling/test_mcmc.py::TestInitNuts::test_checks_seeds_kwarg
7.53s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[map]
6.75s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
5.62s call     tests/sampling/test_mcmc.py::TestType::test_float64
5.43s call     tests/sampling/test_mcmc.py::test_sample_find_MAP_does_not_modify_start
4.31s call     tests/sampling/test_mcmc.py::test_step_args
4.05s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-jitter+adapt_diag]
3.96s call     tests/sampling/test_mcmc.py::TestSampleReturn::test_categorical_gibbs_respects_driver_tune_boundary
3.17s call     tests/sampling/test_mcmc.py::TestShared::test_sample
2.78s call     tests/sampling/test_mcmc.py::TestSample::test_deterministic_of_unobserved[False]
2.47s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[False]
2.36s call     tests/sampling/test_mcmc.py::test_blas_cores
1.93s call     tests/sampling/test_mcmc.py::TestNamedSampling::test_shared_named
1.90s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-advi]
1.80s call     tests/sampling/test_mcmc.py::TestSample::test_deterministic_of_unobserved[True]
1.74s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-map]
1.68s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-1-advi]
1.65s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-advi]
1.59s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-1-advi]
1.59s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-2-advi]
1.50s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-1-advi]
1.40s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-1-map]
1.37s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[ADVI+adapt_diag]
1.35s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-map]
1.33s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[map]
1.33s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-1-advi]
1.29s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-advi]
1.28s call     tests/sampling/test_mcmc.py::TestDefaultTuneSteps::test_compound_step
1.27s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-2-advi]
1.23s call     tests/sampling/test_mcmc.py::TestSample::test_default_sample_does_not_set_global_seed
1.21s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-2-advi]
1.21s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-1-map]
1.15s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi_map]
1.15s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-advi]
1.14s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[True]
1.11s call     tests/sampling/test_mcmc.py::test_setup_chain_receives_draw_counts[sequential]
1.08s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-2-map]
1.05s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-1-map]
0.95s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-map]
0.90s call     tests/sampling/test_mcmc.py::TestDefaultTuneSteps::test_sample
0.90s call     tests/sampling/test_mcmc.py::TestDefaultTuneSteps::test_custom_step
0.90s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-2-map]
0.79s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-map]
0.78s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-2-map]
0.75s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-1-map]
============ 109 passed, 5 skipped, 5 warnings in 215.93s (0:03:35) ============
============================= slowest 50 durations =============================
22.36s call     tests/backends/test_arviz.py::TestDataPyMC::test_mv_missing_data_model
5.64s call     tests/backends/test_arviz.py::TestDataPyMC::test_multivariate_observations
4.70s setup    tests/backends/test_arviz.py::TestDataPyMC::test_to_idata
3.49s call     tests/backends/test_zarr.py::test_record[single_step-include_transformed-frequent_writes]
3.32s call     tests/backends/test_arviz.py::TestDataPyMC::test_autodetect_coords_from_model[True]
2.08s call     tests/backends/test_arviz.py::TestDataPyMC::test_single_observation
2.03s call     tests/backends/test_zarr.py::test_record[compound_step-include_transformed-frequent_writes]
1.78s call     tests/backends/test_arviz.py::TestDataPyMC::test_missing_data_model
1.77s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_idata-keep_warning_stat-sequential_sampling-no_loglike-frequent_writes]
1.55s call     tests/backends/test_arviz.py::TestDataPyMC::test_to_idata
1.51s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_zarr-discard_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
1.45s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_zarr-keep_warning_stat-parallel_sampling-compute_loglike-sparse_writes]
1.35s call     tests/backends/test_zarr.py::test_sample[single_step-discard_transformed-discard_tuning-return_idata-discard_warning_stat-parallel_sampling-no_loglike-frequent_writes]
1.25s call     tests/backends/test_zarr.py::test_sample[single_step-discard_transformed-keep_tuning-return_zarr-discard_warning_stat-parallel_sampling-compute_loglike-sparse_writes]
1.25s setup    tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-discard_tuning-return_idata-discard_warning_stat-parallel_sampling-no_loglike-sparse_writes]
1.22s setup    tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-keep_tuning-return_zarr-discard_warning_stat-sequential_sampling-no_loglike-frequent_writes]
1.19s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-discard_tuning-return_idata-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
1.18s setup    tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-discard_tuning-return_zarr-discard_warning_stat-parallel_sampling-no_loglike-sparse_writes]
1.12s setup    tests/backends/test_zarr.py::test_sample[single_step-discard_transformed-discard_tuning-return_idata-discard_warning_stat-sequential_sampling-no_loglike-frequent_writes]
1.11s setup    tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-keep_tuning-return_idata-discard_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
1.11s call     tests/backends/test_zarr.py::test_sample[single_step-discard_transformed-keep_tuning-return_idata-keep_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
1.09s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_idata-discard_warning_stat-parallel_sampling-no_loglike-sparse_writes]
1.04s call     tests/backends/test_zarr.py::test_sampling_consistency[compound_step-frequent_writes]
1.03s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-discard_tuning-return_zarr-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
1.02s call     tests/backends/test_zarr.py::test_record[single_step-discard_transformed-frequent_writes]
1.01s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_zarr-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
1.01s call     tests/backends/test_arviz.py::TestDataPyMC::test_constant_data[True]
0.99s call     tests/backends/test_arviz.py::TestDataPyMC::test_overwrite_model_coords_dims
0.96s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-keep_tuning-return_zarr-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.91s call     tests/backends/test_arviz.py::TestDataPyMC::test_predictions_constant_data
0.91s call     tests/backends/test_zarr.py::test_sample[single_step-discard_transformed-discard_tuning-return_idata-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.89s call     tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-keep_tuning-return_idata-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.87s call     tests/backends/test_arviz.py::TestDataPyMC::test_multiple_observed_rv[True]
0.83s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_idata-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.82s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_idata-keep_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
0.80s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-keep_tuning-return_zarr-keep_warning_stat-parallel_sampling-compute_loglike-sparse_writes]
0.79s setup    tests/backends/test_zarr.py::test_sample[single_step-include_transformed-keep_tuning-return_idata-discard_warning_stat-parallel_sampling-compute_loglike-sparse_writes]
0.78s call     tests/backends/test_zarr.py::test_sampling_consistency[single_step-frequent_writes]
0.78s call     tests/backends/test_zarr.py::test_sampling_consistency[compound_step-sparse_writes]
0.76s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_idata-keep_warning_stat-parallel_sampling-no_loglike-frequent_writes]
0.75s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_idata-keep_warning_stat-parallel_sampling-compute_loglike-sparse_writes]
0.73s setup    tests/backends/test_zarr.py::test_sample[single_step-include_transformed-keep_tuning-return_zarr-discard_warning_stat-parallel_sampling-no_loglike-frequent_writes]
0.72s call     tests/backends/test_arviz.py::TestDataPyMC::test_observed_data_also_constant[True]
0.72s call     tests/backends/test_zarr.py::test_sample[single_step-discard_transformed-discard_tuning-return_idata-discard_warning_stat-parallel_sampling-compute_loglike-sparse_writes]
0.72s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_idata-keep_warning_stat-parallel_sampling-no_loglike-sparse_writes]
0.70s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_zarr-discard_warning_stat-parallel_sampling-compute_loglike-sparse_writes]
0.70s setup    tests/backends/test_zarr.py::test_sample[single_step-include_transformed-discard_tuning-return_idata-keep_warning_stat-sequential_sampling-no_loglike-sparse_writes]
0.65s call     tests/backends/test_zarr.py::test_record[compound_step-discard_transformed-frequent_writes]
0.63s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_zarr-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.62s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-keep_tuning-return_idata-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
========== 481 passed, 64 skipped, 266 warnings in 242.32s (0:04:02) ===========
============================= slowest 50 durations =============================
21.65s call     tests/sampling/test_mcmc_external.py::TestNutpieAutoSelection::test_falls_back_to_pymc_for_disqualifying_kwargs[init-advi]
11.50s call     tests/sampling/test_jax.py::test_get_log_likelihood
10.15s call     tests/sampling/test_mcmc_external.py::test_sample_var_names[pymc]
9.57s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[nutpie]
7.56s call     tests/sampling/test_mcmc_external.py::test_sample_var_names[nutpie]
6.73s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-None-sampler0]
6.55s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-None-sampler0]
6.40s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[pymc]
6.25s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-None-sampler1]
6.21s call     tests/distributions/test_random_alternative_backends.py::test_mvstudentt[NUMBA]
5.76s call     tests/sampling/test_jax.py::test_get_batched_jittered_initial_points
5.48s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-cpu-sampler0]
5.46s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-None-sampler0]
5.45s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-cpu-sampler0]
5.30s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-cpu-sampler0]
5.20s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[blackjax]
5.09s call     tests/sampling/test_mcmc_external.py::test_sample_var_names[blackjax]
4.94s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-None-sampler0]
4.88s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-cpu-sampler0]
4.66s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-cpu-sampler1]
4.65s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-None-sampler1]
4.51s call     tests/sampling/test_mcmc_external.py::test_sample_var_names[numpyro]
4.33s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-None-sampler1]
4.31s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-cpu-sampler1]
4.29s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[numpyro]
4.16s call     tests/sampling/test_mcmc_external.py::test_initvals[pymc-negative_control]
4.09s call     tests/sampling/test_jax.py::test_initvals_without_jitter[sampler0]
4.05s call     tests/sampling/test_jax.py::test_seeding[2-None-sampler0]
3.93s call     tests/distributions/test_random_alternative_backends.py::test_dirichlet_multinomial[NUMBA]
3.78s call     tests/sampling/test_jax.py::test_seeding[2-123-sampler0]
3.49s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-cpu-sampler1]
3.49s call     tests/sampling/test_jax.py::test_seeding[1-None-sampler0]
3.47s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-None-sampler1]
3.44s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-cpu-sampler1]
3.39s call     tests/sampling/test_jax.py::test_seeding[1-123-sampler0]
3.00s call     tests/distributions/test_random_alternative_backends.py::test_dirichlet_multinomial_dims[JAX]
2.93s call     tests/sampling/test_mcmc_external.py::test_initvals[nutpie-negative_control]
2.92s call     tests/sampling/test_jax.py::test_initvals_without_jitter[sampler1]
2.69s call     tests/sampling/test_jax.py::test_seeding[2-None-sampler1]
2.65s call     tests/sampling/test_jax.py::test_convergence_warnings[blackjax]
2.63s call     tests/sampling/test_jax.py::test_seeding[2-123-sampler1]
2.53s call     tests/distributions/test_random_alternative_backends.py::test_mvstudentt[FAST_RUN]
2.40s call     tests/sampling/test_jax.py::test_seeding[1-None-sampler1]
2.29s call     tests/sampling/test_jax.py::test_idata_contains_stats[sample_blackjax_nuts]
2.28s call     tests/sampling/test_jax.py::test_deterministic_samples[sampler0]
2.28s call     tests/sampling/test_mcmc_external.py::test_initvals[blackjax-full_constrained]
2.22s call     tests/sampling/test_jax.py::test_idata_kwargs[None-idata_kwargs0-sampler1]
2.19s call     tests/sampling/test_jax.py::test_idata_kwargs[None-idata_kwargs0-sampler0]
2.10s call     tests/sampling/test_jax.py::test_seeding[1-123-sampler1]
2.07s call     tests/sampling/test_mcmc_external.py::test_initvals[blackjax-unconstrained_partial]
================= 131 passed, 30 warnings in 325.34s (0:05:25) =================
============================ slowest 50 durations =============================
32.23s call     tests/variational/test_inference.py::test_fit_oo[SVGD-full]
31.27s call     tests/variational/test_inference.py::test_replacements[FullRankADVI]
28.62s call     tests/variational/test_inference.py::test_replacements[ASVGD]
23.84s call     tests/variational/test_inference.py::TestUntransformedData::test_state_dirichlet
21.93s call     tests/variational/test_inference.py::test_replacements[ADVI]
20.99s call     tests/variational/test_inference.py::test_fit_data_coords
19.00s call     tests/variational/test_inference.py::test_replacements[SVGD]
17.31s call     tests/variational/test_approximations.py::test_seeding_advi_fit
15.67s call     tests/variational/test_inference.py::test_fit_with_nans[True]
14.95s call     tests/variational/test_inference.py::test_fit_oo[ASVGD-full]
14.54s call     tests/variational/test_inference.py::test_fit_oo[FullRankADVI-full]
14.30s call     tests/variational/test_inference.py::test_fit_oo[ADVI-mini]
12.63s call     tests/variational/test_inference.py::test_fit_data[SVGD-full]
9.99s call     tests/variational/test_approximations.py::test_elbo
8.70s call     tests/variational/test_inference.py::test_fit_fn_text[svgd-kwargs4-None]
8.04s call     tests/variational/test_inference.py::test_fit_oo[FullRankADVI-mini]
7.81s call     tests/variational/test_inference.py::test_multiple_minibatch_variables
7.77s call     tests/variational/test_inference.py::test_fit_fn_text[fullrank_advi-kwargs3-None]
7.09s call     tests/variational/test_inference.py::test_sample_posterior_after_minibatch
6.69s call     tests/variational/test_inference.py::TestUntransformedData::test_state_full_rank
6.62s call     tests/variational/test_approximations.py::test_empirical_from_trace
6.32s call     tests/variational/test_inference.py::test_fit_fn_text[advi-kwargs2-None]
5.80s call     tests/variational/test_inference.py::test_fit_data[ASVGD-full]
5.44s call     tests/variational/test_inference.py::test_fit_oo[ADVI-full]
5.24s call     tests/variational/test_approximations.py::test_empirical_does_not_support_inference_data
5.16s call     tests/variational/test_inference.py::test_var_replacement
4.78s call     tests/variational/test_opvi.py::test_init_groups[raises0-grouping0]
4.75s call     tests/variational/test_inference.py::test_fit_fn_text[asvgd-kwargs7-None]
3.94s call     tests/variational/test_inference.py::test_fit_with_nans[False]
3.91s call     tests/variational/test_opvi.py::test_logq_globals[FullRankGroup: None, MeanFieldGroup: ['one']]
3.52s call     tests/variational/test_inference.py::TestUntransformedData::test_state_empirical_std_is_none
3.19s call     tests/variational/test_inference.py::TestUntransformedData::test_state_mean_field
2.91s call     tests/variational/test_inference.py::test_fit_data[FullRankADVI-mini]
2.84s call     tests/variational/test_opvi.py::test_logq_mini_2_sample_2_var[MeanFieldGroup: {}]
2.55s call     tests/variational/test_inference.py::TestUntransformedData::test_state_in_callback
2.49s call     tests/variational/test_approximations.py::test_scale_cost_to_minibatch_works[5]
2.31s call     tests/variational/test_inference.py::test_fit_start[SVGD-full]
2.24s call     tests/variational/test_inference.py::test_fit_data[ADVI-mini]
2.21s call     tests/variational/test_inference.py::test_fit_data[FullRankADVI-full]
2.09s call     tests/variational/test_opvi.py::test_init_groups[raises1-grouping1]
2.04s call     tests/variational/test_opvi.py::test_group_api_params[raises0-params0-MeanFieldGroup-kw0-None]
1.91s call     tests/variational/test_inference.py::test_fit_start[FullRankADVI-full]
1.81s call     tests/variational/test_approximations.py::test_elbo_beta_kl[5]
1.75s call     tests/variational/test_approximations.py::test_scale_cost_to_minibatch_works[8]
1.69s call     tests/variational/test_inference.py::TestUntransformedData::test_state_include_transformed
1.67s call     tests/variational/test_approximations.py::test_elbo_beta_kl[8]
1.40s call     tests/test_initial_point.py::TestInitvalEvaluation::test_dependent_initval_from_OFG[False]
1.40s call     tests/variational/test_opvi.py::test_logq_globals[MeanFieldGroup: None]
1.33s call     tests/variational/test_inference.py::TestUntransformedData::test_state_include_transformed_dirichlet
1.25s call     tests/variational/test_inference.py::test_remove_scan_op
=========== 132 passed, 10 skipped, 3 warnings in 461.87s (0:07:41) ===========
============================= slowest 50 durations =============================
16.81s call     tests/sampling/test_parallel.py::test_context
13.91s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_separable
9.90s call     tests/sampling/test_parallel.py::test_remote_pipe_closed
7.26s call     tests/model/test_core.py::TestValueGradFunction::test_missing_data
7.03s call     tests/model/test_core.py::TestImputationMissingData::test_missing_dual_observations
6.88s call     tests/model/test_core.py::TestImputationMissingData::test_coordinates
6.28s call     tests/test_data.py::TestData::test_sample
5.78s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter[logp]
5.18s call     tests/model/test_core.py::test_model_logp_fast_compile
5.16s call     tests/model/test_core.py::TestTransformArgs::test_default_transform_is_applied
4.76s call     tests/model/test_core.py::TestImputationMissingData::test_interval_missing_observations
4.66s call     tests/sampling/test_parallel.py::test_explicit_sample[spawn]
4.56s call     tests/sampling/test_parallel.py::test_spawn_densitydist_bound_method
4.43s call     tests/test_data.py::TestData::test_shared_data_as_index
4.43s call     tests/model/test_core.py::TestModelDebug::test_invalid_observed_value
4.12s call     tests/model/test_core.py::TestValueGradFunction::test_edge_case
4.04s call     tests/model/test_core.py::test_model_d2logp[True]
3.98s call     tests/sampling/test_parallel.py::test_spawn_densitydist_function
3.88s call     tests/test_data.py::TestData::test_shared_data_as_rv_input
3.70s call     tests/model/test_core.py::TestImputationMissingData::test_truncated_normal
3.30s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter_cant_be_evaluated[logp-True]
2.98s call     tests/sampling/test_parallel.py::test_bad_unpickle
2.66s call     tests/variational/test_minibatch_rv.py::TestMinibatchRandomVariable::test_minibatch_parameter_and_value
2.65s call     tests/model/test_core.py::TestPytensorRelatedLogpBugs::test_pytensor_switch_broadcast_edge_cases_2
2.50s call     tests/test_data.py::test_scaling_data_works_in_likelihood
2.20s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_unseparable
2.13s call     tests/test_data.py::TestData::test_sample_posterior_predictive_after_set_data_with_coords
2.11s call     tests/model/test_core.py::TestImputationMissingData::test_missing_basic[pandas]
2.02s call     tests/model/test_core.py::TestFrozenModelCaching::test_logp_dlogp_function_weights_are_per_call
2.01s call     tests/model/test_core.py::TestPytensorRelatedLogpBugs::test_pytensor_switch_broadcast_edge_cases_1
1.99s call     tests/model/test_core.py::TestFrozenModelCaching::test_logp_dlogp_function_extra_values_are_per_call
1.77s call     tests/model/test_core.py::test_model_dlogp[True]
1.73s call     tests/test_data.py::TestData::test_set_data_to_non_data_container_variables
1.68s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter[dlogp]
1.65s call     tests/model/test_core.py::test_tempered_logp_dlogp[False]
1.64s call     tests/model/test_core.py::TestImputationMissingData::test_missing_logp1
1.62s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter_cant_be_evaluated[dlogp-True]
1.60s call     tests/model/test_core.py::TestImputationMissingData::test_missing_vector_parameter
1.57s call     tests/model/test_core.py::test_datalogp_multiple_shapes
1.54s call     tests/model/test_core.py::TestCheckStartVals::test_valid_start_point
1.46s call     tests/model/test_core.py::TestImputationMissingData::test_missing_basic[masked]
1.43s call     tests/model/test_core.py::TestImputationMissingData::test_missing_logp2
1.39s call     tests/model/test_core.py::TestFrozenModelCaching::test_compile_fn_jax_bypasses_cache_but_stays_seeded
1.34s call     tests/model/test_core.py::TestFrozenModelCaching::test_unfrozen_model_does_not_cache
1.32s call     tests/model/test_core.py::test_model_d2logp[False]
1.31s call     tests/model/test_core.py::TestFrozenModelCaching::test_logp_dlogp_d2logp_graphs_are_cached
1.31s call     tests/model/test_core.py::TestImputationMissingData::test_symbolic_random_variable
1.24s call     tests/model/test_core.py::TestImputationMissingData::test_missing_with_predictors
1.16s call     tests/model/test_core.py::TestModelDebug::test_invalid_value
1.12s call     tests/test_data.py::TestData::test_deterministic
====== 198 passed, 2 skipped, 1 xfailed, 17 warnings in 233.76s (0:03:53) ======
============================= slowest 50 durations =============================
10.03s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
3.19s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
2.84s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[map]
2.61s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
2.05s call     tests/sampling/test_mcmc.py::test_sample_var_names
1.79s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-2-advi]
1.73s setup    tests/sampling/test_mcmc.py::TestInitNuts::test_checks_seeds_kwarg
1.73s call     tests/sampling/test_mcmc.py::test_step_args
1.69s call     tests/sampling/test_mcmc.py::TestSampleReturn::test_categorical_gibbs_respects_driver_tune_boundary
1.65s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-advi]
1.60s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-1-advi]
1.59s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-advi]
1.58s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-1-advi]
1.55s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-advi]
1.52s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-advi]
1.51s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-1-advi]
1.50s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi_map]
1.47s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-2-advi]
1.47s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-map]
1.46s call     tests/sampling/test_mcmc.py::TestShared::test_sample
1.45s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-2-advi]
1.43s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-1-advi]
1.40s call     tests/sampling/test_mcmc.py::TestType::test_float32
1.35s call     tests/sampling/test_mcmc.py::TestType::test_float64
1.20s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi_map]
1.18s call     tests/sampling/test_mcmc.py::TestNamedSampling::test_shared_named
1.08s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[ADVI+adapt_diag]
1.04s call     tests/sampling/test_mcmc.py::test_sample_find_MAP_does_not_modify_start
0.91s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-map]
0.86s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-jitter+adapt_diag]
0.80s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[map]
0.73s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-1-map]
0.72s call     tests/sampling/test_mcmc.py::TestSample::test_deterministic_of_unobserved[True]
0.68s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-1-map]
0.67s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-map]
0.63s call     tests/sampling/test_mcmc.py::TestSample::test_sample_does_not_rely_on_external_global_seeding
0.62s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-1-map]
0.62s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-2-map]
0.59s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-map]
0.57s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-map]
0.57s call     tests/sampling/test_mcmc.py::TestSample::test_deterministic_of_unobserved[False]
0.55s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[False]
0.54s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[jitter+adapt_diag]
0.49s call     tests/sampling/test_mcmc.py::test_blas_cores
0.48s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-2-map]
0.47s call     tests/sampling/test_mcmc.py::TestSample::test_default_sample_does_not_set_global_seed
0.47s call     tests/sampling/test_mcmc.py::TestNamedSampling::test_constant_named
0.46s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-1-map]
0.46s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-2-map]
0.37s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-1-jitter+adapt_diag]
============ 109 passed, 5 skipped, 5 warnings in 98.99s (0:01:38) =============
============================= slowest 50 durations =============================
21.29s call     tests/model/test_core.py::test_model_logp_fast_compile
8.24s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_separable
6.39s call     tests/model/test_core.py::test_model_d2logp[True]
5.84s call     tests/model/test_core.py::TestImputationMissingData::test_coordinates
5.50s call     tests/model/test_core.py::TestImputationMissingData::test_missing_dual_observations
5.12s call     tests/sampling/test_parallel.py::test_context
3.96s call     tests/model/test_core.py::TestImputationMissingData::test_missing_with_predictors
3.90s call     tests/model/test_core.py::TestModelDebug::test_invalid_observed_value
3.57s call     tests/test_data.py::TestMinibatch::test_1d
3.54s call     tests/sampling/test_parallel.py::test_spawn_densitydist_bound_method
3.48s call     tests/model/test_core.py::TestImputationMissingData::test_interval_missing_observations
2.98s call     tests/sampling/test_parallel.py::test_remote_pipe_closed
2.77s call     tests/test_data.py::TestData::test_sample
2.42s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter_cant_be_evaluated[dlogp-True]
2.39s call     tests/model/test_core.py::TestImputationMissingData::test_missing_basic[pandas]
2.38s call     tests/test_data.py::TestData::test_shared_data_as_rv_input
2.38s call     tests/sampling/test_parallel.py::test_spawn_densitydist_function
2.18s call     tests/test_data.py::TestData::test_shared_data_as_index
1.93s call     tests/sampling/test_parallel.py::test_explicit_sample[spawn]
1.75s call     tests/model/test_core.py::TestTransformArgs::test_default_transform_is_applied
1.64s call     tests/sampling/test_parallel.py::test_bad_unpickle
1.61s call     tests/model/test_core.py::TestValueGradFunction::test_missing_data
1.57s call     tests/model/test_core.py::test_model_d2logp[False]
1.50s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter[logp]
1.46s call     tests/model/test_core.py::TestFrozenModelCaching::test_compile_fn_jax_bypasses_cache_but_stays_seeded
1.32s call     tests/model/test_core.py::TestSetUpdateCoords::test_resize_from_set_data_and_set_dim
1.28s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter_cant_be_evaluated[logp-True]
1.27s call     tests/model/test_core.py::TestImputationMissingData::test_missing_basic[masked]
1.25s call     tests/test_data.py::TestData::test_sample_after_set_data
1.24s call     tests/model/test_core.py::TestValueGradFunction::test_edge_case
1.20s call     tests/test_data.py::TestData::test_sample_posterior_predictive_after_set_data
1.19s call     tests/model/test_core.py::TestModelDebug::test_invalid_value
1.18s call     tests/model/test_core.py::test_model_dlogp[True]
1.14s call     tests/model/test_core.py::TestFrozenModelCaching::test_logp_dlogp_function_extra_values_are_per_call
1.11s call     tests/model/test_core.py::test_model_logp[True]
0.94s call     tests/test_data.py::TestData::test_sample_posterior_predictive_after_set_data_with_coords
0.93s call     tests/model/test_core.py::TestPytensorRelatedLogpBugs::test_pytensor_switch_broadcast_edge_cases_2
0.92s call     tests/model/test_core.py::TestNested::test_nested_model_to_netcdf
0.90s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter[random]
0.85s call     tests/variational/test_minibatch_rv.py::TestMinibatchRandomVariable::test_minibatch_parameter_and_value
0.85s call     tests/test_data.py::TestMinibatch::test_assert
0.84s call     tests/model/test_core.py::TestImputationMissingData::test_missing_logp2
0.83s call     tests/test_data.py::TestData::test_set_data_to_non_data_container_variables
0.83s call     tests/model/test_core.py::TestImputationMissingData::test_truncated_normal
0.80s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter[dlogp]
0.79s call     tests/model/test_core.py::test_tempered_logp_dlogp[True]
0.76s call     tests/model/test_core.py::test_model_logp[False]
0.70s call     tests/model/test_core.py::TestImputationMissingData::test_symbolic_random_variable
0.69s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_unseparable
0.68s call     tests/model/test_core.py::test_datalogp_multiple_shapes
====== 198 passed, 2 skipped, 1 xfailed, 20 warnings in 169.30s (0:02:49) ======
============================= slowest 50 durations =============================
22.81s call     tests/backends/test_arviz.py::TestDataPyMC::test_mv_missing_data_model
10.15s setup    tests/backends/test_arviz.py::TestDataPyMC::test_to_idata
6.88s call     tests/backends/test_arviz.py::TestDataPyMC::test_autodetect_coords_from_model[True]
6.52s call     tests/backends/test_arviz.py::TestDataPyMC::test_multivariate_observations
5.18s call     tests/backends/test_arviz.py::TestDataPyMC::test_single_observation
4.64s setup    tests/backends/test_zarr.py::test_record[single_step-include_transformed-frequent_writes]
4.23s call     tests/backends/test_arviz.py::TestDataPyMC::test_missing_data_model
3.01s call     tests/backends/test_arviz.py::TestDataPyMC::test_overwrite_model_coords_dims
2.63s call     tests/backends/test_arviz.py::TestDataPyMC::test_constant_data[True]
2.60s setup    tests/backends/test_zarr.py::test_record[compound_step-include_transformed-frequent_writes]
1.85s call     tests/backends/test_zarr.py::test_record[single_step-include_transformed-frequent_writes]
1.68s call     tests/backends/test_arviz.py::TestDataPyMC::test_to_idata
1.39s call     tests/backends/test_arviz.py::TestDataPyMC::test_potential
1.28s call     tests/backends/test_arviz.py::TestDataPyMC::test_predictions_constant_data
1.18s call     tests/backends/test_arviz.py::TestDataPyMC::test_single_chain[1]
1.04s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-discard_tuning-return_idata-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.67s call     tests/backends/test_zarr.py::test_sampling_consistency[compound_step-frequent_writes]
0.67s call     tests/backends/test_arviz.py::TestDataPyMC::test_include_transformed
0.64s call     tests/backends/test_zarr.py::test_sample[single_step-discard_transformed-discard_tuning-return_idata-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.64s call     tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-discard_tuning-return_idata-keep_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
0.63s call     tests/backends/test_arviz.py::TestPyMCWarmupHandling::test_save_warmup[0-50-1-False]
0.62s call     tests/backends/test_arviz.py::TestDataPyMC::test_conversion_from_variables_subset
0.60s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_idata-discard_warning_stat-parallel_sampling-no_loglike-frequent_writes]
0.57s call     tests/backends/test_zarr.py::test_sample[single_step-discard_transformed-keep_tuning-return_zarr-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.54s call     tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-keep_tuning-return_idata-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.54s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-keep_tuning-return_zarr-discard_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
0.53s call     tests/backends/test_zarr.py::test_sampling_consistency[single_step-frequent_writes]
0.51s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_zarr-discard_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
0.50s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_idata-keep_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
0.49s call     tests/backends/test_zarr.py::test_record[single_step-discard_transformed-frequent_writes]
0.48s call     tests/backends/test_zarr.py::test_sample[single_step-discard_transformed-keep_tuning-return_idata-discard_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
0.47s call     tests/backends/test_zarr.py::test_record[compound_step-discard_transformed-frequent_writes]
0.46s call     tests/backends/test_zarr.py::test_sample[single_step-discard_transformed-keep_tuning-return_idata-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.45s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-discard_tuning-return_zarr-discard_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
0.45s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-keep_tuning-return_zarr-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.44s call     tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-discard_tuning-return_zarr-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.44s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_zarr-keep_warning_stat-parallel_sampling-compute_loglike-sparse_writes]
0.44s call     tests/backends/test_zarr.py::test_record[compound_step-include_transformed-frequent_writes]
0.43s call     tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-keep_tuning-return_idata-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.42s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_zarr-keep_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
0.42s call     tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-discard_tuning-return_idata-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.41s call     tests/backends/test_arviz.py::TestDataPyMC::test_multiple_observed_rv[log_likelihood2]
0.41s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_idata-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.41s call     tests/backends/test_zarr.py::test_sample[single_step-discard_transformed-discard_tuning-return_idata-keep_warning_stat-sequential_sampling-no_loglike-frequent_writes]
0.40s call     tests/backends/test_zarr.py::test_sampling_consistency[compound_step-sparse_writes]
0.40s setup    tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-keep_tuning-return_zarr-discard_warning_stat-parallel_sampling-no_loglike-sparse_writes]
0.40s call     tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-keep_tuning-return_zarr-keep_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
0.40s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_idata-discard_warning_stat-parallel_sampling-no_loglike-frequent_writes]
0.39s call     tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-keep_tuning-return_zarr-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
0.39s call     tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_idata-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
========== 481 passed, 64 skipped, 264 warnings in 207.23s (0:03:27) ===========
============================ slowest 50 durations =============================
53.42s call     tests/variational/test_inference.py::test_replacements[FullRankADVI]
52.55s call     tests/variational/test_inference.py::test_replacements[ASVGD]
40.67s call     tests/variational/test_inference.py::test_replacements[SVGD]
38.73s call     tests/variational/test_inference.py::test_replacements[ADVI]
33.73s call     tests/variational/test_inference.py::test_fit_oo[SVGD-full]
21.55s call     tests/variational/test_approximations.py::test_seeding_advi_fit
21.24s call     tests/variational/test_inference.py::TestUntransformedData::test_state_dirichlet
16.95s call     tests/variational/test_inference.py::test_fit_oo[ASVGD-full]
16.34s call     tests/variational/test_inference.py::test_fit_data_coords
15.69s call     tests/variational/test_inference.py::test_fit_oo[ADVI-mini]
14.45s call     tests/variational/test_inference.py::test_fit_oo[FullRankADVI-full]
12.94s call     tests/variational/test_inference.py::test_fit_data[SVGD-full]
11.15s call     tests/variational/test_inference.py::test_fit_fn_text[asvgd-kwargs7-None]
10.48s call     tests/variational/test_inference.py::test_fit_fn_text[svgd-kwargs4-None]
10.40s call     tests/variational/test_inference.py::test_multiple_minibatch_variables
10.22s call     tests/variational/test_inference.py::test_fit_with_nans[True]
10.16s call     tests/variational/test_inference.py::test_fit_oo[FullRankADVI-mini]
9.76s call     tests/variational/test_inference.py::test_sample_posterior_after_minibatch
7.68s call     tests/variational/test_inference.py::TestUntransformedData::test_state_full_rank
7.62s call     tests/variational/test_inference.py::test_var_replacement
7.28s call     tests/variational/test_approximations.py::test_empirical_from_trace
6.99s call     tests/variational/test_approximations.py::test_elbo
6.77s call     tests/variational/test_inference.py::TestUntransformedData::test_state_empirical_std_is_none
6.65s call     tests/variational/test_inference.py::test_fit_oo[ADVI-full]
6.51s call     tests/variational/test_inference.py::test_fit_fn_text[fullrank_advi-kwargs3-None]
6.34s call     tests/variational/test_inference.py::test_fit_data[ASVGD-full]
5.99s call     tests/variational/test_inference.py::test_fit_fn_text[advi-kwargs2-None]
5.70s call     tests/variational/test_inference.py::test_fit_fn_text[svgd-kwargs5-None]
5.50s call     tests/variational/test_opvi.py::test_logq_globals[FullRankGroup: None, MeanFieldGroup: ['one']]
5.31s call     tests/variational/test_inference.py::test_fit_with_nans[False]
5.16s call     tests/variational/test_inference.py::test_fit_data[FullRankADVI-mini]
5.06s call     tests/variational/test_inference.py::TestUntransformedData::test_state_mean_field
4.96s call     tests/variational/test_inference.py::test_fit_data[ADVI-mini]
4.84s call     tests/variational/test_opvi.py::test_init_groups[raises4-grouping4]
4.67s call     tests/variational/test_inference.py::test_profile[FullRankADVI-full]
4.47s call     tests/variational/test_inference.py::test_profile[FullRankADVI-mini]
4.39s call     tests/variational/test_inference.py::TestUntransformedData::test_state_include_transformed
4.17s call     tests/variational/test_inference.py::test_sample_replacements[FullRankADVI]
4.10s call     tests/variational/test_approximations.py::test_empirical_does_not_support_inference_data
4.08s call     tests/variational/test_opvi.py::test_init_groups[raises0-grouping0]
3.73s call     tests/variational/test_inference.py::test_profile[ADVI-full]
3.69s call     tests/variational/test_inference.py::test_profile[ADVI-mini]
3.56s call     tests/variational/test_inference.py::test_fit_data[FullRankADVI-full]
3.27s call     tests/variational/test_inference.py::test_remove_scan_op
3.22s call     tests/variational/test_approximations.py::test_scale_cost_to_minibatch_works[5]
3.12s call     tests/variational/test_opvi.py::test_logq_globals[MeanFieldGroup: None]
3.02s call     tests/variational/test_inference.py::test_sample_replacements[SVGD]
2.98s call     tests/variational/test_opvi.py::test_init_groups[raises1-grouping1]
2.80s call     tests/variational/test_approximations.py::test_scale_cost_to_minibatch_works[8]
2.37s call     tests/variational/test_inference.py::TestUntransformedData::test_state_is_single_group_approx_attr
=========== 132 passed, 10 skipped, 3 warnings in 618.04s (0:10:18) ===========
============================ slowest 50 durations =============================
20.48s call     tests/sampling/test_mcmc.py::test_step_args
19.48s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
13.24s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_separable
12.42s call     tests/sampling/test_mcmc.py::test_sample_var_names
11.86s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-advi]
11.09s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-advi]
11.09s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-advi]
11.06s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
9.50s call     tests/model/test_core.py::test_model_d2logp[True]
9.49s call     tests/model/test_core.py::TestImputationMissingData::test_missing_dual_observations
8.68s call     tests/sampling/test_mcmc.py::test_blas_cores
8.46s call     tests/sampling/test_mcmc.py::test_sample_find_MAP_does_not_modify_start
7.12s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
6.76s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-2-advi]
6.72s call     tests/sampling/test_mcmc.py::TestSample::test_sample_does_not_rely_on_external_global_seeding
6.48s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-map]
6.46s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-map]
6.32s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-map]
6.31s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-advi]
6.19s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-1-advi]
6.18s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-1-advi]
6.17s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-1-advi]
6.14s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-2-advi]
6.11s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-2-advi]
6.08s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-1-advi]
5.94s call     tests/model/test_core.py::TestNested::test_nested_model_to_netcdf
5.72s call     tests/model/test_core.py::TestImputationMissingData::test_coordinates
5.68s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[map]
5.52s call     tests/model/test_core.py::TestTransformArgs::test_default_transform_is_applied
5.51s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi_map]
5.47s call     tests/sampling/test_mcmc.py::TestShared::test_sample
5.46s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-jitter+adapt_diag]
5.42s call     tests/model/test_core.py::TestImputationMissingData::test_interval_missing_observations
5.35s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-jitter+adapt_diag]
5.35s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-jitter+adapt_diag]
5.21s call     tests/model/test_core.py::TestImputationMissingData::test_missing_basic[pandas]
4.80s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[False]
4.42s call     tests/sampling/test_mcmc.py::TestType::test_float32
4.27s call     tests/model/test_core.py::TestValueGradFunction::test_missing_data
4.19s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[True]
4.09s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-map]
3.99s call     tests/model/test_core.py::TestImputationMissingData::test_missing_with_predictors
3.81s call     tests/model/test_core.py::test_model_logp_fast_compile
3.80s call     tests/sampling/test_mcmc.py::TestSample::test_default_sample_does_not_set_global_seed
3.77s call     tests/sampling/test_mcmc.py::TestSample::test_sample_callback
3.73s call     tests/sampling/test_mcmc.py::test_setup_chain_receives_draw_counts[multiprocess]
3.71s call     tests/sampling/test_mcmc.py::test_no_init_nuts_compound
3.67s call     tests/sampling/test_mcmc.py::TestSampleReturn::test_logs_sampler_warnings[2]
3.49s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[ADVI+adapt_diag]
3.35s call     tests/sampling/test_mcmc.py::TestType::test_float64
========== 257 passed, 6 skipped, 209 warnings in 496.15s (0:08:16) ===========
============================ slowest 50 durations =============================
63.54s setup    tests/step_methods/test_metropolis.py::TestMetropolisUniform::test_kstest
43.86s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist4]
43.32s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist3]
38.18s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSLKJCholeskyCov::test_kstest
20.68s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSUniform::test_kstest
20.26s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSUniform3::test_kstest
19.41s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSUniform2::test_kstest
13.25s setup    tests/step_methods/test_slicer.py::TestSliceUniform::test_kstest
11.63s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSBetaBinomial::test_kstest
11.39s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSNormal::test_kstest
11.19s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSStudentT::test_kstest
8.09s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist2]
6.98s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_multiple_samplers
6.59s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist1]
6.26s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_emits_energy_warnings
5.88s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist0]
5.11s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update_different_scales
4.79s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_continuous[<function TestStepMetropolis.<lambda> at 0x000001C1589FC180>-4000]
4.74s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_categorical[uniform]
4.14s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_discrete
3.77s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_epsilon_parallel
3.18s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_tuning_reset
2.70s call     tests/step_methods/test_compound.py::TestCompoundStep::test_name
2.54s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_bad_init_nonparallel
2.30s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_categorical[proportional]
2.12s call     tests/step_methods/test_slicer.py::TestStepSlicer::test_step_continuous[<function TestStepSlicer.<lambda> at 0x000001C1589FD6C0>-2000]
2.05s call     tests/step_methods/test_compound.py::TestStatsBijection::test_flatten_steps
1.91s call     tests/step_methods/test_compound.py::TestStepCompound::test_step_continuous[<function TestStepCompound.<lambda> at 0x000001C1589FF6A0>-1000]
1.74s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_sampler_stats
1.62s call     tests/step_methods/test_compound.py::TestCompoundStep::test_non_blocked
1.57s call     tests/step_methods/hmc/test_nuts.py::TestStepNUTS::test_step_continuous[<lambda>-1000_0]
1.53s call     tests/step_methods/hmc/test_nuts.py::TestStepNUTS::test_step_continuous[<lambda>-1000_1]
1.43s call     tests/step_methods/test_slicer.py::TestStepSlicer::test_step_continuous[<function TestStepSlicer.<lambda> at 0x000001C1589FD800>-2000]
1.40s call     tests/step_methods/test_metropolis.py::TestDEMetropolis::test_demcmc_tune_parameter
1.11s call     tests/step_methods/hmc/test_hmc.py::TestStepHamiltonianMC::test_step_continuous[<lambda>-1000_1]
1.09s call     tests/step_methods/hmc/test_hmc.py::TestStepHamiltonianMC::test_step_continuous[<lambda>-1000_0]
1.07s call     tests/step_methods/hmc/test_hmc.py::test_leapfrog_reversible
0.95s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_lambda_sequential
0.92s call     tests/step_methods/test_compound.py::TestStatsMetadata::test_stats_from_steps
0.91s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_reset
0.84s call     tests/step_methods/test_metropolis.py::TestRVsAssignmentMetropolis::test_discrete_steps[BinaryMetropolis]
0.79s call     tests/step_methods/test_compound.py::TestCompoundStep::test_blocked
0.68s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_none
0.29s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_custom_proposal_dist
0.28s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_multinomial_no_elemwise_update
0.27s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tune_drop_fraction
0.26s call     tests/step_methods/test_metropolis.py::TestRVsAssignmentMetropolis::test_discrete_steps[CategoricalGibbsMetropolis]
0.23s call     tests/step_methods/hmc/test_hmc.py::test_nuts_tuning
0.23s call     tests/step_methods/test_compound.py::TestRVsAssignmentCompound::test_compound_step
0.21s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_bad_init_parallel
=========== 118 passed, 6 skipped, 34 warnings in 408.54s (0:06:48) ===========
============================= slowest 50 durations =============================
18.77s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_log_jac_det
16.24s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
16.18s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a1-None]
15.08s call     tests/distributions/test_transform.py::test_simplex
11.87s call     tests/distributions/test_transform.py::test_interval
11.77s call     tests/distributions/test_dist_math.py::test_multinomial_check_parameters
11.68s call     tests/tuning/test_starting.py::test_find_MAP
11.50s call     tests/sampling/test_mcmc.py::test_step_args
10.40s call     tests/distributions/test_transform.py::test_simplex_bounds
10.14s call     tests/sampling/test_mcmc.py::test_sample_var_names
9.95s call     tests/distributions/test_transform.py::test_ordered
9.17s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_half_normal_ordered[sigma1-size1]
9.15s call     tests/tuning/test_starting.py::test_find_MAP_discrete
7.53s call     tests/distributions/test_dist_math.py::TestI0e::test_grad
7.39s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_backward
7.33s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_other[lower2-upper2-size2-transform2]
7.24s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
6.57s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_other[0.5-5.5-size1-transform1]
6.27s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
5.85s call     tests/sampling/test_mcmc.py::TestType::test_float32
5.84s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_mvnormal_transform[transform0-mu0-cov0-None-shape0]
5.72s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a0-None]
5.66s call     tests/sampling/test_mcmc.py::test_sample_find_MAP_does_not_modify_start
5.37s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_backward_produces_valid_cholesky_corr[3]
5.31s call     tests/sampling/test_parallel.py::test_remote_pipe_closed
5.30s call     tests/distributions/test_transform.py::TestCholeskyCovTransform::test_log_jac_det
5.04s call     tests/distributions/test_transform.py::test_logodds
5.02s call     tests/distributions/test_transform.py::test_log_exp_m1
4.60s call     tests/tuning/test_starting.py::test_find_MAP_issue_4488
4.28s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_triangular[-10-0-200-size1]
4.06s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi_map]
4.02s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_ordered[0.0-1.0-size0]
4.02s call     tests/sampling/test_mcmc.py::test_blas_cores
3.97s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform[0.5-5.5-size1]
3.93s call     tests/tuning/test_scaling.py::test_guess_scaling
3.93s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_ordered[lower1-upper1-size1]
3.91s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_other[0.0-1.0-size0-transform0]
3.87s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_beta[0.5-0.5-size1]
3.87s call     tests/distributions/test_transform.py::test_log
3.84s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_half_normal[5.0-size1]
3.78s call     tests/distributions/test_dist_math.py::test_logpow_stabilization_reaches_logp
3.69s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-advi]
3.69s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-advi]
3.67s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-advi]
3.65s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_half_normal_ordered[2.5-size0]
3.56s call     tests/distributions/test_transform.py::test_upperbound
3.47s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_vonmises[-0.5-5.5-size1]
3.43s call     tests/sampling/test_mcmc.py::TestType::test_float64
3.37s call     tests/distributions/test_transform.py::test_chain_jacob_det
3.29s call     tests/sampling/test_parallel.py::test_explicit_sample[spawn]
=========== 256 passed, 8 skipped, 14 warnings in 527.68s (0:08:47) ============
============================ slowest 50 durations =============================
32.21s setup    tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
20.89s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
15.49s call     tests/smc/test_smc.py::TestSMC::test_marginal_likelihood
15.04s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[VFE]
12.15s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_2_param_tensor
10.30s call     tests/smc/test_smc.py::TestMHKernel::test_normal_model
9.64s call     tests/smc/test_smc.py::TestSMC::test_return_datatype[2]
9.13s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_2_param
8.04s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_1_param
7.96s call     tests/sampling/test_parallel.py::test_remote_pipe_closed
6.72s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_1_param
6.57s call     tests/smc/test_smc.py::TestSMC::test_sample[sequential]
6.47s call     tests/ode/test_ode.py::test_simulate
6.44s call     tests/smc/test_smc.py::TestSMC::test_sample[parallel]
5.84s call     tests/smc/test_smc.py::TestSMC::test_parallel_custom
5.54s call     tests/sampling/test_parallel.py::test_sampling_with_random_generator_matches[2]
5.53s call     tests/smc/test_smc.py::test_smc_with_custom_op
5.33s call     tests/sampling/test_parallel.py::test_spawn_densitydist_bound_method
5.13s call     tests/sampling/test_parallel.py::test_iterator
5.10s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_3_params
4.98s call     tests/smc/test_smc.py::TestSMC::test_unobserved_categorical
4.65s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveLatent
4.57s call     tests/smc/test_smc.py::TestSMC::test_convergence_checks
4.56s call     tests/gp/test_cov.py::TestCovAdd::test_rightadd_whitenoise
4.40s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[DTC]
4.39s call     tests/gp/test_cov.py::TestCovAdd::test_symadd_cov
4.38s call     tests/smc/test_smc.py::TestSMC::test_start
4.36s call     tests/gp/test_cov.py::TestCovKron::test_multiops
4.36s call     tests/smc/test_smc.py::TestSMC::test_unobserved_bernoulli
4.26s setup    tests/gp/test_gp.py::TestMarginalKron::testMarginalKronvsMarginalpredict
4.09s call     tests/smc/test_smc.py::TestSMC::test_ordered
3.99s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[FITC]
3.94s call     tests/gp/test_cov.py::TestStability::test_stable
3.93s call     tests/sampling/test_parallel.py::test_explicit_sample[spawn]
3.93s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[VFE]
3.90s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[DTC]
3.90s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_2_param
3.79s call     tests/gp/test_gp.py::TestMarginalVsLatent::testLatentMultioutput
3.64s call     tests/gp/test_gp.py::TestLatentKron::testLatentKronvsLatent
3.57s call     tests/sampling/test_parallel.py::test_context
3.35s setup    tests/gp/test_gp.py::TestMarginalKron::testMarginalKronvsMarginal
3.32s setup    tests/gp/test_gp.py::TestMarginalKron::testMarginalKronRaises
3.25s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginal
3.25s call     tests/sampling/test_parallel.py::test_spawn_densitydist_function
3.10s call     tests/gp/test_gp.py::TestTP::testTPvsLatent
2.90s call     tests/sampling/test_parallel.py::test_bad_unpickle
2.77s call     tests/gp/test_cov.py::TestCovExponentiation::test_covexp_shared
2.63s call     tests/gp/test_gp.py::TestMarginalKron::testMarginalKronvsMarginalpredict
2.43s call     tests/ode/test_ode.py::TestDiffEqModel::test_scalar_ode_2_param
2.42s call     tests/ode/test_ode.py::TestDiffEqModel::test_scalar_ode_1_param
=========== 180 passed, 5 skipped, 28 warnings in 375.91s (0:06:15) ===========
============================= slowest 50 durations =============================
45.48s call     tests/dims/test_model.py::test_complex_model
24.84s call     tests/dims/test_model.py::test_simple_model
18.88s call     tests/dims/distributions/test_vector.py::test_dirichlet
15.81s call     tests/dims/distributions/test_core.py::test_multivariate_distribution_dims
13.18s call     tests/dims/distributions/test_vector.py::test_zerosumnormal_batch_sigma
13.17s call     tests/dims/test_model.py::test_zerosumnormal_model
7.76s call     tests/dims/distributions/test_core.py::test_density_helpers[True-True-True]
7.18s call     tests/dims/distributions/test_core.py::test_density_helpers[True-False-True]
6.98s call     tests/dims/distributions/test_core.py::test_density_helpers[True-False-False]
6.74s call     tests/dims/distributions/test_core.py::test_density_helpers[True-True-False]
4.81s call     tests/dims/distributions/test_core.py::test_distribution_dims
3.79s call     tests/dims/distributions/test_vector.py::test_categorical
3.78s call     tests/dims/distributions/test_scalar.py::test_uniform_transposed
0.68s call     tests/dims/distributions/test_core.py::test_density_helpers[False-True-True]
0.67s call     tests/dims/distributions/test_core.py::test_density_helpers[False-True-False]
0.55s call     tests/dims/test_model.py::test_data
0.44s call     tests/dims/distributions/test_scalar.py::test_truncated_normal
0.39s call     tests/dims/distributions/test_scalar.py::test_studentt
0.22s call     tests/dims/distributions/test_scalar.py::test_halfstudentt
0.22s call     tests/dims/distributions/test_scalar.py::test_uniform
0.20s call     tests/dims/distributions/test_scalar.py::test_beta
0.17s call     tests/dims/distributions/test_core.py::test_density_helpers[False-False-False]
0.17s call     tests/dims/distributions/test_scalar.py::test_inverse_gamma
0.17s call     tests/dims/distributions/test_core.py::test_density_helpers[False-False-True]
0.16s call     tests/dims/distributions/test_scalar.py::test_lognormal
0.14s call     tests/dims/distributions/test_censored.py::test_censored_basic[1--1]
0.14s call     tests/dims/distributions/test_scalar.py::test_halfnormal
0.14s call     tests/dims/distributions/test_vector.py::test_zerosumnormal
0.13s call     tests/dims/distributions/test_scalar.py::test_normal
0.12s call     tests/dims/distributions/test_scalar.py::test_negative_binomial
0.12s call     tests/dims/distributions/test_vector.py::test_mvnormal
0.12s call     tests/dims/distributions/test_censored.py::test_censored_basic[1-None]
0.11s call     tests/dims/distributions/test_scalar.py::test_exponential
0.11s call     tests/dims/distributions/test_censored.py::test_censored_basic[None--1]
0.10s call     tests/dims/distributions/test_scalar.py::test_gamma
0.08s call     tests/dims/distributions/test_censored.py::test_censored_basic[None-None]
0.07s call     tests/dims/distributions/test_scalar.py::test_laplace
0.07s call     tests/dims/distributions/test_scalar.py::test_weibull
0.06s call     tests/dims/distributions/test_scalar.py::test_halfcauchy
0.06s call     tests/dims/distributions/test_scalar.py::test_poisson
0.05s call     tests/dims/test_model.py::test_model_table_dims
0.05s call     tests/dims/distributions/test_scalar.py::test_cauchy
0.03s call     tests/dims/distributions/test_scalar.py::test_halfflat
0.03s call     tests/dims/distributions/test_scalar.py::test_diracdelta[float]
0.03s call     tests/dims/distributions/test_scalar.py::test_diracdelta[int]
0.02s call     tests/dims/distributions/test_scalar.py::test_diracdelta_observed
0.02s call     tests/dims/distributions/test_scalar.py::test_diracdelta_scalar
0.02s call     tests/dims/distributions/test_scalar.py::test_flat
0.01s call     tests/dims/test_model.py::test_model_table_xtensor_without_registered_dims
0.01s call     tests/dims/distributions/test_censored.py::test_censored_dims
======================== 52 passed in 186.30s (0:03:06) ========================
============================= slowest 50 durations =============================
17.10s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_log_jac_det
10.86s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
10.29s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a0-None]
9.70s call     tests/sampling/test_mcmc.py::test_step_args
9.28s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_backward
8.71s call     tests/distributions/test_dist_math.py::TestI0e::test_grad
8.11s call     tests/distributions/test_transform.py::test_simplex_bounds
7.74s call     tests/distributions/test_transform.py::test_ordered
7.62s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a1-None]
7.61s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_half_normal_ordered[sigma1-size1]
6.93s call     tests/sampling/test_mcmc.py::test_sample_var_names
6.31s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_mvnormal_transform[transform0-mu0-cov0-None-shape0]
5.88s call     tests/distributions/test_transform.py::test_interval
5.87s call     tests/distributions/test_transform.py::TestLJKCholeskyCorrTransform::test_backward_produces_valid_cholesky_corr[3]
5.71s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_beta_ordered[a1-b1-size1]
5.68s call     tests/distributions/test_transform.py::TestCholeskyCovTransform::test_log_jac_det
5.67s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
5.43s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-advi]
5.42s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[map]
5.41s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-advi]
5.34s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-advi]
4.75s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
4.52s call     tests/tuning/test_starting.py::test_find_MAP
4.47s call     tests/distributions/test_transform.py::test_simplex
4.29s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-2-advi]
4.19s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a2-size2]
4.19s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_other[0.5-5.5-size1-transform1]
4.11s call     tests/sampling/test_mcmc.py::test_sample_find_MAP_does_not_modify_start
4.05s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_vonmises_ordered[mu1-kappa1-size1]
4.05s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_ordered[lower1-upper1-size1]
4.04s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_mvnormal_transform[transform0-mu1-cov1-size1-shape1]
4.02s call     tests/distributions/test_dist_math.py::test_multinomial_check_parameters
3.91s call     tests/sampling/test_mcmc.py::test_blas_cores
3.77s call     tests/distributions/test_transform.py::test_logodds
3.54s call     tests/tuning/test_starting.py::test_find_MAP_discrete
3.40s call     tests/distributions/test_transform.py::test_chain_jacob_det
3.38s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-1-advi]
3.37s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-jitter+adapt_diag]
3.34s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-1-advi]
3.33s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-advi]
3.33s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-1-advi]
3.30s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-1-advi]
3.27s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-2-advi]
3.27s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-2-advi]
3.18s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-map]
3.13s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-map]
3.12s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-map]
3.11s call     tests/sampling/test_mcmc.py::TestSample::test_sample_does_not_rely_on_external_global_seeding
2.98s call     tests/distributions/test_transform.py::test_log_exp_m1
2.94s call     tests/distributions/test_transform.py::test_sum_to_1
=========== 256 passed, 8 skipped, 22 warnings in 435.61s (0:07:15) ============
============================= slowest 50 durations =============================
24.03s call     tests/logprob/test_transform_value.py::test_default_value_transform_logprob[pt_dist17-dist_params17-<lambda>-size17]
16.95s call     tests/logprob/test_transform_value.py::test_default_value_transform_logprob[pt_dist16-dist_params16-<lambda>-None]
14.06s call     tests/backends/test_mcbackend.py::test_make_runmeta_and_point_fn
13.31s call     tests/logprob/test_transforms.py::TestTransform::test_check_jac_det[transform2]
11.51s call     tests/logprob/test_transform_value.py::test_default_value_transform_logprob[pt_dist15-dist_params15-<lambda>-size15]
10.17s call     tests/distributions/test_truncated.py::test_truncated_multiple_rngs
9.17s call     tests/logprob/test_transforms.py::test_extra_bijective_rv_transforms[pt_transform2-transform2]
8.47s setup    tests/backends/test_zarr.py::test_record[single_step-include_transformed-frequent_writes]
8.41s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[size19-10-size29-10-2-False-2]
8.24s call     tests/logprob/test_scan.py::test_scan_joint_logprob[False]
7.28s call     tests/distributions/test_truncated.py::test_truncated_inference
6.81s call     tests/distributions/test_truncated.py::test_truncation_continuous_random[False-True-rejection--1-inf]
6.55s call     tests/backends/test_mcbackend.py::TestMcBackendSampling::test_return_multitrace[False]
6.09s call     tests/logprob/test_binary.py::test_discrete_rv_comparison_bitwise[comparison_op0-<lambda>-<lambda>-inputs0]
6.02s call     tests/logprob/test_mixture.py::test_hetero_mixture_binomial[p_val2-size2-True]
5.92s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[size15-10-size25-10-1-True-1]
5.90s call     tests/distributions/test_truncated.py::test_truncated_custom_dist_indexed_argument[rv_op0]
5.51s call     tests/logprob/test_transform_value.py::test_scan_transform
5.40s call     tests/logprob/test_composite_logprob.py::test_affine_join_interdependent[False]
5.02s call     tests/logprob/test_transforms.py::test_monotonically_decreasing_transform_logcdf[pt_transform1-transform1]
4.95s call     tests/logprob/test_mixture.py::test_switch_mixture_vector[True]
4.84s call     tests/logprob/test_transforms.py::TestLocScaleRVTransform::test_scale_transform_rv[rv_size2-matrix-False]
4.82s call     tests/distributions/test_truncated.py::test_truncation_continuous_random[False-False-rejection--1-inf]
4.59s call     tests/logprob/test_transforms.py::TestLocScaleRVTransform::test_loc_transform_rv[rv_size2-col-True]
4.54s call     tests/logprob/test_cumsum.py::test_bernoulli_cumsum[size4-0]
4.48s call     tests/logprob/test_cumsum.py::test_normal_cumsum[size4-0]
4.46s call     tests/logprob/test_cumsum.py::test_bernoulli_cumsum[size2-0]
4.40s call     tests/distributions/test_truncated.py::test_change_truncated_size
4.36s call     tests/logprob/test_transforms.py::TestTransform::test_check_jac_det[transform0]
4.35s call     tests/logprob/test_tensor.py::TestMeasurableIdentityOps::test_scalar_from_tensor
4.24s call     tests/logprob/test_tensor.py::TestMeasurableJoinSplitDims::test_multivariate_indirect_join_within_batch
4.16s call     tests/distributions/test_truncated.py::test_vectorized_bounds
4.15s call     tests/backends/test_mcbackend.py::test_get_variables_and_point_fn
4.12s call     tests/logprob/test_scan.py::test_scan_non_pure_rv_output
4.01s call     tests/logprob/test_tensor.py::TestMeasurableJoinSplitDims::test_multivariate_indirect_join_within_support
3.94s call     tests/logprob/test_cumsum.py::test_normal_cumsum[size2-0]
3.94s call     tests/logprob/test_mixture.py::test_scalar_switch_mixture
3.90s call     tests/logprob/test_tensor.py::test_measurable_broadcast
3.82s call     tests/logprob/test_transform_value.py::test_weakref_leak
3.78s call     tests/logprob/test_mixture.py::test_hetero_mixture_categorical[X_args6-Y_args6-Z_args6-p_val6-comp_size6-idx_size6-extra_indices6-0-True]
3.77s call     tests/logprob/test_mixture.py::test_mixture_basics
3.76s call     tests/logprob/test_cumsum.py::test_bernoulli_cumsum[size5-1]
3.75s call     tests/logprob/test_tensor.py::test_measurable_join_univariate[size16-size26-2-False]
3.75s call     tests/logprob/test_tensor.py::test_measurable_join_univariate[size14-size24-0-False]
3.75s call     tests/logprob/test_cumsum.py::test_normal_cumsum[size5-1]
3.74s call     tests/logprob/test_tensor.py::test_measurable_join_univariate[size15-size25-1-False]
3.68s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[size17-5-size27-5-0-False-0]
3.68s call     tests/logprob/test_transforms.py::TestLocScaleRVTransform::test_scale_transform_rv[1-scale_type1-True]
3.63s call     tests/logprob/test_checks.py::test_specify_shape_logprob
3.60s call     tests/logprob/test_transforms.py::TestLocScaleRVTransform::test_scale_transform_rv[rv_size0-scalar-True]
====
============================ slowest 50 durations =============================
41.80s setup    tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
22.14s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
15.45s call     tests/smc/test_smc.py::TestSMC::test_marginal_likelihood
14.95s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[VFE]
12.31s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[DTC]
10.41s call     tests/smc/test_smc.py::TestMHKernel::test_normal_model
10.27s call     tests/smc/test_smc.py::TestSMC::test_return_datatype[2]
10.15s setup    tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[DTC]
10.05s setup    tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[VFE]
9.63s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_1_param
9.11s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_2_param
8.28s call     tests/smc/test_smc.py::test_smc_with_custom_op
7.46s call     tests/smc/test_smc.py::TestSMC::test_unobserved_categorical
7.29s call     tests/smc/test_smc.py::TestSMC::test_sample[parallel]
6.97s call     tests/gp/test_gp.py::TestMarginalVsLatent::testLatentMultioutput
6.75s call     tests/gp/test_gp.py::TestLatentKron::testLatentKronvsLatent
6.62s call     tests/smc/test_smc.py::TestSMC::test_sample[sequential]
6.21s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveLatent
6.16s call     tests/ode/test_ode.py::test_simulate
6.09s call     tests/sampling/test_parallel.py::test_sampling_with_random_generator_matches[2]
6.03s call     tests/gp/test_cov.py::TestCovAdd::test_rightadd_whitenoise
5.94s call     tests/smc/test_smc.py::TestSMC::test_parallel_custom
5.44s call     tests/sampling/test_parallel.py::test_iterator
5.26s call     tests/sampling/test_parallel.py::test_remote_pipe_closed
5.22s call     tests/smc/test_smc.py::TestSMC::test_convergence_checks
4.97s call     tests/sampling/test_parallel.py::test_spawn_densitydist_bound_method
4.76s call     tests/gp/test_cov.py::TestCovKron::test_multiops
4.23s call     tests/gp/test_cov.py::TestCovAdd::test_symadd_cov
4.12s setup    tests/gp/test_gp.py::TestMarginalKron::testMarginalKronvsMarginalpredict
4.11s call     tests/smc/test_smc.py::TestSMC::test_unobserved_bernoulli
4.07s call     tests/ode/test_ode.py::TestDiffEqModel::test_scalar_ode_2_param
4.00s call     tests/smc/test_smc.py::TestSMC::test_start
3.84s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginal
3.83s call     tests/gp/test_gp.py::TestMarginalVsLatent::testLatent2
3.78s call     tests/sampling/test_parallel.py::test_context
3.60s call     tests/gp/test_cov.py::TestStability::test_stable
3.55s call     tests/sampling/test_parallel.py::test_explicit_sample[spawn]
3.51s call     tests/gp/test_gp.py::TestTP::testTPvsLatent
3.50s call     tests/sampling/test_parallel.py::test_spawn_densitydist_function
3.47s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[FITC]
3.42s setup    tests/gp/test_gp.py::TestMarginalKron::testMarginalKronRaises
3.38s setup    tests/gp/test_gp.py::TestMarginalKron::testMarginalKronvsMarginal
3.31s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[DTC]
3.17s call     tests/ode/test_ode.py::TestDiffEqModel::test_scalar_ode_1_param
3.10s call     tests/gp/test_cov.py::TestWhiteNoise::test_1d
3.06s setup    tests/gp/test_gp.py::TestLatentKron::testLatentKronvsLatent
3.05s call     tests/sampling/test_parallel.py::test_bad_unpickle
3.01s call     tests/gp/test_gp.py::TestTP::testTPvsLatentReparameterized
2.83s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[VFE]
2.73s call     tests/smc/test_smc.py::TestSMC::test_discrete_rounding_proposal
=========== 180 passed, 5 skipped, 42 warnings in 414.68s (0:06:54) ===========
============================= slowest 50 durations =============================
19.64s call     tests/logprob/test_transform_value.py::test_default_value_transform_logprob[pt_dist17-dist_params17-<lambda>-size17]
18.60s call     tests/logprob/test_transform_value.py::test_weakref_leak
15.27s call     tests/distributions/test_truncated.py::test_truncated_multiple_rngs
13.20s call     tests/logprob/test_transforms.py::TestTransform::test_check_jac_det[transform2]
12.94s call     tests/logprob/test_transform_value.py::test_default_value_transform_logprob[pt_dist16-dist_params16-<lambda>-None]
8.02s call     tests/logprob/test_composite_logprob.py::test_affine_join_interdependent[False]
8.02s call     tests/logprob/test_transforms.py::test_extra_bijective_rv_transforms[pt_transform2-transform2]
7.56s call     tests/logprob/test_scan.py::test_convert_outer_out_to_in_sit_sot
7.38s call     tests/logprob/test_scan.py::test_scan_joint_logprob[False]
6.56s call     tests/logprob/test_mixture.py::test_switch_mixture_vector[True]
5.96s call     tests/logprob/test_transform_value.py::test_default_value_transform_logprob[pt_dist15-dist_params15-<lambda>-size15]
5.91s call     tests/logprob/test_scan.py::test_generative_graph_unchanged
5.38s call     tests/logprob/test_binary.py::test_discrete_rv_comparison_bitwise[comparison_op0-<lambda>-<lambda>-inputs0]
4.99s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[size19-10-size29-10-2-False-2]
4.85s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[size15-10-size25-10-1-True-1]
4.80s call     tests/logprob/test_tensor.py::test_measurable_join_univariate[size15-size25-1-False]
4.35s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[None-2-None-2-0-True-0]
4.35s call     tests/backends/test_mcbackend.py::TestMcBackendSampling::test_return_multitrace[False]
4.26s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[size12-2-size22-2-0-True-0]
4.24s call     tests/backends/test_zarr.py::test_record[single_step-include_transformed-frequent_writes]
4.22s call     tests/distributions/test_truncated.py::test_truncated_inference
4.06s call     tests/logprob/test_transform_value.py::test_scan_transform
3.92s call     tests/logprob/test_scan.py::test_convert_outer_out_to_in_mit_sot
3.43s call     tests/logprob/test_tensor.py::TestMeasurableSplit::test_multivariate
3.24s call     tests/logprob/test_transforms.py::test_monotonically_decreasing_transform_logcdf[pt_transform1-transform1]
3.20s call     tests/logprob/test_transforms.py::TestTransform::test_check_jac_det[transform0]
3.15s call     tests/backends/test_mcbackend.py::TestMcBackendSampling::test_return_inferencedata[1]
3.10s call     tests/logprob/test_transforms.py::TestTransform::test_fallback_log_jac_det[1]
3.08s call     tests/logprob/test_tensor.py::test_measurable_join_univariate[size12-size22-0-True]
3.03s call     tests/distributions/test_truncated.py::test_truncation_continuous_random[False-False-rejection--1-inf]
2.97s call     tests/distributions/test_truncated.py::test_change_truncated_size
2.96s call     tests/logprob/test_mixture.py::test_hetero_mixture_categorical[X_args1-Y_args1-Z_args1-p_val1-None-idx_size1-extra_indices1-0-True]
2.82s call     tests/logprob/test_tensor.py::test_measurable_join_univariate[size14-size24-0-False]
2.69s call     tests/backends/test_mcbackend.py::TestMcBackendSampling::test_return_inferencedata[3]
2.69s call     tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-discard_tuning-return_idata-discard_warning_stat-parallel_sampling-compute_loglike-sparse_writes]
2.65s call     tests/logprob/test_tensor.py::test_measurable_join_with_constant_input
2.62s call     tests/logprob/test_scan.py::test_scan_joint_logprob[True]
2.61s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[size17-5-size27-5-0-False-0]
2.58s call     tests/logprob/test_transforms.py::TestLocScaleRVTransform::test_scale_transform_rv[rv_size2-matrix-False]
2.52s call     tests/logprob/test_tensor.py::test_measurable_join_univariate[size16-size26-2-False]
2.49s call     tests/logprob/test_transforms.py::TestPowerRVTransform::test_reciprocal_rv_transform[1.0]
2.47s call     tests/logprob/test_composite_logprob.py::test_affine_join_interdependent[True]
2.44s call     tests/logprob/test_scan.py::test_scan_multiple_output_types
2.44s call     tests/logprob/test_composite_logprob.py::test_nested_scalar_mixtures
2.44s call     tests/logprob/test_tensor.py::TestMeasurableSplit::test_univariate
2.42s call     tests/distributions/test_truncated.py::test_truncation_discrete_random[rejection-2-inf]
2.42s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-keep_tuning-return_zarr-discard_warning_stat-parallel_sampling-compute_loglike-sparse_writes]
2.37s call     tests/logprob/test_tensor.py::test_unmeasurable_dimshuffles
2.35s call     tests/backends/test_zarr.py::test_sampling_consistency[compound_step-frequent_writes]
2.31s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-discard_tuning-return_zarr-keep_warning_stat-parallel_sampling-compute_loglike-sparse_writes]
==== 768 passed, 64 skipped, 15 xfailed, 1936 warnings in 881.42s (0:14:41) ====
============================ slowest 50 durations =============================
23.25s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
22.51s call     tests/sampling/test_mcmc.py::test_step_args
16.24s call     tests/sampling/test_mcmc.py::test_sample_var_names
15.69s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_separable
12.50s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
10.04s call     tests/model/test_core.py::TestImputationMissingData::test_missing_dual_observations
9.80s call     tests/sampling/test_mcmc.py::test_sample_find_MAP_does_not_modify_start
8.86s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
8.66s call     tests/sampling/test_mcmc.py::test_blas_cores
8.56s call     tests/sampling/test_mcmc.py::TestType::test_float32
8.50s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-advi]
8.40s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-advi]
8.39s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-advi]
8.18s call     tests/model/test_core.py::TestValueGradFunction::test_missing_data
7.32s call     tests/model/test_core.py::TestNested::test_nested_model_to_netcdf
7.31s call     tests/sampling/test_mcmc.py::TestSample::test_sample_does_not_rely_on_external_global_seeding
6.98s call     tests/model/test_core.py::TestImputationMissingData::test_coordinates
6.90s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-map]
6.86s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-map]
6.82s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-map]
6.82s call     tests/model/test_core.py::TestTransformArgs::test_default_transform_is_applied
6.61s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter[logp]
6.47s call     tests/model/test_core.py::TestValueGradFunction::test_edge_case
6.28s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[False]
5.63s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-jitter+adapt_diag]
5.55s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-jitter+adapt_diag]
5.55s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-jitter+adapt_diag]
5.41s call     tests/model/test_core.py::test_model_d2logp[True]
5.30s call     tests/model/test_core.py::TestImputationMissingData::test_interval_missing_observations
5.19s call     tests/model/test_core.py::TestModelDebug::test_invalid_observed_value
5.13s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[map]
4.97s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[True]
4.88s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi_map]
4.54s call     tests/sampling/test_mcmc.py::TestType::test_float64
4.36s call     tests/sampling/test_mcmc.py::TestSample::test_default_sample_does_not_set_global_seed
4.31s call     tests/model/test_core.py::TestImputationMissingData::test_truncated_normal
4.28s call     tests/sampling/test_mcmc.py::TestShared::test_sample
4.19s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter_cant_be_evaluated[logp-True]
3.96s call     tests/sampling/test_mcmc.py::test_no_init_nuts_compound
3.81s call     tests/sampling/test_mcmc.py::test_setup_chain_receives_draw_counts[multiprocess]
3.75s call     tests/sampling/test_mcmc.py::TestSampleReturn::test_logs_sampler_warnings[2]
3.64s call     tests/sampling/test_mcmc.py::TestSample::test_sample_callback
3.63s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-advi]
3.56s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-1-advi]
3.55s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-2-advi]
3.43s call     tests/model/test_core.py::TestPytensorRelatedLogpBugs::test_pytensor_switch_broadcast_edge_cases_2
3.39s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-1-advi]
3.34s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-2-advi]
3.34s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-1-advi]
3.33s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-1-advi]
========== 257 passed, 6 skipped, 209 warnings in 502.79s (0:08:22) ===========
============================= slowest 50 durations =============================
21.09s call     tests/distributions/test_timeseries.py::TestAR::test_batched_sigma
17.10s setup    tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
16.51s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_prior[cov_func0-centered]
15.49s call     tests/gp/test_hsgp_approx.py::TestHSGPPeriodic::test_prior[100.0-cov_func0]
14.70s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_conditional[cov_func0-centered]
14.63s call     tests/gp/test_hsgp_approx.py::TestHSGPPeriodic::test_conditional_periodic[cov_func0]
14.47s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_conditional[cov_func1-noncentered]
14.41s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_prior[cov_func1-noncentered]
14.22s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[True-omega]
12.25s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
11.11s call     tests/distributions/test_timeseries.py::TestGARCH11::test_support_point[size1-expected1]
10.46s call     tests/distributions/test_timeseries.py::TestPredefinedRandomWalk::test_mvgaussian_with_chol_cov_rv[chol]
8.11s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[True-beta_1]
7.91s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[VFE]
7.64s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[False-beta_1]
7.59s call     tests/distributions/test_timeseries.py::TestAR::test_support_point[size1-expected1]
7.59s call     tests/distributions/test_timeseries.py::TestAR::test_batched_size[False]
7.28s call     tests/model/test_core.py::test_model_logp_fast_compile
7.11s call     tests/distributions/test_timeseries.py::TestEulerMaruyama::test_linear_model
6.90s setup    tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[DTC]
6.85s call     tests/distributions/test_timeseries.py::TestPredefinedRandomWalk::test_gaussian_inference
6.81s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[DTC]
6.70s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[True-alpha_1]
6.55s call     tests/distributions/test_timeseries.py::TestEulerMaruyama::test_batched_size[True-1]
6.39s call     tests/distributions/test_timeseries.py::TestGARCH11::test_support_point[None-expected0]
6.07s call     tests/model/test_fgraph.py::test_multivariate_transform
5.86s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[False-omega]
5.81s call     tests/distributions/test_timeseries.py::TestAR::test_batched_rhos
5.79s setup    tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[VFE]
5.74s call     tests/distributions/test_timeseries.py::TestRandomWalk::test_init_logp_multivariate[init_dist0]
5.72s call     tests/distributions/test_timeseries.py::TestEulerMaruyama::test_change_dist_size2
5.61s call     tests/distributions/test_timeseries.py::TestAR::test_change_dist_size
5.54s call     tests/model/test_core.py::TestImputationMissingData::test_missing_dual_observations
5.52s call     tests/distributions/test_timeseries.py::TestRandomWalk::test_support_point[init_dist4-innovation_dist4-9-size4-expected4-False]
5.20s call     tests/distributions/test_timeseries.py::TestAR::test_support_point[None-expected0]
5.20s call     tests/model/test_core.py::test_model_d2logp[True]
5.15s call     tests/distributions/test_timeseries.py::TestRandomWalk::test_inconsistent_steps_and_shape[init_dist0-innovation_dist0-12-shape0]
5.13s call     tests/model/test_core.py::TestImputationMissingData::test_interval_missing_observations
5.06s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_1_param
5.06s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[False-alpha_1]
4.98s call     tests/gp/test_gp.py::TestLatentKron::testLatentKronvsLatent
4.72s call     tests/distributions/test_timeseries.py::TestAR::test_batched_init_dist
4.59s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[True-initial_vol]
4.58s call     tests/distributions/test_timeseries.py::TestAR::test_batched_size[True]
4.56s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveLatent
4.45s call     tests/distributions/test_timeseries.py::TestAR::test_multivariate_init_dist
4.44s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_separable
4.43s call     tests/distributions/test_timeseries.py::TestEulerMaruyama::test_change_dist_size1
4.28s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[False-initial_vol]
4.25s call     tests/distributions/test_timeseries.py::TestGARCH11::test_change_dist_size
=========== 498 passed, 2 xfailed, 41 warnings in 582.91s (0:09:42) ============
============================= slowest 50 durations =============================
31.48s call     tests/dims/test_model.py::test_complex_model
17.04s call     tests/dims/distributions/test_core.py::test_multivariate_distribution_dims
14.19s call     tests/dims/test_model.py::test_zerosumnormal_model
11.22s call     tests/dims/test_model.py::test_simple_model
11.06s call     tests/dims/distributions/test_vector.py::test_dirichlet
4.46s call     tests/dims/distributions/test_vector.py::test_categorical
4.16s call     tests/dims/distributions/test_core.py::test_distribution_dims
4.14s call     tests/dims/distributions/test_vector.py::test_zerosumnormal_batch_sigma
4.14s call     tests/dims/distributions/test_core.py::test_density_helpers[True-True-True]
4.10s call     tests/dims/distributions/test_core.py::test_density_helpers[True-False-True]
2.91s call     tests/dims/distributions/test_core.py::test_density_helpers[True-False-False]
2.57s call     tests/dims/distributions/test_core.py::test_density_helpers[False-True-True]
2.29s call     tests/dims/distributions/test_core.py::test_density_helpers[False-True-False]
1.90s call     tests/dims/distributions/test_core.py::test_density_helpers[True-True-False]
1.75s call     tests/dims/distributions/test_core.py::test_density_helpers[False-False-False]
1.19s call     tests/dims/distributions/test_scalar.py::test_uniform_transposed
0.59s call     tests/dims/distributions/test_vector.py::test_mvnormal
0.40s call     tests/dims/test_model.py::test_data
0.40s call     tests/dims/distributions/test_scalar.py::test_truncated_normal
0.23s call     tests/dims/distributions/test_core.py::test_density_helpers[False-False-True]
0.22s call     tests/dims/distributions/test_scalar.py::test_uniform
0.21s call     tests/dims/distributions/test_scalar.py::test_halfstudentt
0.19s call     tests/dims/distributions/test_scalar.py::test_studentt
0.18s call     tests/dims/distributions/test_scalar.py::test_beta
0.16s call     tests/dims/distributions/test_scalar.py::test_inverse_gamma
0.15s call     tests/dims/distributions/test_scalar.py::test_lognormal
0.14s call     tests/dims/distributions/test_censored.py::test_censored_basic[1--1]
0.14s call     tests/dims/distributions/test_scalar.py::test_halfnormal
0.12s call     tests/dims/distributions/test_vector.py::test_zerosumnormal
0.12s call     tests/dims/distributions/test_scalar.py::test_normal
0.11s call     tests/dims/distributions/test_scalar.py::test_negative_binomial
0.11s call     tests/dims/distributions/test_censored.py::test_censored_basic[1-None]
0.11s call     tests/dims/distributions/test_censored.py::test_censored_basic[None--1]
0.10s call     tests/dims/distributions/test_scalar.py::test_exponential
0.10s call     tests/dims/distributions/test_scalar.py::test_gamma
0.07s call     tests/dims/distributions/test_censored.py::test_censored_basic[None-None]
0.07s call     tests/dims/distributions/test_scalar.py::test_weibull
0.07s call     tests/dims/distributions/test_scalar.py::test_laplace
0.06s call     tests/dims/distributions/test_scalar.py::test_halfcauchy
0.06s call     tests/dims/distributions/test_scalar.py::test_poisson
0.05s call     tests/dims/distributions/test_scalar.py::test_cauchy
0.04s call     tests/dims/test_model.py::test_model_table_dims
0.03s call     tests/dims/distributions/test_scalar.py::test_halfflat
0.03s call     tests/dims/distributions/test_scalar.py::test_diracdelta[float]
0.03s call     tests/dims/distributions/test_scalar.py::test_diracdelta[int]
0.02s call     tests/dims/distributions/test_scalar.py::test_diracdelta_observed
0.02s call     tests/dims/distributions/test_scalar.py::test_flat
0.02s call     tests/dims/distributions/test_scalar.py::test_diracdelta_scalar
0.01s call     tests/dims/test_model.py::test_model_table_xtensor_without_registered_dims
0.01s call     tests/dims/distributions/test_censored.py::test_censored_dims
======================== 52 passed in 131.34s (0:02:11) ========================
============================= slowest 50 durations =============================
24.29s call     tests/distributions/test_timeseries.py::TestAR::test_batched_sigma
22.48s setup    tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
20.61s call     tests/distributions/test_timeseries.py::TestPredefinedRandomWalk::test_gaussian_inference
19.50s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_prior[cov_func0-centered]
17.37s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[True-omega]
17.22s call     tests/gp/test_hsgp_approx.py::TestHSGPPeriodic::test_prior[100.0-cov_func0]
17.09s call     tests/distributions/test_timeseries.py::TestAR::test_batched_size[False]
16.80s call     tests/gp/test_hsgp_approx.py::TestHSGPPeriodic::test_conditional_periodic[cov_func0]
16.39s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
15.82s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_conditional[cov_func0-centered]
15.81s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_conditional[cov_func1-noncentered]
15.71s call     tests/distributions/test_timeseries.py::TestPredefinedRandomWalk::test_mvgaussian_with_chol_cov_rv[chol]
14.71s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_prior[cov_func1-noncentered]
12.89s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_separable
12.70s call     tests/distributions/test_timeseries.py::TestEulerMaruyama::test_linear_model
11.63s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[VFE]
9.58s call     tests/distributions/test_timeseries.py::TestEulerMaruyama::test_batched_size[True-1]
9.22s call     tests/distributions/test_timeseries.py::TestGARCH11::test_support_point[size1-expected1]
8.06s call     tests/model/transform/test_conditioning.py::test_change_value_transforms
7.88s call     tests/distributions/test_timeseries.py::TestAR::test_batched_rhos
7.69s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[True-alpha_1]
7.62s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[True-beta_1]
7.41s call     tests/model/test_core.py::TestImputationMissingData::test_missing_dual_observations
7.10s call     tests/distributions/test_timeseries.py::TestAR::test_batched_size[True]
6.26s call     tests/distributions/test_timeseries.py::TestEulerMaruyama::test_batched_size[True-2]
5.87s call     tests/distributions/test_timeseries.py::TestGARCH11::test_logp
5.66s call     tests/model/test_core.py::TestValueGradFunction::test_missing_data
5.65s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_2_param_tensor
5.58s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_1_param
4.91s call     tests/distributions/test_timeseries.py::TestRandomWalk::test_init_logp_multivariate[init_dist0]
4.88s call     tests/model/test_core.py::TestModelDebug::test_invalid_observed_value
4.60s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[False-omega]
4.49s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_3_params
4.45s call     tests/model/test_core.py::test_model_logp_fast_compile
4.36s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[False-alpha_1]
4.33s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[False-beta_1]
4.17s call     tests/model/test_core.py::TestImputationMissingData::test_interval_missing_observations
4.04s call     tests/model/test_core.py::TestTransformArgs::test_default_transform_is_applied
3.93s call     tests/model/test_core.py::TestImputationMissingData::test_truncated_normal
3.88s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_1_param
3.80s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveLatent
3.68s call     tests/distributions/test_timeseries.py::TestGARCH11::test_support_point[None-expected0]
3.65s call     tests/distributions/test_timeseries.py::TestAR::test_batched_init_dist
3.62s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_2_param
3.51s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_2_param
3.47s call     tests/model/test_fgraph.py::test_fgraph_rewrite
3.40s call     tests/gp/test_cov.py::TestCovKron::test_multiops
3.30s call     tests/distributions/test_timeseries.py::TestAR::test_support_point[size1-expected1]
3.27s call     tests/gp/test_gp.py::TestMarginalVsLatent::testLatentMultioutput
3.19s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[DTC]
=========== 498 passed, 2 xfailed, 31 warnings in 621.25s (0:10:21) ============
============================= slowest 50 durations =============================
10.23s call     tests/distributions/test_simulator.py::TestSimulator::test_multiple_simulators
9.31s call     tests/distributions/test_simulator.py::TestSimulator::test_one_gaussian
8.02s call     tests/distributions/test_simulator.py::TestSimulator::test_nested_simulators
6.22s call     tests/sampling/test_forward.py::TestSamplePriorPredictive::test_multivariate2
5.85s call     tests/distributions/test_discrete.py::TestLogitCategorical::test_distribution
5.22s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_inferred_logp_mixture
5.10s call     tests/distributions/test_simulator.py::TestSimulator::test_simulator_metropolis_mcmc
5.06s call     tests/distributions/test_censored.py::TestCensored::test_censored_workflow[False]
5.02s call     tests/distributions/test_censored.py::TestCensored::test_censored_categorical
4.74s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point_scan
4.72s call     tests/distributions/test_censored.py::TestCensored::test_censored_workflow[True]
4.26s call     tests/sampling/test_forward.py::TestNestedRandom::test_ZeroInflatedNegativeBinomial[10-(4, 3)-(0.5, (3,))-(None, (3,))-(None, (3,))]
4.19s setup    tests/sampling/test_forward.py::TestSamplePPC::test_logging_sampled_basic_rvs_posterior_mutable[MultiTrace]
3.87s call     tests/sampling/test_forward.py::TestSamplePriorPredictive::test_zeroinflatedpoisson
3.78s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_hypergeometric
3.76s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_scan
3.38s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point_scan_recurring
3.17s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_beta_binomial
3.15s call     tests/distributions/test_simulator.py::TestSimulator::test_custom_dist_sum_stat[float32]
3.12s call     tests/distributions/test_discrete.py::TestBetaBinomial::test_distribution
2.93s call     tests/distributions/test_distribution.py::TestPartialObservedRV::test_multivariate_constant_mask_separable[True-True]
2.90s call     tests/sampling/test_forward.py::TestSamplePPC::test_observed_data_needed_in_pp
2.89s call     tests/distributions/test_discrete.py::TestDiscreteWeibull::test_distribution
2.83s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_negative_binomial
2.62s call     tests/sampling/test_forward.py::TestSamplePPC::test_deterministic_of_observed_modified_interface
2.47s call     tests/distributions/test_custom.py::TestCustomDist::test_serialize_custom_dist
2.41s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_geometric
2.27s call     tests/sampling/test_forward.py::TestSamplePPC::test_deterministic_of_observed
2.26s call     tests/sampling/test_forward.py::TestSamplePriorPredictive::test_transformed
2.23s call     tests/distributions/test_discrete.py::TestDiscreteUniform::test_distribution
2.20s call     tests/distributions/test_distribution.py::TestPartialObservedRV::test_multivariate_shared_mask_unseparable[False]
2.15s call     tests/distributions/test_simulator.py::TestSimulator::test_named_model
2.13s call     tests/distributions/test_discrete.py::TestCategorical::test_distribution
2.07s call     tests/sampling/test_forward.py::TestSamplePosteriorPredictive::test_forward_function_reused_across_set_data_batches
2.04s call     tests/sampling/test_forward.py::TestSamplePPC::test_normal_scalar
2.01s call     tests/distributions/test_simulator.py::TestSimulator::test_custom_dist_sum_stat[float64]
1.98s call     tests/sampling/test_forward.py::TestSamplePPC::test_model_shared_variable
1.98s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_bernoulli
1.92s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_discrete_unif
1.89s call     tests/distributions/test_distribution.py::TestPartialObservedRV::test_multivariate_shared_mask_unseparable[True]
1.88s call     tests/sampling/test_forward.py::TestSamplePPC::test_normal_scalar_idata
1.84s call     tests/distributions/test_distribution.py::TestPartialObservedRV::test_multivariate_shared_mask_separable
1.75s call     tests/sampling/test_forward.py::TestDraw::test_multivariate
1.74s call     tests/distributions/test_custom.py::TestCustomDist::test_custom_dist_custom_support_point_univariate[(2,)]
1.72s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_basic
1.72s setup    tests/sampling/test_forward.py::TestSamplePosteriorPredictive::test_point_list_arg_bug_spp
1.68s call     tests/sampling/test_forward.py::TestSamplePPC::test_normal_vector_idata
1.67s call     tests/sampling/test_forward.py::TestSamplePriorPredictive::test_respects_shape
1.63s call     tests/sampling/test_deterministic.py::test_docstring_example
1.61s call     tests/sampling/test_forward.py::TestSamplePosteriorPredictive::test_reused_forward_function_is_reproducible_across_seeds
=========== 405 passed, 1 xfailed, 258 warnings in 305.84s (0:05:05) ===========
============================= slowest 50 durations =============================
29.97s call     tests/test_pytensorf.py::test_rewrite_pregrad_inner_graphs
25.96s call     tests/distributions/test_mixture.py::TestMixture::test_list_mvnormals_predictive_sampling_shape
19.61s call     tests/progress_bar/test_manager.py::test_progressbar_nested_compound
19.49s call     tests/test_func_utils.py::test_find_constrained_prior[0.5-StudentT-0-1-init_guess3-fixed_params3-None]
13.62s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_nd[(3, 3)-5]
13.29s call     tests/distributions/test_mixture.py::TestMixture::test_single_univariate_component_deterministic_weights[None-component0-weights1]
12.81s call     tests/distributions/test_mixture.py::TestMixture::test_list_normals_sampling
12.35s call     tests/distributions/test_mixture.py::TestMixture::test_single_univariate_component_deterministic_weights[None-component3-weights0]
12.16s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_nd[3-5]
11.76s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_nd[1-5]
9.17s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_nd[()-5]
9.07s call     tests/distributions/test_mixture.py::TestMixture::test_single_poisson_sampling
8.66s call     tests/test_func_utils.py::test_find_constrained_prior[0.5-Gamma-0.1-0.4-init_guess0-fixed_params0-None]
8.39s call     tests/distributions/test_mixture.py::TestMixture::test_single_univariate_component_deterministic_weights[None-component0-weights0]
8.31s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_sampling
8.06s call     tests/distributions/test_mixture.py::TestMixture::test_single_poisson_predictive_sampling_shape
7.21s call     tests/distributions/test_mixture.py::TestMixtureSameFamily::test_with_multinomial[(3, 4)]
7.08s call     tests/distributions/test_mixture.py::TestMixtureVsLatent::test_vector_components
7.08s call     tests/distributions/test_mixture.py::TestMixture::test_list_poissons_sampling
6.62s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_nd[(3, 3)-3]
6.34s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zeroinflatednegativebinomial_logp
6.04s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_nd[3-3]
5.79s call     tests/distributions/test_mixture.py::TestMixtureVsLatent::test_scalar_components
5.73s call     tests/test_func_utils.py::test_find_constrained_prior[0.95-StudentT-0-1-init_guess3-fixed_params3-None]
5.61s call     tests/distributions/test_mixture.py::TestMixture::test_list_multivariate_components_deterministic_weights[None-components0-weights0]
5.43s call     tests/test_func_utils.py::test_find_constrained_prior[0.75-StudentT-0-1-init_guess3-fixed_params3-None]
5.25s call     tests/distributions/test_mixture.py::TestMixture::test_change_dist_size[True-comp_dists3]
5.17s call     tests/distributions/test_mixture.py::TestNormalMixture::test_random
5.08s call     tests/test_math.py::test_kronecker
5.01s call     tests/distributions/test_mixture.py::TestMixtureSameFamily::test_with_multinomial[(20,)]
4.65s call     tests/distributions/test_mixture.py::TestMixture::test_nested_mixture
4.56s call     tests/distributions/test_mixture.py::TestMixtureMoments::test_single_multivariate_component[weights0-comp_dists0-None-expected0]
4.53s call     tests/distributions/test_mixture.py::TestMixture::test_change_dist_size[True-comp_dists2]
4.38s call     tests/distributions/test_mixture.py::TestMixture::test_list_multivariate_components_deterministic_weights[size2-components0-weights0]
3.95s call     tests/distributions/test_mixture.py::TestMixture::test_list_multivariate_components_deterministic_weights[size2-components0-weights1]
3.94s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zeroinflatedbinomial_logp
3.86s call     tests/distributions/test_mixture.py::TestMixture::test_list_multivariate_components_deterministic_weights[None-components0-weights1]
3.85s call     tests/test_func_utils.py::test_find_constrained_prior_error_too_large[Gamma-0.1-0.4-init_guess0-fixed_params0]
3.82s call     tests/test_func_utils.py::test_find_constrained_prior[0.5-StudentT-0.1-0.4-init_guess2-fixed_params2-None]
3.69s call     tests/distributions/test_mixture.py::TestMixture::test_single_multivariate_component_deterministic_weights[size2-component0-weights0]
3.66s call     tests/distributions/test_mixture.py::TestMixture::test_change_dist_size[False-comp_dists0]
3.64s call     tests/step_methods/hmc/test_hmc.py::TestStepHamiltonianMC::test_step_continuous[<lambda>-1000_0]
3.60s call     tests/distributions/test_mixture.py::TestMixture::test_single_multivariate_component_deterministic_weights[None-component0-weights0]
3.58s call     tests/test_func_utils.py::test_find_constrained_prior[0.5-Normal-155-180-init_guess1-fixed_params1-None]
3.54s call     tests/distributions/test_mixture.py::TestMixture::test_change_dist_size[False-comp_dists3]
3.51s call     tests/test_func_utils.py::test_find_constrained_prior[0.75-Gamma-0.1-0.4-init_guess0-fixed_params0-None]
3.48s call     tests/distributions/test_mixture.py::TestMixtureSameFamily::test_with_mvnormal
3.40s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zero_inflated_poisson_support_point[0.2-mu3-size3-expected3]
3.39s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zeroinflatedpoisson_logp
3.34s call     tests/distributions/test_mixture.py::TestMixture::test_change_dist_size[True-comp_dists0]
=========== 631 passed, 8 skipped, 81 warnings in 633.34s (0:10:33) ============
============================= slowest 50 durations =============================
10.25s call     tests/sampling/test_forward.py::TestCompileForwardSampler::test_lkj_cholesky_cov
9.62s setup    tests/sampling/test_forward.py::TestSamplePPC::test_logging_sampled_basic_rvs_posterior_mutable[MultiTrace]
9.43s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_hypergeometric
8.68s call     tests/distributions/test_censored.py::TestCensored::test_censored_workflow[False]
7.14s call     tests/distributions/test_simulator.py::TestSimulator::test_multiple_simulators
6.29s call     tests/distributions/test_censored.py::TestCensored::test_censored_workflow[True]
6.04s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_negative_binomial
6.02s call     tests/distributions/test_distribution.py::TestPartialObservedRV::test_multivariate_constant_mask_separable[True-True]
5.88s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_inferred_logp_mixture
5.27s call     tests/distributions/test_censored.py::TestCensored::test_censored_categorical
4.89s call     tests/distributions/test_simulator.py::TestSimulator::test_custom_dist_sum_stat[float32]
4.78s call     tests/sampling/test_forward.py::TestSamplePriorPredictive::test_multivariate2
4.68s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_beta_binomial
4.60s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_geometric
4.38s call     tests/distributions/test_simulator.py::TestSimulator::test_nested_simulators
4.16s call     tests/distributions/test_simulator.py::TestSimulator::test_simulator_metropolis_mcmc
4.05s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_bernoulli
4.05s call     tests/distributions/test_simulator.py::TestSimulator::test_one_gaussian
3.77s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_orderedprobit[3]
3.43s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_discrete_unif
3.31s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_categorical[2]
3.29s call     tests/distributions/test_distribution.py::TestPartialObservedRV::test_multivariate_shared_mask_separable
3.28s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point_scan
3.24s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_binomial
3.21s call     tests/distributions/test_custom.py::TestCustomDist::test_serialize_custom_dist
3.13s call     tests/stats/test_log_density.py::TestComputeLogLikelihood::test_multivariate
3.13s call     tests/distributions/test_discrete.py::TestDiscreteWeibull::test_distribution
3.04s call     tests/distributions/test_discrete.py::TestLogitCategorical::test_distribution
2.93s call     tests/sampling/test_forward.py::TestSamplePPC::test_model_shared_variable
2.90s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_discrete_weibull
2.86s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_basic
2.72s call     tests/sampling/test_forward.py::TestSamplePriorPredictive::test_zeroinflatedpoisson
2.65s call     tests/distributions/test_discrete.py::test_negative_binomial_logp_stable_when_mu_overflows
2.49s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_poisson
2.44s call     tests/distributions/test_distribution.py::TestPartialObservedRV::test_multivariate_constant_mask_unseparable
2.40s call     tests/sampling/test_forward.py::TestSamplePPC::test_normal_scalar
2.39s call     tests/sampling/test_forward.py::TestSamplePosteriorPredictive::test_forward_function_reused_across_set_data_batches
2.25s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point_scan_recurring
2.21s call     tests/sampling/test_forward.py::TestSamplePosteriorPredictive::test_reused_forward_function_is_reproducible_across_seeds
2.18s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_scan
2.05s call     tests/distributions/test_discrete.py::test_negative_binomial_logp_large_n
2.04s call     tests/distributions/test_custom.py::TestCustomDist::test_custom_dist_multivariate_logp[None]
1.89s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_orderedprobit[2]
1.87s setup    tests/sampling/test_forward.py::TestSamplePosteriorPredictive::test_point_list_arg_bug_spp
1.87s call     tests/distributions/test_distribution.py::TestPartialObservedRV::test_multivariate_shared_mask_unseparable[False]
1.81s call     tests/distributions/test_distribution.py::TestDiracDelta::TestDiracDelta::test_distribution
1.72s call     tests/distributions/test_distribution.py::TestPartialObservedRV::test_multivariate_shared_mask_unseparable[True]
1.70s call     tests/sampling/test_population.py::TestPopulationSamplers::test_demcmc_warning_on_small_populations
1.70s call     tests/distributions/test_custom.py::TestCustomDist::test_custom_dist_without_random
1.69s call     tests/sampling/test_forward.py::TestSamplePPC::test_deterministic_of_observed_modified_interface
=========== 404 passed, 2 xfailed, 173 warnings in 279.17s (0:04:39) ===========
============================ slowest 50 durations =============================
52.97s setup    tests/step_methods/test_metropolis.py::TestMetropolisUniform::test_kstest
36.05s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSLKJCholeskyCov::test_kstest
28.31s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist3]
25.87s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist4]
18.38s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSUniform::test_kstest
17.99s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSUniform2::test_kstest
17.94s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSUniform3::test_kstest
12.11s setup    tests/step_methods/test_slicer.py::TestSliceUniform::test_kstest
9.53s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSStudentT::test_kstest
9.41s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSNormal::test_kstest
8.45s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSBetaBinomial::test_kstest
8.03s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist1]
7.84s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist2]
5.52s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist0]
5.34s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_categorical[uniform]
5.29s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update_different_scales
5.11s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_emits_energy_warnings
4.78s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_continuous[<function TestStepMetropolis.<lambda> at 0x00000208BD3B0400>-4000]
4.77s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_multiple_samplers
4.19s call     tests/step_methods/hmc/test_nuts.py::TestStepNUTS::test_step_continuous[<lambda>-1000_0]
4.05s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_epsilon_parallel
1.98s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_tuning_reset
1.90s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_discrete
1.63s call     tests/step_methods/test_slicer.py::TestStepSlicer::test_step_continuous[<function TestStepSlicer.<lambda> at 0x00000208BD3B16C0>-2000]
1.60s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_categorical[proportional]
1.51s call     tests/step_methods/test_compound.py::TestStepCompound::test_step_continuous[<function TestStepCompound.<lambda> at 0x00000208BD3B3740>-1000]
1.27s call     tests/step_methods/test_compound.py::TestCompoundStep::test_non_blocked
1.18s call     tests/step_methods/hmc/test_nuts.py::TestStepNUTS::test_step_continuous[<lambda>-1000_1]
1.17s call     tests/step_methods/test_compound.py::TestCompoundStep::test_blocked
1.16s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_bad_init_nonparallel
1.09s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_sampler_stats
1.03s call     tests/step_methods/test_slicer.py::TestStepSlicer::test_step_continuous[<function TestStepSlicer.<lambda> at 0x00000208BD3B1800>-2000]
0.89s call     tests/step_methods/hmc/test_hmc.py::TestStepHamiltonianMC::test_step_continuous[<lambda>-1000_0]
0.86s call     tests/step_methods/hmc/test_hmc.py::TestStepHamiltonianMC::test_step_continuous[<lambda>-1000_1]
0.80s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_lambda_sequential
0.80s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_reset
0.76s call     tests/step_methods/hmc/test_hmc.py::test_leapfrog_reversible
0.72s call     tests/step_methods/test_metropolis.py::TestDEMetropolis::test_demcmc_tune_parameter
0.68s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_bad_init_parallel
0.59s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_none
0.41s call     tests/step_methods/test_metropolis.py::TestRVsAssignmentMetropolis::test_discrete_steps[BinaryMetropolis]
0.35s call     tests/step_methods/test_compound.py::TestStatsBijection::test_flatten_steps
0.30s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_custom_proposal_dist
0.27s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tune_drop_fraction
0.25s call     tests/step_methods/test_compound.py::TestCompoundStep::test_name
0.22s call     tests/step_methods/hmc/test_hmc.py::test_nuts_tuning
0.20s call     tests/step_methods/test_compound.py::TestRVsAssignmentCompound::test_compound_step
0.18s call     tests/step_methods/test_compound.py::TestStatsMetadata::test_stats_from_steps
0.18s call     tests/step_methods/hmc/test_nuts.py::test_nuts_step_legacy_value_grad_function
0.17s call     tests/step_methods/hmc/test_hmc.py::TestRVsAssignmentHamiltonianMC::test_continuous_steps[HamiltonianMC-step_kwargs0]
=========== 118 passed, 6 skipped, 35 warnings in 352.39s (0:05:52) ===========

You can find more information on how to contribute here

Automatically generated by GitHub Action Latest run date: 2026-08-13 Run logs: 31695199392

Contributor guide