Skip to content

Speed up test times 🚀 #7686

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
williambdean opened this issue Feb 20, 2025 · 2 comments
Open

Speed up test times 🚀 #7686

williambdean opened this issue Feb 20, 2025 · 2 comments

Comments

@williambdean
Copy link
Contributor

williambdean commented Feb 20, 2025 •
edited
Loading

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

Here are some of the slowest test times:

(6 minutes 33 seconds) macos (macos-latest, float64, 3.13, tests/sampling/test_parallel.py tests/test_data.py tests/vari... 
============================= slowest 50 durations =============================
22.58s call     tests/variational/test_minibatch_rv.py::TestMinibatchRandomVariable::test_multidim_scaling
21.50s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_separable
18.15s call     tests/sampling/test_parallel.py::test_remote_pipe_closed
17.09s call     tests/sampling/test_parallel.py::test_context
15.36s call     tests/model/test_core.py::TestValueGradFunction::test_missing_data
12.41s call     tests/variational/test_minibatch_rv.py::TestMinibatchRandomVariable::test_derived_rv
11.99s call     tests/model/test_core.py::TestImputationMissingData::test_truncated_normal
11.21s call     tests/model/test_core.py::TestImputationMissingData::test_interval_missing_observations
10.93s call     tests/test_data.py::TestData::test_shared_data_as_rv_input
10.20s call     tests/model/test_core.py::test_model_d2logp[True]
9.85s call     tests/sampling/test_parallel.py::test_spawn_densitydist_bound_method
9.72s call     tests/variational/test_minibatch_rv.py::TestMinibatchRandomVariable::test_density_scaling
8.78s call     tests/test_data.py::TestData::test_sample
8.37s call     tests/model/test_core.py::TestValueGradFunction::test_edge_case
8.00s call     tests/model/test_core.py::TestImputationMissingData::test_missing_logp1
7.29s call     tests/sampling/test_parallel.py::test_spawn_densitydist_function
6.97s call     tests/sampling/test_parallel.py::test_explicit_sample[spawn]
6.38s call     tests/model/test_core.py::TestImputationMissingData::test_missing_dual_observations
5.35s call     tests/model/test_core.py::TestTransformArgs::test_default_transform_is_applied
5.27s call     tests/model/test_core.py::TestModelDebug::test_invalid_observed_value
4.94s call     tests/test_data.py::TestData::test_shared_data_as_index
4.14s call     tests/sampling/test_parallel.py::test_bad_unpickle
4.06s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_unseparable
3.89s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter[logp]
3.71s call     tests/model/test_core.py::TestImputationMissingData::test_missing_logp2
3.71s call     tests/model/test_core.py::TestImputationMissingData::test_missing_basic[pandas]
3.58s call     tests/model/test_core.py::test_model_logp_fast_compile
3.55s call     tests/model/test_core.py::TestImputationMissingData::test_coordinates
3.11s call     tests/test_data.py::test_scaling_data_works_in_likelihood
3.10s call     tests/model/test_core.py::test_tempered_logp_dlogp[False]
3.06s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter[dlogp]
2.65s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter_cant_be_evaluated[logp-True]
2.62s call     tests/model/test_core.py::TestPytensorRelatedLogpBugs::test_pytensor_switch_broadcast_edge_cases_2
2.49s call     tests/model/test_core.py::TestProfile::test_profile_model
2.26s call     tests/test_data.py::TestData::test_sample_posterior_predictive_after_set_data_with_coords
2.26s call     tests/model/test_core.py::TestObserved::test_observed_compute_test_value
2.22s call     tests/model/test_core.py::TestModelDebug::test_invalid_parameter_cant_be_evaluated[dlogp-True]
1.90s call     tests/test_data.py::TestData::test_sample_after_set_data
1.89s call     tests/test_data.py::TestMinibatch::test_multiple_vars
1.85s call     tests/model/test_core.py::test_model_d2logp[False]
1.85s call     tests/test_data.py::TestData::test_deterministic
1.84s call     tests/model/test_core.py::TestImputationMissingData::test_missing_vector_parameter
1.71s call     tests/test_data.py::TestMinibatch::test_assert
1.56s call     tests/model/test_core.py::test_model_dlogp[True]
1.51s call     tests/test_data.py::TestData::test_set_data_to_non_data_container_variables
1.50s call     tests/model/test_core.py::TestCheckStartVals::test_valid_start_point
1.44s call     tests/test_data.py::TestData::test_model_to_graphviz_for_model_with_data_container
1.43s call     tests/model/test_core.py::test_datalogp_multiple_shapes
1.42s call     tests/model/test_core.py::TestImputationMissingData::test_symbolic_random_variable
1.36s call     tests/variational/test_minibatch_rv.py::TestMinibatchRandomVariable::test_minibatch_parameter_and_value
====== 167 passed, 2 skipped, 1 xfailed, 5 warnings in 330.96s (0:05:30) =======
(5 minutes 25 seconds) macos (macos-latest, float64, 3.13, tests/sampling/test_mcmc.py ) 
============================= slowest 50 durations =============================
44.44s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
14.55s setup    tests/sampling/test_mcmc.py::TestInitNuts::test_checks_seeds_kwarg
10.78s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
9.63s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-map]
9.17s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-jitter+adapt_diag]
8.80s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[False]
8.42s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[map]
8.26s call     tests/sampling/test_mcmc.py::test_step_args
7.92s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
7.79s call     tests/sampling/test_mcmc.py::test_sample_var_names
7.12s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[True]
6.91s call     tests/sampling/test_mcmc.py::TestType::test_float32
5.88s call     tests/sampling/test_mcmc.py::TestSample::test_deterministic_of_unobserved[True]
4.78s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi_map]
4.50s call     tests/sampling/test_mcmc.py::test_sample_find_MAP_does_not_modify_start
4.45s call     tests/sampling/test_mcmc.py::TestNamedSampling::test_shared_named
3.85s call     tests/sampling/test_mcmc.py::TestType::test_float64
3.29s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-2-advi]
3.24s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-1-advi]
3.00s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[map]
2.99s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-1-map]
2.94s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-2-advi]
2.87s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-2-advi]
2.86s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-1-advi]
2.83s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-advi]
2.83s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-1-advi]
2.80s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-1-advi]
2.70s call     tests/sampling/test_mcmc.py::TestShared::test_sample
2.53s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-advi]
2.46s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[ADVI+adapt_diag]
2.40s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-1-map]
2.26s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-2-map]
2.23s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-advi]
2.13s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-advi]
2.13s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-2-map]
2.09s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-1-1-map]
2.07s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-map]
1.96s call     tests/sampling/test_mcmc.py::TestSample::test_default_sample_does_not_set_global_seed
1.93s call     tests/sampling/test_mcmc.py::TestSample::test_deterministic_of_unobserved[False]
1.92s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-2-map]
1.91s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-map]
1.90s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-seeds2-1-map]
1.69s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-map]
1.66s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi_map]
1.59s call     tests/sampling/test_mcmc.py::test_blas_cores
1.47s call     tests/sampling/test_mcmc.py::TestSample::test_sample_does_not_rely_on_external_global_seeding
1.41s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-map]
0.83s call     tests/sampling/test_mcmc.py::TestAssignStepMethods::test_categorical
0.74s call     tests/sampling/test_mcmc.py::TestNamedSampling::test_constant_named
0.72s call     tests/sampling/test_mcmc.py::TestSampleReturn::test_sample_return_lengths
============ 98 passed, 5 skipped, 25 warnings in 262.43s (0:04:22) ============
(13 minutes 6 seconds) macos (macos-latest, float64, 3.13, tests/backends/test_arviz.py tests/backends/test_zarr.py test... 
============================= slowest 50 durations =============================
71.77s call     tests/backends/test_arviz.py::TestDataPyMC::test_mv_missing_data_model
49.97s setup    tests/backends/test_arviz.py::TestDataPyMC::test_to_idata
32.34s call     tests/backends/test_arviz.py::TestDataPyMC::test_multivariate_observations
23.90s call     tests/backends/test_arviz.py::TestDataPyMC::test_single_observation
16.39s setup    tests/backends/test_zarr.py::test_record[single_step-include_transformed-frequent_writes]
16.09s call     tests/backends/test_arviz.py::TestDataPyMC::test_autodetect_coords_from_model[True]
12.84s call     tests/backends/test_arviz.py::TestDataPyMC::test_missing_data_model
9.18s call     tests/backends/test_arviz.py::TestDataPyMC::test_overwrite_model_coords_dims
4.76s call     tests/variational/test_updates.py::test_updates_fast[scalar-without_args-all_params-adagrad_window]
4.61s call     tests/backends/test_arviz.py::TestDataPyMC::test_constant_data[True]
4.30s call     tests/backends/test_arviz.py::TestDataPyMC::test_to_idata
3.85s setup    tests/backends/test_zarr.py::test_record[compound_step-include_transformed-frequent_writes]
3.69s call     tests/backends/test_zarr.py::test_record[single_step-include_transformed-frequent_writes]
2.88s call     tests/variational/test_updates.py::test_updates_fast[scalar-without_args-all_params-adam]
2.82s call     tests/backends/test_arviz.py::TestDataPyMC::test_potential
2.43s call     tests/backends/test_arviz.py::TestDataPyMC::test_posterior_predictive_thinned
2.38s call     tests/variational/test_updates.py::test_updates_fast[scalar-without_args-all_params-rmsprop]
2.21s call     tests/variational/test_updates.py::test_updates_fast[mixed-without_args-all_params-adadelta]
2.19s call     tests/backends/test_arviz.py::TestDataPyMC::test_predictions_constant_data
2.09s call     tests/backends/test_arviz.py::TestDataPyMC::test_single_chain[1]
2.07s call     tests/variational/test_updates.py::test_updates_fast[matrix-without_args-all_params-adadelta]
1.77s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-discard_tuning-return_zarr-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
1.56s call     tests/backends/test_arviz.py::TestDataPyMC::test_multiple_observed_rv[False]
1.55s 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]
1.50s call     tests/backends/test_arviz.py::TestDataPyMC::test_include_transformed
1.46s setup    tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-discard_tuning-return_idata-keep_warning_stat-parallel_sampling-no_loglike-frequent_writes]
1.43s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_zarr-discard_warning_stat-parallel_sampling-no_loglike-sparse_writes]
1.41s call     tests/backends/test_zarr.py::test_sampling_consistency[compound_step-frequent_writes]
1.41s call     tests/backends/test_arviz.py::TestDataPyMC::test_multiple_observed_rv[True]
1.40s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-keep_tuning-return_idata-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
1.38s call     tests/backends/test_arviz.py::TestDataPyMC::test_autodetect_coords_from_model[False]
1.34s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_zarr-keep_warning_stat-parallel_sampling-no_loglike-frequent_writes]
1.32s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_zarr-keep_warning_stat-parallel_sampling-no_loglike-sparse_writes]
1.32s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_zarr-keep_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
1.31s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_zarr-keep_warning_stat-parallel_sampling-no_loglike-sparse_writes]
1.30s setup    tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-keep_tuning-return_idata-keep_warning_stat-parallel_sampling-no_loglike-sparse_writes]
1.30s call     tests/backends/test_arviz.py::TestPyMCWarmupHandling::test_save_warmup[0-50-1-False]
1.30s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-keep_tuning-return_idata-discard_warning_stat-parallel_sampling-no_loglike-sparse_writes]
1.30s setup    tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-keep_tuning-return_idata-discard_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
1.30s call     tests/backends/test_zarr.py::test_sampling_consistency[single_step-frequent_writes]
1.29s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_idata-keep_warning_stat-parallel_sampling-compute_loglike-frequent_writes]
1.28s call     tests/backends/test_zarr.py::test_record[compound_step-include_transformed-frequent_writes]
1.26s 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.26s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-keep_tuning-return_zarr-discard_warning_stat-sequential_sampling-no_loglike-frequent_writes]
1.25s call     tests/backends/test_zarr.py::test_sample[single_step-include_transformed-discard_tuning-return_zarr-keep_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
1.25s 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]
1.22s 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.22s call     tests/backends/test_zarr.py::test_sample[compound_step-include_transformed-keep_tuning-return_idata-discard_warning_stat-sequential_sampling-compute_loglike-sparse_writes]
1.21s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_idata-discard_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
1.21s setup    tests/backends/test_zarr.py::test_sample[compound_step-discard_transformed-discard_tuning-return_zarr-discard_warning_stat-sequential_sampling-compute_loglike-frequent_writes]
=========== 480 passed, 64 skipped, 70 warnings in 695.47s (0:11:35) ===========
(5 minutes 53 seconds) alternative_backends (ubuntu-20.04, float64, 3.12, tests/distributions/test_random_alternative_ba... 
============================= slowest 50 durations =============================
22.18s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-None-sampler0]
16.51s call     tests/sampling/test_jax.py::test_get_log_likelihood
13.39s call     tests/distributions/test_random_alternative_backends.py::test_dirichlet_multinomial[NUMBA]
12.14s call     tests/distributions/test_random_alternative_backends.py::test_dirichlet_multinomial[FAST_RUN]
11.51s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[nutpie]
11.46s call     tests/distributions/test_random_alternative_backends.py::test_dirichlet_multinomial_dims[NUMBA]
9.06s call     tests/sampling/test_jax.py::test_get_batched_jittered_initial_points
7.45s call     tests/distributions/test_random_alternative_backends.py::test_mvstudentt[NUMBA]
7.14s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-None-sampler0]
7.06s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-cpu-sampler0]
6.95s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-None-sampler0]
6.77s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-cpu-sampler0]
6.64s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-cpu-sampler0]
6.61s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-cpu-sampler0]
6.58s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-None-sampler0]
6.34s call     tests/sampling/test_jax.py::test_sample_partially_observed
6.10s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[pymc]
6.01s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[blackjax]
5.65s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-None-sampler1]
5.39s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-cpu-sampler1]
5.24s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-cpu-sampler1]
5.24s call     tests/distributions/test_random_alternative_backends.py::test_mvstudentt[FAST_RUN]
5.17s call     tests/sampling/test_jax.py::test_transform_samples[scan-2-None-sampler1]
4.84s call     tests/sampling/test_jax.py::test_transform_samples[vmap-2-None-sampler1]
4.73s call     tests/sampling/test_jax.py::test_initvals_without_jitter[sampler0]
4.65s call     tests/sampling/test_jax.py::test_seeding[2-123-sampler0]
4.15s call     tests/sampling/test_jax.py::test_transform_samples[scan-1-cpu-sampler1]
4.12s call     tests/sampling/test_mcmc_external.py::test_external_nuts_sampler[numpyro]
3.96s call     tests/sampling/test_jax.py::test_idata_kwargs[None-idata_kwargs0-sampler0]
3.84s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-None-sampler1]
3.81s call     tests/sampling/test_jax.py::test_transform_samples[vmap-1-cpu-sampler1]
3.78s call     tests/sampling/test_jax.py::test_seeding[2-None-sampler0]
3.73s call     tests/sampling/test_jax.py::test_seeding[1-None-sampler0]
3.67s call     tests/sampling/test_jax.py::test_seeding[1-123-sampler0]
3.30s call     tests/sampling/test_jax.py::test_idata_kwargs[None-idata_kwargs1-sampler0]
3.12s call     tests/sampling/test_jax.py::test_deterministic_samples[sampler0]
2.81s call     tests/sampling/test_jax.py::test_initvals_without_jitter[sampler1]
2.54s call     tests/sampling/test_jax.py::test_seeding[2-None-sampler1]
2.53s call     tests/sampling/test_jax.py::test_idata_kwargs[cpu-idata_kwargs1-sampler0]
2.52s call     tests/sampling/test_jax.py::test_seeding[2-123-sampler1]
2.41s call     tests/sampling/test_jax.py::test_idata_kwargs[cpu-idata_kwargs3-sampler0]
2.41s call     tests/sampling/test_jax.py::test_idata_kwargs[cpu-idata_kwargs4-sampler0]
2.41s call     tests/sampling/test_jax.py::test_idata_kwargs[cpu-idata_kwargs0-sampler0]
2.39s call     tests/sampling/test_jax.py::test_idata_kwargs[cpu-idata_kwargs2-sampler0]
2.38s call     tests/sampling/test_jax.py::test_idata_kwargs[None-idata_kwargs4-sampler0]
2.38s call     tests/sampling/test_jax.py::test_idata_kwargs[None-idata_kwargs3-sampler0]
2.36s call     tests/sampling/test_jax.py::test_idata_kwargs[None-idata_kwargs2-sampler0]
2.19s call     tests/sampling/test_jax.py::test_convergence_warnings[blackjax]
2.15s call     tests/sampling/test_jax.py::test_idata_kwargs[cpu-idata_kwargs2-sampler1]
2.13s call     tests/sampling/test_jax.py::test_idata_contains_stats[sample_blackjax_nuts]
================= 79 passed, 24 warnings in 317.73s (0:05:17) ==================
(24 minutes 12 seconds) float32 (windows-latest, float32, 3.13, tests/sampling/test_mcmc.py tests/ode/test_ode.py tests/o... 
============================ slowest 50 durations =============================
68.14s call     tests/sampling/test_mcmc.py::test_step_args
46.10s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
35.84s call     tests/sampling/test_mcmc.py::test_sample_var_names
34.06s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a1-None]
32.64s call     tests/sampling/test_mcmc.py::test_blas_cores
30.06s call     tests/sampling/test_mcmc.py::test_sample_find_MAP_does_not_modify_start
24.81s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-advi]
24.80s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-advi]
24.71s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-advi]
23.58s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-map]
23.25s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-map]
23.15s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-map]
22.81s call     tests/sampling/test_mcmc.py::TestSample::test_sample_does_not_rely_on_external_global_seeding
22.03s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[False]
20.76s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-jitter+adapt_diag]
20.76s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-jitter+adapt_diag]
20.59s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-jitter+adapt_diag]
18.20s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a0-None]
17.66s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[True]
16.98s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_2_param_tensor
16.62s call     tests/distributions/test_transform.py::test_interval
16.46s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_triangular[-10-0-200-size1]
16.41s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_1_param
16.27s call     tests/distributions/test_transform.py::test_simplex_bounds
14.96s call     tests/ode/test_ode.py::test_simulate
14.80s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
14.74s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[map]
14.02s call     tests/distributions/test_transform.py::test_ordered
14.01s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_half_normal_ordered[sigma1-size1]
13.87s setup    tests/sampling/test_mcmc.py::TestInitNuts::test_checks_seeds_kwarg
12.87s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
12.77s call     tests/sampling/test_mcmc.py::TestSample::test_default_sample_does_not_set_global_seed
12.01s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a2-size2]
11.43s call     tests/distributions/test_transform.py::test_simplex
11.04s call     tests/sampling/test_mcmc.py::test_no_init_nuts_compound
11.00s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_other[0.5-5.5-size1-transform1]
10.87s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_beta[a2-b2-size2]
10.72s call     tests/ode/test_ode.py::test_logp_scalar_ode
10.46s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform[lower2-upper2-size2]
10.34s call     tests/sampling/test_mcmc.py::test_no_init_nuts_step
10.34s call     tests/sampling/test_mcmc.py::TestSample::test_parallel_start
10.27s call     tests/sampling/test_mcmc.py::test_init_nuts
10.22s call     tests/sampling/test_mcmc.py::TestSampleReturn::test_logs_sampler_warnings[2]
10.13s call     tests/sampling/test_mcmc.py::TestSample::test_sample_callback
9.94s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_other[lower2-upper2-size2-transform2]
9.92s call     tests/ode/test_ode.py::TestDiffEqModel::test_scalar_ode_1_param
9.74s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-map]
9.73s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_2_param
9.70s call     tests/sampling/test_mcmc.py::TestType::test_float64
9.60s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_beta[0.5-0.5-size1]
===== 179 passed, 7 skipped, 4 xfailed, 40 warnings in 1299.82s (0:21:39) =====
(18 minutes 24 seconds) ubuntu (ubuntu-20.04, float64, 3.13, tests/test_util.py tests/test_pytensorf.py tests/test_math.p... 
============================= slowest 50 durations =============================
49.93s call     tests/distributions/test_mixture.py::TestMixtureSameFamily::test_with_multinomial[(3, 4)]
41.88s call     tests/distributions/test_mixture.py::TestMixtureSameFamily::test_with_multinomial[(20,)]
31.29s call     tests/distributions/test_mixture.py::TestMixture::test_single_poisson_sampling
25.41s call     tests/test_func_utils.py::test_find_constrained_prior[0.5-Gamma-0.1-0.4-init_guess0-fixed_params0-None]
25.01s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zero_inflated_negative_binomial_support_point[psi3-mu3-alpha3-size3-expected3]
24.09s call     tests/distributions/test_mixture.py::TestMixture::test_single_multivariate_component_deterministic_weights[size2-component0-weights0]
22.27s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_nd[(3, 3)-5]
21.84s call     tests/test_func_utils.py::test_find_constrained_prior[0.5-StudentT-0-1-init_guess3-fixed_params3-None]
21.00s call     tests/distributions/test_mixture.py::TestMixture::test_list_mvnormals_predictive_sampling_shape
19.46s call     tests/distributions/test_mixture.py::TestMixture::test_single_univariate_component_deterministic_weights[None-component0-weights1]
19.44s call     tests/distributions/test_mixture.py::TestMixture::test_single_multivariate_component_deterministic_weights[None-component0-weights1]
19.38s call     tests/distributions/test_mixture.py::TestHurdleMixtures::test_hurdle_zero_draws[dist-0.1-non_psi_args0]
18.33s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_nd[3-5]
16.83s call     tests/distributions/test_mixture.py::TestMixture::test_single_multivariate_component_deterministic_weights[None-component0-weights0]
16.72s call     tests/distributions/test_mixture.py::TestMixtureVsLatent::test_scalar_components
16.29s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_sampling
16.16s call     tests/distributions/test_mixture.py::TestMixture::test_list_normals_sampling
15.41s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zero_inflated_poisson_support_point[0.2-mu3-size3-expected3]
14.65s call     tests/distributions/test_mixture.py::TestMixture::test_list_mvnormals_logp
13.51s call     tests/distributions/test_mixture.py::TestMixtureVsLatent::test_vector_components
13.46s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_nd[1-5]
13.04s call     tests/test_func_utils.py::test_find_constrained_prior[0.5-Normal-155-180-init_guess1-fixed_params1-None]
12.74s call     tests/test_pytensorf.py::TestBroadcasting::test_metropolis_sampling
12.64s call     tests/distributions/test_mixture.py::TestMixture::test_single_univariate_component_deterministic_weights[None-component3-weights0]
12.50s call     tests/distributions/test_mixture.py::TestMixture::test_list_poissons_sampling
11.93s call     tests/test_func_utils.py::test_find_constrained_prior[0.5-Binomial-0-8-init_guess6-fixed_params6-None]
11.91s call     tests/test_func_utils.py::test_find_constrained_prior[0.5-StudentT-0.1-0.4-init_guess2-fixed_params2-None]
11.32s call     tests/backends/test_base.py::TestInitTrace::test_init_trace_continuation_unsupported
11.23s call     tests/distributions/test_shape_utils.py::TestSizeShapeDimsObserved::test_observed_with_column_vector
11.16s call     tests/distributions/test_mixture.py::TestMixture::test_single_univariate_component_deterministic_weights[None-component0-weights0]
11.15s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zeroinflatednegativebinomial_logp
10.52s call     tests/test_pytensorf.py::TestCompilePyMC::test_scan_updates
10.50s call     tests/distributions/test_mixture.py::TestMixture::test_nested_mixture
10.36s call     tests/step_methods/hmc/test_hmc.py::TestStepHamiltonianMC::test_step_continuous[<lambda>-1000_0]
9.96s call     tests/test_func_utils.py::test_find_constrained_prior_error_too_large[Gamma-0.1-0.4-init_guess0-fixed_params0]
9.05s call     tests/test_func_utils.py::test_find_constrained_prior[0.75-StudentT-0-1-init_guess3-fixed_params3-None]
8.88s call     tests/test_func_utils.py::test_find_constrained_prior[0.95-StudentT-0-1-init_guess3-fixed_params3-None]
8.87s call     tests/distributions/test_mixture.py::TestNormalMixture::test_normal_mixture_nd[()-5]
8.46s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zero_inflated_poisson_support_point[0.9-3.0-None-3]
8.05s call     tests/distributions/test_mixture.py::TestHurdleMixtures::test_hurdle_logp_at_zero[dist-0.25-non_psi_args5]
7.96s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zero_inflated_binomial_support_point[0.4-25-p3-size3-expected3]
7.95s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zero_inflated_poisson_support_point[0.8-2.9-5-expected1]
7.92s call     tests/distributions/test_mixture.py::TestMixture::test_list_multivariate_components_deterministic_weights[size2-components0-weights0]
7.03s call     tests/step_methods/hmc/test_hmc.py::test_leapfrog_reversible
6.93s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zero_inflated_negative_binomial_support_point[0.2-10-4-5-expected1]
6.64s call     tests/test_func_utils.py::test_find_constrained_prior[0.5-Poisson-1-15-init_guess7-fixed_params7-None]
6.13s call     tests/distributions/test_mixture.py::TestMixture::test_list_multivariate_components_deterministic_weights[None-components0-weights1]
6.13s call     tests/distributions/test_mixture.py::TestMixtureSameFamily::test_broadcasting_in_shape
6.10s call     tests/distributions/test_mixture.py::TestZeroInflatedMixture::test_zero_inflated_negative_binomial_support_point[0.4-mu2-alpha2-None-expected2]
6.09s call     tests/distributions/test_shape_utils.py::test_implicit_size_from_params
================ 605 passed, 55 warnings in 1076.02s (0:17:56) =================
(31 minutes 52 seconds) ubuntu (ubuntu-20.04, float64, 3.13, tests/distributions/test_continuous.py tests/distributions/t... 
============================= slowest 50 durations =============================
67.51s call     tests/distributions/test_multivariate.py::test_mvnormal_no_cholesky_in_model_logp
38.39s call     tests/distributions/test_multivariate.py::TestLKJCorr::test_distribution
27.16s call     tests/distributions/test_multivariate.py::TestMoments::test_stickbreakingweights_support_point[0.5-5-size3-expected3]
26.53s call     tests/distributions/test_continuous.py::TestMatchesScipy::test_truncated_normal
25.46s call     tests/distributions/test_continuous.py::TestMatchesScipy::test_triangular
24.30s call     tests/distributions/test_continuous.py::TestMoments::test_truncatednormal_support_point[mu2-1-None-10-size2-expected2]
24.25s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_mvnormal_indef
23.46s call     tests/distributions/test_multivariate.py::TestMoments::test_dirichlet_multinomial_support_point[a5-n5-size5-expected5]
21.67s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_multinomial_vectorized[extra_size2-p0-n2]
20.38s call     tests/distributions/test_multivariate.py::TestLKJCholeskCov::test_dist
20.27s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_multinomial_vectorized[extra_size2-p2-10]
20.04s call     tests/distributions/test_continuous.py::TestMatchesScipy::test_interpolated
19.91s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_dirichlet_vectorized[extra_size2-a2]
19.22s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_dirichlet_vectorized[extra_size1-a2]
18.90s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_dirichlet_vectorized[extra_size0-a2]
18.20s call     tests/distributions/test_continuous.py::TestMatchesScipy::test_uniform
17.34s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_multinomial_vectorized[extra_size2-p1-n2]
17.15s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_multinomial_vectorized[extra_size0-p1-n2]
17.11s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_multinomial_vectorized[extra_size0-p2-10]
16.31s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_dirichlet_vectorized[extra_size2-a1]
16.05s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_multinomial_vectorized[extra_size0-p0-n2]
15.68s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_multinomial_vectorized[extra_size0-p1-10]
15.64s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_kroneckernormal[0-3-2]
15.08s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_stickbreakingweights_vectorized[alpha1-5]
14.75s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_multinomial_vectorized[extra_size0-p0-n1]
14.50s call     tests/distributions/test_multivariate.py::TestZeroSumNormal::test_zsn_dims[dims0-None]
14.31s call     tests/distributions/test_continuous.py::TestMoments::test_truncatednormal_support_point[1-sigma1--10-inf-None-expected1]
14.01s call     tests/distributions/test_continuous.py::TestMoments::test_skewstudentt_support_point[10-10-mu3-sigma3-size3-expected3]
13.65s call     tests/distributions/test_continuous.py::TestMoments::test_beta_support_point[1-beta3-size3-expected3]
13.61s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_dirichlet_vectorized[extra_size1-a1]
13.44s call     tests/distributions/test_multivariate.py::TestMoments::test_zerosum_normal_support_point[shape1-2-expected1]
13.42s call     tests/distributions/test_continuous.py::TestMoments::test_exgaussian_support_point[1-1-sigma1-None-expected1]
12.86s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_multinomial_vectorized[extra_size1-p0-n2]
12.63s call     tests/distributions/test_continuous.py::TestMoments::test_interpolated_support_point[x_points3-pdf_points3-size3-expected3]
12.54s call     tests/distributions/test_continuous.py::TestMoments::test_halfstudentt_support_point[1-sigma2-size2-expected2]
12.43s call     tests/distributions/test_continuous.py::TestMoments::test_uniform_support_point[0-upper3-size3-expected3]
12.17s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_dirichlet_vectorized[extra_size0-a1]
12.07s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_mvnormal[1]
11.94s call     tests/distributions/test_multivariate.py::TestZeroSumNormal::test_zsn_dims[dims2-2]
11.57s call     tests/distributions/test_continuous.py::TestMoments::test_interpolated_support_point[x_points2-pdf_points2-size2-expected2]
11.47s call     tests/distributions/test_multivariate.py::TestMoments::test_zerosum_normal_support_point[shape2-3-expected2]
11.31s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_dirichlet_vectorized[extra_size1-a0]
11.15s call     tests/distributions/test_continuous.py::TestMatchesScipy::test_get_tau_sigma
10.83s call     tests/distributions/test_continuous.py::TestMoments::test_uniform_support_point[0-upper2-None-expected2]
10.77s call     tests/distributions/test_multivariate.py::TestMvStudentTCov::test_distribution
10.57s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_multinomial_vectorized[extra_size1-p0-n1]
10.53s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_dirichlet_vectorized[extra_size2-a0]
10.43s call     tests/distributions/test_multivariate.py::TestZeroSumNormal::test_zsn_logp[5-shape3-2-mvn_axes3]
10.19s call     tests/distributions/test_multivariate.py::TestMatchesScipy::test_multinomial_invalid_value
10.14s call     tests/distributions/test_continuous.py::TestMoments::test_gamma_support_point[alpha3-beta3-size3-expected3]
=========== 647 passed, 3 xfailed, 43 warnings in 1884.23s (0:31:24) ===========
(12 minutes 36 seconds) windows (windows-latest, float64, 3.10, tests/variational/test_approximations.py tests/variationa... 
============================ slowest 50 durations =============================
62.59s call     tests/variational/test_inference.py::test_replacements[FullRankADVI]
57.31s call     tests/variational/test_inference.py::test_replacements[ASVGD]
51.26s call     tests/variational/test_inference.py::test_fit_oo[SVGD-full]
41.36s call     tests/variational/test_inference.py::test_replacements[ADVI]
33.97s call     tests/variational/test_inference.py::test_replacements[SVGD]
33.43s call     tests/variational/test_inference.py::test_fit_oo[ASVGD-full]
31.14s call     tests/variational/test_approximations.py::test_seeding_advi_fit
24.38s call     tests/variational/test_inference.py::test_fit_data_coords
23.45s call     tests/variational/test_approximations.py::test_elbo
20.35s call     tests/variational/test_inference.py::test_fit_oo[FullRankADVI-full]
18.07s call     tests/variational/test_approximations.py::test_empirical_from_trace
17.45s call     tests/variational/test_inference.py::test_fit_oo[ADVI-mini]
17.16s call     tests/variational/test_inference.py::test_fit_with_nans[True]
14.84s call     tests/variational/test_approximations.py::test_empirical_does_not_support_inference_data
12.65s call     tests/variational/test_inference.py::test_fit_data[SVGD-full]
12.31s call     tests/variational/test_inference.py::test_multiple_minibatch_variables
11.06s call     tests/variational/test_inference.py::test_fit_fn_text[svgd-kwargs4-None]
8.54s call     tests/variational/test_inference.py::test_fit_fn_text[advi-kwargs2-None]
7.52s call     tests/variational/test_inference.py::test_fit_fn_text[fullrank_advi-kwargs3-None]
6.95s call     tests/variational/test_inference.py::test_fit_oo[FullRankADVI-mini]
6.74s call     tests/variational/test_inference.py::test_fit_fn_text[asvgd-kwargs7-None]
6.25s call     tests/variational/test_inference.py::test_fit_with_nans[False]
5.99s call     tests/variational/test_inference.py::test_fit_oo[ADVI-full]
5.46s call     tests/variational/test_opvi.py::test_init_groups[raises0-grouping0]
5.21s call     tests/variational/test_inference.py::test_fit_data[FullRankADVI-mini]
5.06s call     tests/variational/test_inference.py::test_fit_data[FullRankADVI-full]
4.98s call     tests/variational/test_inference.py::test_fit_data[ASVGD-full]
4.83s call     tests/variational/test_inference.py::test_fit_start[SVGD-full]
4.74s call     tests/test_initial_point.py::TestInitvalEvaluation::test_dependent_initvals[False]
4.09s call     tests/variational/test_opvi.py::test_group_api_params[raises0-params0-MeanFieldGroup-kw0-None]
4.00s call     tests/variational/test_opvi.py::test_logq_globals[FullRankGroup: None, MeanFieldGroup: ['one']]
3.83s call     tests/variational/test_approximations.py::test_scale_cost_to_minibatch_works[2]
3.42s setup    tests/variational/test_inference.py::test_replacements[FullRankADVI]
3.32s call     tests/variational/test_inference.py::test_fit_data[ADVI-mini]
3.17s call     tests/variational/test_approximations.py::test_scale_cost_to_minibatch_works[5]
2.75s call     tests/variational/test_inference.py::test_fit_fn_text[svgd-kwargs5-None]
2.61s call     tests/variational/test_inference.py::test_var_replacement
2.59s call     tests/variational/test_inference.py::test_fit_start[ADVI-mini]
2.55s call     tests/variational/test_inference.py::test_fit_start[ADVI-full]
2.25s call     tests/variational/test_approximations.py::test_elbo_beta_kl[8]
2.20s call     tests/variational/test_approximations.py::test_scale_cost_to_minibatch_works[8]
2.10s call     tests/variational/test_approximations.py::test_elbo_beta_kl[5]
2.04s call     tests/variational/test_inference.py::test_fit_start[FullRankADVI-mini]
1.89s call     tests/variational/test_inference.py::test_fit_start[FullRankADVI-full]
1.88s call     tests/variational/test_inference.py::test_fit_data[ADVI-full]
1.88s call     tests/variational/test_inference.py::test_remove_scan_op
1.79s call     tests/variational/test_inference.py::test_clear_cache
1.63s call     tests/variational/test_inference.py::test_profile[FullRankADVI-mini]
1.51s call     tests/variational/test_inference.py::test_profile[ADVI-mini]
1.48s call     tests/variational/test_inference.py::test_profile[FullRankADVI-full]
================= 121 passed, 10 skipped in 639.76s (0:10:39) =================
(14 minutes 42 seconds) ubuntu (ubuntu-20.04, float64, 3.13, tests/distributions/test_censored.py tests/distributions/tes... 
============================= slowest 50 durations =============================
34.06s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_scan
31.42s call     tests/distributions/test_discrete.py::test_orderedlogistic_dimensions[(3, 1)]
25.67s call     tests/distributions/test_discrete.py::test_orderedlogistic_dimensions[(3, 2)]
22.97s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point_scan
21.37s call     tests/distributions/test_censored.py::TestCensored::test_censored_workflow[False]
20.32s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point_scan_recurring
19.74s call     tests/distributions/test_distribution.py::TestPartialObservedRV::test_multivariate_constant_mask_separable[True-False]
18.49s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_categorical_logp_batch_dims
17.92s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_discrete_unif
16.52s call     tests/distributions/test_censored.py::TestCensored::test_censored_workflow[True]
15.36s call     tests/sampling/test_forward.py::TestSamplePriorPredictive::test_multivariate2
14.97s call     tests/distributions/test_discrete.py::TestMoments::test_hyper_geometric_support_point[50-10-n3-size3-expected3]
14.40s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point_nested[-3-1-size2-expected2]
14.15s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_hypergeometric
13.47s call     tests/distributions/test_discrete.py::TestMoments::test_categorical_support_point[p3-size3-expected3]
13.44s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point_nested[-1-1-None-5]
12.84s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_inferred_logp_mixture
12.28s setup    tests/sampling/test_forward.py::TestSamplePPC::test_logging_sampled_basic_rvs_posterior_mutable[MultiTrace]
11.95s call     tests/distributions/test_discrete.py::TestMoments::test_beta_binomial_support_point[10-1-beta3-size3-expected3]
11.50s call     tests/distributions/test_discrete.py::TestMoments::test_negative_binomial_support_point[10-p3-size3-expected3]
11.24s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_categorical_negative_p_symbolic
10.84s call     tests/sampling/test_forward.py::TestCompileForwardSampler::test_lkj_cholesky_cov
10.74s call     tests/distributions/test_discrete.py::TestOrderedProbit::test_expected_categorical
10.43s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point[dist_params3-size3-expected3-<lambda>]
9.73s call     tests/distributions/test_distribution.py::TestPartialObservedRV::test_multivariate_constant_mask_separable[True-True]
9.55s call     tests/distributions/test_discrete.py::test_orderedlogistic_dimensions[(1,)]
8.77s call     tests/distributions/test_custom.py::TestCustomDist::test_serialize_custom_dist
8.64s call     tests/stats/test_log_density.py::TestComputeLogLikelihood::test_multivariate
8.56s call     tests/distributions/test_discrete.py::TestMoments::test_discrete_weibull_support_point[q3-beta3-None-expected3]
8.51s call     tests/distributions/test_discrete.py::TestMoments::test_hyper_geometric_support_point[50-10-23-5-expected1]
8.12s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_negative_binomial
8.07s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point[dist_params2-None-4.4816890703380645-<lambda>]
7.61s call     tests/distributions/test_simulator.py::TestSimulator::test_multiple_simulators
7.58s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point[dist_params1-None-expected1-<lambda>]
7.58s call     tests/distributions/test_simulator.py::TestSimulator::test_custom_dist_sum_stat[float32]
7.43s call     tests/distributions/test_custom.py::TestCustomDist::test_custom_dist_multivariate_logp[None]
7.09s call     tests/distributions/test_discrete.py::TestMoments::test_discrete_uniform_support_point[1-upper3-size3-expected3]
6.94s call     tests/distributions/test_discrete.py::TestDiscreteWeibull::test_distribution
6.92s call     tests/distributions/test_discrete.py::TestMoments::test_hyper_geometric_support_point[50-10-n2-None-expected2]
6.62s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_beta_binomial
6.47s call     tests/distributions/test_simulator.py::TestSimulator::test_nested_simulators
5.91s call     tests/distributions/test_custom.py::TestCustomSymbolicDist::test_custom_dist_default_support_point[dist_params0-None-148.4131591025766-<lambda>]
5.86s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_categorical_bounds
5.63s call     tests/sampling/test_forward.py::TestSamplePPC::test_model_not_drawable_prior
5.54s call     tests/distributions/test_discrete.py::TestMoments::test_beta_binomial_support_point[10-1-1-5-expected1]
5.45s call     tests/distributions/test_distribution.py::TestBugfixes::test_issue_3051[1-MvStudentT-kwargs1]
5.29s call     tests/distributions/test_simulator.py::TestSimulator::test_one_gaussian
5.16s call     tests/distributions/test_custom.py::TestCustomDist::test_dist
5.11s call     tests/distributions/test_discrete.py::TestMoments::test_beta_binomial_support_point[10-1-1-None-5]
5.09s call     tests/distributions/test_discrete.py::TestMatchesScipy::test_bernoulli
=========== 350 passed, 1 xfailed, 344 warnings in 844.98s (0:14:04) ===========
(21 minutes 0 seconds) windows (windows-latest, float64, 3.10, tests/model/test_core.py tests/sampling/test_mcmc.py) 
============================ slowest 50 durations =============================
58.29s call     tests/sampling/test_mcmc.py::test_step_args
35.88s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_separable
34.56s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
29.70s call     tests/sampling/test_mcmc.py::test_sample_var_names
26.14s call     tests/sampling/test_mcmc.py::test_blas_cores
25.95s call     tests/sampling/test_mcmc.py::test_sample_find_MAP_does_not_modify_start
25.44s call     tests/model/test_core.py::TestValueGradFunction::test_missing_data
21.05s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-advi]
20.84s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-advi]
20.75s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-advi]
20.67s call     tests/model/test_core.py::TestImputationMissingData::test_truncated_normal
20.29s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-map]
20.10s call     tests/sampling/test_mcmc.py::TestSample::test_sample_does_not_rely_on_external_global_seeding
19.73s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-map]
19.69s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-map]
19.07s call     tests/model/test_core.py::TestValueGradFunction::test_edge_case
18.84s call     tests/model/test_core.py::TestNested::test_nested_model_to_netcdf
18.39s call     tests/model/test_core.py::TestImputationMissingData::test_interval_missing_observations
17.79s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-2-jitter+adapt_diag]
17.46s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-None-2-jitter+adapt_diag]
17.31s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-1-2-jitter+adapt_diag]
15.03s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[True]
14.57s call     tests/model/test_core.py::test_model_d2logp[True]
14.44s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[False]
11.20s call     tests/model/test_core.py::TestImputationMissingData::test_missing_logp1
10.82s call     tests/sampling/test_mcmc.py::TestSample::test_default_sample_does_not_set_global_seed
10.68s call     tests/model/test_core.py::TestImputationMissingData::test_missing_dual_observations
10.14s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
9.98s call     tests/model/test_core.py::TestPytensorRelatedLogpBugs::test_pytensor_switch_broadcast_edge_cases_2
9.75s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[map]
9.50s call     tests/sampling/test_mcmc.py::TestType::test_float32
8.94s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
8.89s call     tests/sampling/test_mcmc.py::TestSample::test_parallel_start
8.88s call     tests/sampling/test_mcmc.py::test_no_init_nuts_compound
8.69s call     tests/sampling/test_mcmc.py::test_init_nuts
8.61s call     tests/sampling/test_mcmc.py::TestSampleReturn::test_logs_sampler_warnings[2]
8.61s call     tests/sampling/test_mcmc.py::test_no_init_nuts_step
8.58s call     tests/sampling/test_mcmc.py::TestSample::test_sample_callback
8.35s call     tests/model/test_core.py::TestTransformArgs::test_default_transform_is_applied
7.26s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi_map]
6.65s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_unseparable
6.42s call     tests/model/test_core.py::TestModelDebug::test_invalid_observed_value
6.34s call     tests/model/test_core.py::TestImputationMissingData::test_missing_basic[pandas]
6.32s call     tests/sampling/test_mcmc.py::TestNamedSampling::test_shared_named
6.30s call     tests/model/test_core.py::test_model_logp_fast_compile
6.22s call     tests/model/test_core.py::TestImputationMissingData::test_coordinates
5.59s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-map]
5.40s call     tests/model/test_core.py::test_tempered_logp_dlogp[False]
5.11s call     tests/model/test_core.py::TestImputationMissingData::test_missing_logp2
4.75s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[2-seeds5-1-advi]
=========== 222 passed, 5 skipped, 30 warnings in 958.34s (0:15:58) ===========
(14 minutes 9 seconds) ubuntu (ubuntu-20.04, float64, 3.13, tests/tuning/test_scaling.py tests/tuning/test_starting.py t... 
============================= slowest 50 durations =============================
27.66s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a1-None]
27.61s call     tests/sampling/test_mcmc.py::TestSample::test_random_seed[1-None-1-advi]
27.47s call     tests/tuning/test_starting.py::test_find_MAP_discrete
21.85s call     tests/distributions/test_dist_math.py::test_multinomial_check_parameters
20.30s call     tests/tuning/test_starting.py::test_find_MAP_issue_4488
20.06s call     tests/tuning/test_starting.py::test_find_MAP
14.85s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_triangular[-10-0-200-size1]
14.60s call     tests/distributions/test_transform.py::test_interval
14.48s call     tests/tuning/test_starting.py::test_mle_jacobian[True]
14.05s call     tests/distributions/test_transform.py::test_simplex_bounds
13.66s call     tests/tuning/test_scaling.py::test_guess_scaling
13.40s call     tests/distributions/test_transform.py::test_simplex
12.46s call     tests/distributions/test_transform.py::test_ordered
12.13s call     tests/sampling/test_mcmc.py::test_step_args
11.71s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_half_normal_ordered[sigma1-size1]
11.62s call     tests/sampling/test_mcmc.py::test_sample_var_names
10.03s call     tests/sampling/test_parallel.py::test_remote_pipe_closed
9.78s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[map]
9.56s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_beta[0.5-0.5-size1]
9.33s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_other[0.5-5.5-size1-transform1]
8.98s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi]
8.76s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_beta[a2-b2-size2]
8.72s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_half_normal[5.0-size1]
8.60s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a2-size2]
8.52s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_other[lower2-upper2-size2-transform2]
8.38s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_ordered[lower1-upper1-size1]
8.36s call     tests/sampling/test_mcmc.py::TestType::test_float32
8.34s call     tests/distributions/test_dist_math.py::TestI0e::test_grad
8.32s call     tests/tuning/test_starting.py::test_accuracy_non_normal
8.15s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform[0.5-5.5-size1]
8.08s call     tests/sampling/test_parallel.py::test_spawn_densitydist_bound_method
7.90s call     tests/sampling/test_mcmc.py::TestSample::test_sample_init[advi]
7.74s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[True]
7.37s call     tests/tuning/test_starting.py::test_accuracy_normal
7.28s call     tests/sampling/test_parallel.py::test_bad_unpickle
7.16s call     tests/sampling/test_parallel.py::test_context
7.15s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_dirichlet[a0-None]
7.14s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform[lower2-upper2-size2]
7.03s call     tests/sampling/test_parallel.py::test_spawn_densitydist_function
6.95s call     tests/sampling/test_mcmc.py::TestSample::test_transform_with_rv_dependency[False]
6.93s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_uniform_ordered[0.0-1.0-size0]
6.93s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_beta_ordered[a1-b1-size1]
6.74s call     tests/distributions/test_transform.py::test_chain_jacob_det
6.66s call     tests/sampling/test_mcmc.py::test_exec_nuts_init[advi_map]
6.33s call     tests/distributions/test_transform.py::test_log_exp_m1
6.27s call     tests/sampling/test_parallel.py::test_explicit_sample[spawn]
6.15s call     tests/tuning/test_starting.py::test_mle_jacobian[False]
6.03s call     tests/distributions/test_transform.py::TestElementWiseLogp::test_half_normal[sigma2-size2]
5.86s call     tests/sampling/test_mcmc.py::TestNamedSampling::test_shared_named
5.24s call     tests/distributions/test_transform.py::test_sum_to_1
====== 214 passed, 7 skipped, 1 xfailed, 39 warnings in 811.19s (0:13:31) ======
(16 minutes 59 seconds) windows (windows-latest, float64, 3.10, tests/gp/test_cov.py tests/gp/test_gp.py tests/gp/test_me... 
============================ slowest 50 durations =============================
66.30s setup    tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
30.01s call     tests/smc/test_smc.py::TestSMC::test_marginal_likelihood
27.41s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
27.23s call     tests/smc/test_smc.py::TestSMC::test_unobserved_categorical
21.10s call     tests/gp/test_gp.py::TestMarginalVsLatent::testLatentMultioutput
20.58s call     tests/smc/test_smc.py::TestSMC::test_sample
20.02s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginal
19.09s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_2_param_tensor
18.00s call     tests/sampling/test_parallel.py::test_sampling_with_random_generator_matches[2]
16.51s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[VFE]
15.37s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[VFE]
15.16s call     tests/smc/test_smc.py::TestSMC::test_return_datatype[2]
14.90s call     tests/smc/test_smc.py::TestMHKernel::test_normal_model
14.29s call     tests/smc/test_smc.py::TestSMC::test_start
14.27s call     tests/smc/test_smc.py::TestSMC::test_kernel_kwargs
13.55s call     tests/gp/test_gp.py::TestTP::testTPvsLatent
13.42s call     tests/sampling/test_parallel.py::test_iterator
12.98s call     tests/sampling/test_parallel.py::test_remote_pipe_closed
12.96s call     tests/gp/test_gp.py::TestMarginalKron::testMarginalKronvsMarginal
12.77s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[FITC]
12.55s call     tests/smc/test_smc.py::TestSMC::test_unobserved_bernoulli
12.42s call     tests/smc/test_smc.py::TestSMC::test_return_datatype[1]
12.30s call     tests/gp/test_cov.py::TestCovAdd::test_symadd_cov
11.97s call     tests/gp/test_gp.py::TestMarginalKron::testMarginalKronvsMarginalpredict
11.94s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_1_param
10.79s call     tests/sampling/test_parallel.py::test_spawn_densitydist_bound_method
10.58s call     tests/gp/test_gp.py::TestTP::testTPvsLatentReparameterized
10.19s call     tests/sampling/test_parallel.py::test_spawn_densitydist_function
10.11s call     tests/sampling/test_parallel.py::test_context
9.74s call     tests/smc/test_smc.py::TestSMC::test_parallel_custom
9.52s call     tests/ode/test_ode.py::test_logp_scalar_ode
9.26s call     tests/sampling/test_parallel.py::test_bad_unpickle
9.14s call     tests/gp/test_cov.py::TestCovPSD::test_covpsd_add
8.93s call     tests/ode/test_ode.py::test_simulate
8.38s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveLatent
8.34s call     tests/ode/test_ode.py::TestDiffEqModel::test_scalar_ode_1_param
8.33s call     tests/gp/test_cov.py::TestCovKron::test_multiops
8.09s call     tests/gp/test_cov.py::TestCovProd::test_multiops
7.64s call     tests/smc/test_smc.py::TestSMC::test_convergence_checks
7.36s call     tests/sampling/test_parallel.py::test_explicit_sample[spawn]
7.34s call     tests/gp/test_gp.py::TestLatentKron::testLatentKronvsLatent
6.94s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[DTC]
6.70s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_3_params
6.41s setup    tests/gp/test_gp.py::TestMarginalKron::testMarginalKronvsMarginalpredict
6.21s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[DTC]
6.15s call     tests/gp/test_cov.py::TestCovAdd::test_rightadd_whitenoise
6.06s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_scalar_2_param
6.04s setup    tests/gp/test_gp.py::TestMarginalVsLatent::testLatent1
6.03s call     tests/smc/test_smc.py::TestMHKernel::test_proposal_dist_shape
5.98s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_1_param
====== 177 passed, 2 skipped, 1 xfailed, 5 warnings in 836.66s (0:13:56) ======
(21 minutes 10 seconds) ubuntu (ubuntu-20.04, float64, 3.13, tests/distributions/test_timeseries.py tests/gp/test_cov.py ... 
============================= slowest 50 durations =============================
35.32s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_prior[cov_func0-centered]
35.13s call     tests/distributions/test_timeseries.py::TestAR::test_batched_sigma
31.95s setup    tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
31.42s call     tests/distributions/test_timeseries.py::TestRandomWalk::test_init_logp_univariate[init_dist0]
31.27s call     tests/distributions/test_timeseries.py::TestPredefinedRandomWalk::test_gaussian_inference
27.64s call     tests/gp/test_hsgp_approx.py::TestHSGPPeriodic::test_prior[100.0-cov_func0]
25.38s call     tests/distributions/test_timeseries.py::TestAR::test_batched_size[False]
25.04s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_prior[cov_func1-noncentered]
24.14s call     tests/gp/test_hsgp_approx.py::TestHSGPPeriodic::test_conditional_periodic[cov_func0]
23.38s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_conditional[cov_func0-centered]
23.28s call     tests/gp/test_hsgp_approx.py::TestHSGP::test_conditional[cov_func1-noncentered]
22.60s call     tests/distributions/test_timeseries.py::TestAR::test_multivariate_init_dist
22.38s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[FITC]
21.18s call     tests/model/test_core.py::TestImputationMissingData::test_missing_multivariate_separable
20.21s call     tests/distributions/test_timeseries.py::TestRandomWalk::test_support_point[init_dist1-innovation_dist1-9-size1-expected1-True]
19.09s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[True-omega]
18.84s call     tests/distributions/test_timeseries.py::TestRandomWalk::test_init_logp_multivariate[init_dist0]
18.54s call     tests/distributions/test_timeseries.py::TestRandomWalk::test_init_logp_univariate[init_dist1]
17.90s call     tests/distributions/test_timeseries.py::TestAR::test_batched_size[True]
16.95s call     tests/distributions/test_timeseries.py::TestEulerMaruyama::test_linear_model
15.49s call     tests/distributions/test_timeseries.py::TestRandomWalk::test_support_point[init_dist2-innovation_dist2-9-None-expected2-True]
15.43s call     tests/model/test_core.py::TestImputationMissingData::test_truncated_normal
15.08s call     tests/model/transform/test_conditioning.py::test_change_value_transforms
14.76s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[True-beta_1]
14.39s call     tests/distributions/test_timeseries.py::TestPredefinedRandomWalk::test_mvgaussian_with_chol_cov_rv[chol]
14.10s call     tests/model/test_core.py::TestValueGradFunction::test_missing_data
13.70s call     tests/distributions/test_timeseries.py::TestGARCH11::test_support_point[size1-expected1]
13.32s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[True-alpha_1]
13.24s call     tests/model/test_core.py::TestImputationMissingData::test_interval_missing_observations
12.88s call     tests/gp/test_gp.py::TestMarginalVsLatent::testLatentMultioutput
12.76s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginal
12.51s call     tests/gp/test_gp.py::TestMarginalVsMarginalApprox::test_fits_and_preds[VFE]
11.78s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[False-beta_1]
11.73s call     tests/gp/test_gp.py::TestMarginalKron::testMarginalKronvsMarginal
11.68s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[False-omega]
11.63s call     tests/distributions/test_timeseries.py::TestGARCH11::test_batched_size[False-alpha_1]
11.21s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[VFE]
10.84s call     tests/distributions/test_timeseries.py::TestGARCH11::test_support_point[None-expected0]
10.74s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[FITC]
10.29s call     tests/distributions/test_timeseries.py::TestEulerMaruyama::test_batched_size[True-1]
10.26s call     tests/distributions/test_timeseries.py::TestEulerMaruyama::test_batched_size[True-2]
9.85s call     tests/gp/test_gp.py::TestMarginalKron::testMarginalKronvsMarginalpredict
8.57s call     tests/model/test_core.py::test_model_d2logp[True]
8.43s call     tests/ode/test_ode.py::TestDiffEqModel::test_vector_ode_1_param
7.80s call     tests/model/test_core.py::TestImputationMissingData::test_missing_dual_observations
7.73s call     tests/ode/test_ode.py::TestSensitivityInitialCondition::test_sens_ic_vector_2_param_tensor
7.71s call     tests/distributions/test_timeseries.py::TestAR::test_batched_rhos
7.58s call     tests/distributions/test_timeseries.py::TestGARCH11::test_logp
6.65s call     tests/model/test_core.py::TestImputationMissingData::test_missing_logp1
6.48s call     tests/gp/test_gp.py::TestGPAdditive::testAdditiveMarginalApprox[DTC]
=========== 462 passed, 3 xfailed, 20 warnings in 1242.48s (0:20:42) ===========
(15 minutes 26 seconds) windows (windows-latest, float64, 3.10, tests/step_methods/test_metropolis.py tests/step_methods/... 
============================ slowest 50 durations =============================
101.43s setup    tests/step_methods/test_metropolis.py::TestMetropolisUniform::test_kstest
61.71s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSLKJCholeskyCov::test_kstest
52.68s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist3]
52.61s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist4]
36.41s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSUniform::test_kstest
35.51s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSUniform2::test_kstest
34.34s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSUniform3::test_kstest
28.66s setup    tests/step_methods/test_slicer.py::TestSliceUniform::test_kstest
26.33s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist2]
25.93s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSStudentT::test_kstest
24.74s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSBetaBinomial::test_kstest
22.28s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist1]
21.50s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update[batched_dist0]
20.32s setup    tests/step_methods/hmc/test_nuts.py::TestNUTSNormal::test_kstest
13.42s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_elemwise_update_different_scales
12.94s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_emits_energy_warnings
12.61s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_discrete
11.67s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_tuning_reset
10.37s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_multiple_samplers
9.11s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_epsilon_parallel
8.36s call     tests/step_methods/test_compound.py::TestCompoundStep::test_non_blocked
7.32s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_continuous[<function TestStepMetropolis.<lambda> at 0x000002077DB44790>-4000]
6.45s call     tests/step_methods/test_metropolis.py::TestDEMetropolis::test_demcmc_tune_parameter
5.99s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_categorical[uniform]
5.09s call     tests/step_methods/test_metropolis.py::TestRVsAssignmentMetropolis::test_discrete_steps[BinaryMetropolis]
4.44s call     tests/step_methods/test_metropolis.py::TestMetropolis::test_multinomial_no_elemwise_update
3.87s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_bad_init_nonparallel
3.54s call     tests/step_methods/test_slicer.py::TestStepSlicer::test_step_continuous[<function TestStepSlicer.<lambda> at 0x000002077DB44EE0>-2000]
3.47s call     tests/step_methods/test_metropolis.py::test_sampling_state[Metropolis-simple_model]
3.23s call     tests/step_methods/hmc/test_nuts.py::TestStepNUTS::test_step_continuous[<lambda>-1000_0]
3.06s call     tests/step_methods/test_compound.py::TestCompoundStep::test_name
2.82s call     tests/step_methods/test_metropolis.py::TestStepMetropolis::test_step_categorical[proportional]
2.76s call     tests/step_methods/test_slicer.py::TestStepSlicer::test_step_continuous[<function TestStepSlicer.<lambda> at 0x000002077DB45BD0>-2000]
2.59s call     tests/step_methods/test_compound.py::TestStepCompound::test_step_continuous[<function TestStepCompound.<lambda> at 0x000002077DB460E0>-1000]
2.36s call     tests/step_methods/hmc/test_nuts.py::TestStepNUTS::test_step_continuous[<lambda>-1000_1]
2.26s call     tests/step_methods/test_compound.py::TestStatsBijection::test_flatten_steps
2.07s call     tests/step_methods/hmc/test_nuts.py::TestNutsCheckTrace::test_sampler_stats
1.60s call     tests/step_methods/test_compound.py::TestCompoundStep::test_blocked
1.44s call     tests/step_methods/hmc/test_nuts.py::test_nuts_step_legacy_value_grad_function
1.41s call     tests/step_methods/hmc/test_hmc.py::TestStepHamiltonianMC::test_step_continuous[<lambda>-1000_1]
1.41s call     tests/step_methods/hmc/test_hmc.py::test_leapfrog_reversible
1.40s call     tests/step_methods/hmc/test_hmc.py::TestStepHamiltonianMC::test_step_continuous[<lambda>-1000_0]
1.13s call     tests/step_methods/test_compound.py::TestStatsMetadata::test_stats_from_steps
1.09s call     tests/step_methods/test_metropolis.py::TestRVsAssignmentMetropolis::test_discrete_steps[BinaryGibbsMetropolis]
1.00s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_lambda_sequential
0.97s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_reset
0.75s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tuning_none
0.53s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_tune_drop_fraction
0.51s call     tests/step_methods/hmc/test_hmc.py::TestRVsAssignmentHamiltonianMC::test_continuous_steps[HamiltonianMC-step_kwargs0]
0.40s call     tests/step_methods/test_metropolis.py::TestDEMetropolisZ::test_custom_proposal_dist
========= 118 passed, 6 skipped, 15950 warnings in 709.17s (0:11:49) ==========
(28 minutes 6 seconds) ubuntu (ubuntu-20.04, float64, 3.13, tests/backends/test_mcbackend.py tests/backends/test_zarr.py... 
============================= slowest 50 durations =============================
41.48s call     tests/distributions/test_truncated.py::test_change_truncated_size
31.49s call     tests/distributions/test_truncated.py::test_truncated_multiple_rngs
29.12s call     tests/backends/test_mcbackend.py::test_make_runmeta_and_point_fn
28.00s call     tests/distributions/test_truncated.py::test_truncated_support_point[truncated_dist2--1-None-shape2-expected2]
27.33s call     tests/distributions/test_truncated.py::test_vectorized_bounds
24.99s call     tests/logprob/test_transforms.py::TestTransform::test_check_jac_det[transform2]
24.57s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[size19-10-size29-10-2-False-2]
24.37s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[size15-10-size25-10-1-True-1]
21.28s call     tests/distributions/test_truncated.py::test_truncation_continuous_random[False-True-rejection--1-inf]
21.13s call     tests/logprob/test_transform_value.py::test_default_value_transform_logprob[pt_dist17-dist_params17-<lambda>-size17]
19.41s call     tests/logprob/test_transform_value.py::test_default_value_transform_logprob[pt_dist16-dist_params16-<lambda>-None]
19.07s call     tests/logprob/test_scan.py::test_scan_joint_logprob[False]
16.05s call     tests/distributions/test_truncated.py::test_truncated_support_point[truncated_dist1--1-2-shape1-expected1]
15.72s setup    tests/backends/test_zarr.py::test_record[single_step-include_transformed-frequent_writes]
15.63s call     tests/distributions/test_truncated.py::test_truncated_maxwell_dist
14.68s call     tests/logprob/test_composite_logprob.py::test_affine_join_interdependent[False]
14.50s call     tests/logprob/test_transform_value.py::test_weakref_leak
11.79s call     tests/logprob/test_linalg.py::test_matrix_vector_transform[batch_shape1-True]
11.71s call     tests/logprob/test_scan.py::test_convert_outer_out_to_in_sit_sot
11.60s call     tests/distributions/test_truncated.py::test_truncation_continuous_random[False-False-icdf--1-inf]
11.60s call     tests/logprob/test_transforms.py::test_extra_bijective_rv_transforms[pt_transform2-transform2]
10.99s call     tests/distributions/test_truncated.py::test_truncation_continuous_random[False-True-icdf--1-inf]
10.82s call     tests/distributions/test_truncated.py::test_truncated_inference
10.76s call     tests/logprob/test_linalg.py::test_matrix_vector_transform[batch_shape0-True]
10.75s call     tests/logprob/test_transform_value.py::test_default_value_transform_logprob[pt_dist15-dist_params15-<lambda>-size15]
10.55s call     tests/distributions/test_truncated.py::test_truncated_support_point[truncated_dist3-None-upper3-shape3-expected3]
10.08s call     tests/backends/test_mcbackend.py::TestMcBackendSampling::test_return_multitrace[False]
10.01s call     tests/distributions/test_truncated.py::test_truncation_logprob_bound_check
9.84s 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]
9.63s call     tests/logprob/test_mixture.py::test_expand_indices_moved_subspaces[A_parts0-indices0]
9.58s call     tests/logprob/test_mixture.py::test_expand_indices_basic[A_parts8-indices8]
9.56s call     tests/distributions/test_truncated.py::test_truncation_continuous_random[False-False-rejection--1-inf]
9.24s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[size12-2-size22-2-0-True-0]
9.18s call     tests/distributions/test_truncated.py::test_truncation_continuous_random[True-False-icdf--1-inf]
9.16s call     tests/distributions/test_truncated.py::test_truncation_discrete_random[rejection-2-inf]
8.98s call     tests/logprob/test_cumsum.py::test_normal_cumsum[size4-0]
8.70s call     tests/distributions/test_truncated.py::test_truncated_gamma
8.36s call     tests/backends/test_mcbackend.py::test_get_variables_and_point_fn
8.21s call     tests/logprob/test_order.py::test_max_logprob[shape4-0.9-None]
8.06s call     tests/distributions/test_truncated.py::test_truncation_continuous_random[True-True-icdf--1-inf]
7.46s call     tests/logprob/test_order.py::test_max_logprob[3-0.85--1]
7.35s call     tests/logprob/test_mixture.py::test_expand_indices_basic[A_parts2-indices2]
7.27s call     tests/logprob/test_checks.py::test_specify_shape_logprob
7.20s call     tests/distributions/test_truncated.py::test_truncated_symbolic_rv[<lambda>0]
6.94s call     tests/logprob/test_scan.py::test_scan_multiple_output_types
6.80s call     tests/distributions/test_truncated.py::test_truncated_custom_dist_indexed_argument[rv_op0]
6.42s call     tests/logprob/test_tensor.py::test_measurable_join_univariate[size14-size24-0-False]
6.29s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[size17-5-size27-5-0-False-0]
6.26s call     tests/logprob/test_transform_value.py::test_scan_transform
6.21s call     tests/logprob/test_tensor.py::test_measurable_join_multivariate[None-2-None-2-0-True-0]
===== 667 passed, 64 skipped, 9 xfailed, 96 warnings in 1651.96s (0:27:31) =====

You can find more information on how to contribute here

Automatically generated by GitHub Action
Latest run date: 2025-03-24
Run logs: 13965326294
@williambdean williambdean changed the title Speed up test times 🚀 Speed up test times 🚀 Feb 20, 2025
@nataziel
Copy link
Contributor

@wd60622 the autogenerated post didn't actually include the list of slow tests 🙈

@williambdean
Copy link
Contributor Author

@wd60622 the autogenerated post didn't actually include the list of slow tests 🙈

There is no action yet and I am still working on it 😎 but I need an issue to populate

@williambdean williambdean mentioned this issue Feb 28, 2025
Merged
11 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@williambdean @nataziel