Skip to content
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.

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

Speed up test times 🚀 #1158

Open
wd60622 opened this issue Nov 3, 2024 · 3 comments
Open

Speed up test times 🚀 #1158

wd60622 opened this issue Nov 3, 2024 · 3 comments
Labels

Comments

@wd60622
Copy link
Contributor

wd60622 commented Nov 3, 2024

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

Here are some of the slowest test times:

(13 minutes 14 seconds) test_slow (tests/clv/models/test_beta_geo_beta_binom.py)
============================= slowest 50 durations =============================
683.26s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_model_convergence[mcmc-0.3]
20.62s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_model_convergence[map-0.2]
0.06s setup    tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_model_convergence[mcmc-0.3]

0.005s hidden.  Use -vv to show these durations.)
============= 2 passed, 16 skipped, 1 warning in 705.96s (0:11:45) =============
(10 minutes 59 seconds) test (3.10, false, tests/mmm --ignore tests/mmm/test_tvp.py --ignore tests/mmm/test_budget_optimi...
============================= slowest 50 durations =============================
21.45s call     tests/mmm/test_mmm.py::TestMMM::test_allocate_budget_to_maximize_response
17.08s call     tests/mmm/test_mmm.py::TestMMM::test_init[no_time_varying_media-time_varying_intercept-no_yearly_seasonality-single_channel-no_control-adstock_max_lag=1]
12.79s call     tests/mmm/test_lift_test.py::test_works_with_negative_delta
10.84s call     tests/mmm/test_multidimensional.py::test_fit[tvp intercept only, no fourier-Marginal model]
10.50s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-delayed]
10.40s call     tests/mmm/components/test_adstock.py::test_apply[x1-channel-delayed]
8.84s call     tests/mmm/test_linear_trend.py::test_apply[with_intercept]
8.71s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-weibull_pdf]
7.70s setup    tests/mmm/test_multidimensional.py::test_simple_fit
7.41s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[scalar-vector]
7.21s call     tests/mmm/test_multidimensional.py::test_fit[tvp intercept only, no fourier-County model]
6.87s call     tests/mmm/test_mmm.py::TestMMM::test_channel_contributions_forward_pass_is_consistent
6.30s call     tests/mmm/test_utility.py::test_mean_tightness_score
6.30s call     tests/mmm/test_linear_trend.py::test_apply_additional_dims[scalar]
5.88s call     tests/mmm/test_utility.py::test_conditional_value_at_risk
5.76s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[scalar-scalar]
4.28s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-no_yearly_seasonality-single_channel-no_control-adstock_max_lag=1]
4.05s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[vector-scalar]
3.69s call     tests/mmm/test_multidimensional.py::test_mmm_with_events
3.63s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-weibull_cdf]
3.48s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-geometric]
3.45s call     tests/mmm/test_fourier.py::test_sample_curve[yearly]
3.39s call     tests/mmm/test_multidimensional.py::test_fit[tvp media only with fourier-County model]
3.33s call     tests/mmm/test_multidimensional.py::test_fit[tvp media only with fourier-Marginal model]
3.33s setup    tests/mmm/test_mmm.py::test_save_load_with_media_transformation
3.32s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-no_time_varying_intercept-no_yearly_seasonality-single_channel-no_control-adstock_max_lag=1]
3.26s call     tests/mmm/components/test_base.py::test_transform_sample_curve_with_variable_factory
3.25s call     tests/mmm/test_multidimensional.py::test_fit[no tvps or fourier-Marginal model]
3.11s call     tests/mmm/test_mmm.py::test_save_load_with_tvp[True-True]
2.97s call     tests/mmm/test_utility.py::test_sharpe_ratio
2.97s call     tests/mmm/test_utility.py::test_covariance_matrix_matches_numpy[data0]
2.91s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-no_yearly_seasonality-single_channel-one_control-adstock_max_lag=1]
2.76s call     tests/mmm/test_mmm.py::test_plotting_media_transform_workflow[adstock]
2.75s call     tests/mmm/test_mmm.py::TestMMM::test_init[no_time_varying_media-no_time_varying_intercept-yearly_seasonality-single_channel-no_control-adstock_max_lag=1]
2.73s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-yearly_seasonality-single_channel-one_control-adstock_max_lag=1]
2.63s call     tests/mmm/test_multidimensional.py::test_fit[tvps and fourier-County model]
2.63s setup    tests/mmm/test_mmm.py::test_plotting_media_transform_workflow[adstock]
2.57s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-no_yearly_seasonality-multiple_channel-two_controls-adstock_max_lag=4]
2.47s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-no_yearly_seasonality-single_channel-no_control-adstock_max_lag=4]
2.35s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-no_time_varying_intercept-no_yearly_seasonality-single_channel-one_control-adstock_max_lag=1]
2.35s call     tests/mmm/test_multidimensional.py::test_fit[tvps and fourier-Marginal model]
2.33s call     tests/mmm/test_events.py::test_gaussian_basis_curve_sampling
2.29s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-yearly_seasonality-multiple_channel-one_control-adstock_max_lag=4]
2.28s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-yearly_seasonality-single_channel-no_control-adstock_max_lag=4]
2.26s call     tests/mmm/test_multidimensional.py::test_fit[no tvps with fourier-County model]
2.25s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-yearly_seasonality-single_channel-two_controls-adstock_max_lag=1]
2.24s call     tests/mmm/test_mmm.py::TestMMM::test_init[no_time_varying_media-time_varying_intercept-yearly_seasonality-single_channel-one_control-adstock_max_lag=1]
2.23s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-yearly_seasonality-multiple_channel-no_control-adstock_max_lag=4]
2.21s call     tests/mmm/test_fourier.py::test_additional_dimension[yearly]
2.20s call     tests/mmm/components/test_adstock.py::test_apply[x1-channel-weibull_pdf]
================ 1085 passed, 184 warnings in 565.85s (0:09:25) ================
(8 minutes 50 seconds) test_slow (tests/clv/models/test_pareto_nbd.py)
============================= slowest 50 durations =============================
405.50s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_model_convergence[mcmc-0.1]
25.19s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_model_convergence[map-0.2]
7.87s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_model_convergence[demz-0.2]
0.04s setup    tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_model_convergence[mcmc-0.1]

0.005s hidden.  Use -vv to show these durations.)
=========== 3 passed, 22 skipped, 4000 warnings in 440.95s (0:07:20) ===========
(7 minutes 41 seconds) test (3.10, false, tests/mmm/test_tvp.py tests/mmm/test_budget_optimizer.py tests/mmm/test_hsgp.p...
============================= slowest 50 durations =============================
60.39s call     tests/mmm/test_tvp.py::test_time_varying_prior
30.42s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget[default_minimizer_kwargs]
23.68s call     tests/mmm/test_tvp.py::test_multidimensional
16.15s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_adstock_vectorized[PDF]
14.31s call     tests/mmm/test_hsgp.py::test_curve_workflow[HSGP]
13.44s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_custom_response_constraint[budget=10->resp=5]
8.56s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_adstock_vectorized[CDF]
8.34s call     tests/mmm/test_hsgp.py::test_curve_workflow[HSGPPeriodic]
7.61s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x0-1-1-4]
7.29s call     tests/mmm/test_hsgp.py::test_higher_dimension_hsgp
6.91s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[After]
6.15s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x2-0.7-1-100]
4.84s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_geometric_adstock_vectorized_logistic_saturation
4.42s call     tests/mmm/test_hsgp.py::test_hsgp_with_shared_data
3.97s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[After]
3.65s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x2-0.7-1-100]
3.62s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x6-0.8-1-50]
3.57s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_parameterization_transformation[x0-20-0.5]
3.33s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x0-1-1-4]
3.16s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x1-0.3-0.5-10]
3.13s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x1-0.7-100]
2.97s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_infeasible_constraints[100-budget_bounds0-parameters0-custom_constraints0]
2.87s call     tests/mmm/test_transformers.py::test_batched_convolution_broadcasting
2.70s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_x_zero[After]
2.55s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_parameterization_transformation[x1-100-0.5]
2.16s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[Before]
2.13s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_custom_response_constraint[budget=50->resp=10]
2.04s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x5-0.8-50]
2.02s call     tests/mmm/test_tvp.py::test_calling_without_default_args
2.00s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_bad_alpha[less_than_zero_1]
1.96s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[Overlap]
1.91s call     tests/mmm/test_transformers.py::test_batched_convolution[ndarray-0-After]
1.91s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x3-0.5-7]
1.83s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x0-0.5-1.0]
1.79s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_baselined_range[x0-10-0.5-0.5]
1.78s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_hill_sigmoid_monotonicity[1-1-0]
1.77s call     tests/mmm/test_transformers.py::test_batched_convolution[TensorVariable--1-After]
1.76s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x1-0.3-0.5-10]
1.66s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_custom_minimize_args
1.65s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_delayed_adstock_vectorized_logistic_saturation
1.56s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x1-0.2-19.0]
1.53s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x0-0.3-10]
1.53s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x2-0.6-5.0]
1.42s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget[zero_total_budget]
1.42s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget[custom_minimizer_kwargs]
1.41s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x6-0.8-1-50]
1.37s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[Overlap]
1.37s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[Before]
1.34s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_output_type[After]
1.31s call     tests/mmm/test_hsgp.py::test_hsgp_class_method_ls_upper_changes_distribution
================= 210 passed, 14 warnings in 325.25s (0:05:25) =================
(4 minutes 46 seconds) test_slow (--ignore tests/clv/models/test_beta_geo_beta_binom.py --ignore tests/clv/models/test_p...
============================= slowest 50 durations =============================
44.44s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_model_convergence[mcmc-0.1]
33.18s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_model_convergence
32.27s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModelIndividual::test_model_convergence
25.46s call     tests/clv/models/test_modified_beta_geo.py::TestModifiedBetaGeoModel::test_model_convergence[mcmc-0.075]
24.64s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_model_convergence
15.98s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_model_convergence[map-0.2]
8.60s call     tests/clv/models/test_modified_beta_geo.py::TestModifiedBetaGeoModel::test_model_convergence[map-0.15]
0.63s setup    tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_model_convergence
0.05s setup    tests/clv/models/test_gamma_gamma.py::TestGammaGammaModelIndividual::test_model_convergence
0.04s setup    tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_model_convergence[mcmc-0.1]
0.03s setup    tests/clv/models/test_modified_beta_geo.py::TestModifiedBetaGeoModel::test_model_convergence[mcmc-0.075]

0.005s hidden.  Use -vv to show these durations.)
=========== 7 passed, 1754 skipped, 25 warnings in 195.88s (0:03:15) ===========
(9 minutes 35 seconds) test (3.10, false, tests/clv)
============================= slowest 50 durations =============================
38.30s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_covariate_model_convergence
30.19s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[1]
22.24s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_expected_purchases[1]
20.93s call     tests/clv/test_utils.py::TestCustomerLifetimeValue::test_customer_lifetime_value_as_gg_method[fitted_pnbd]
20.51s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_posterior_distributions[map]
19.55s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_distribution_customer_churn_time
17.77s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_posterior_distributions[map]
15.01s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value3-0.3-p3-10-logp3]
12.74s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_lifetimes[batch_shape1]
12.69s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_spend[True]
12.62s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_logp
10.06s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_expectation_method
9.65s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_save_load
9.63s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_distribution_method
8.71s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_distribution_new_customer
8.36s call     tests/clv/models/test_basic.py::TestCLVModel::test_fit_demz
7.28s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value4-0.55-10.58-0.61-beta4-12]
7.20s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_lifetimes[batch_shape0]
6.79s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_excel
6.66s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value0-0.4-0.15-10--8.39147106159807]
6.43s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_numerically_stable_logp[0-0--0.59947382]
5.95s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value1-lam1-0.15-10-logp1]
5.92s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_invalid_value_logp
5.34s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[6]
5.34s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[3]
5.33s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[1]
4.54s call     tests/clv/models/test_basic.py::TestCLVModel::test_fit_mcmc
4.37s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[6]
4.17s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[3]
3.82s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value1-r1-10.58-0.61-11.67-12]
2.74s call     tests/clv/test_distributions.py::TestContContract::test_continuous_contractual[value3-0.3-p3-10-logp3]
2.54s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value2-lam2-0.15-10-logp2]
2.33s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_spend[False]
2.07s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_posterior_distributions[mcmc]
2.04s call     tests/clv/test_distributions.py::TestBetaGeoNBD::test_bg_nbd[value3-0.55-11.67-0.61-10.58-T3]
2.01s call     tests/clv/test_distributions.py::TestBetaGeoNBD::test_bg_nbd[value0-0.55-10.58-0.61-11.67-12]
1.93s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_posterior_distributions[mcmc]
1.92s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value0-0.55-10.58-0.61-11.67-12]
1.92s call     tests/clv/test_distributions.py::TestContContract::test_continuous_contractual[value1-lam1-0.15-10-logp1]
1.90s setup    tests/clv/test_plotting.py::test_plot_expected_purchases_ppc_exceptions
1.88s call     tests/clv/test_plotting.py::test_plot_expected_purchases_ppc[prior-10-100-None]
1.76s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_extract_predictive_covariates
1.60s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_fit_result_without_fit
1.52s call     tests/clv/models/test_modified_beta_geo.py::TestModifiedBetaGeoModel::test_numerically_stable_logp[0-0--0.41792826]
1.52s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value2-r2-10.58-s2-11.67-12]
1.48s call     tests/clv/test_distributions.py::TestModifiedBetaGeoNBD::test_invalid_value_logp
1.41s call     tests/clv/test_distributions.py::TestBetaGeoNBD::test_bg_nbd[value2-r2-10.58-a2-11.67-12]
1.38s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_beta_geo_beta_binom_sample_prior[None-None-gamma_size3-delta_size3-beta_geo_beta_binom_size3-expected_size3]
1.32s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModelIndividual::test_save_load
1.28s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_beta_geo_beta_binom_sample_prior[None-None-None-None-beta_geo_beta_binom_size4-expected_size4]
=========== 269 passed, 12 skipped, 40 warnings in 424.41s (0:07:04) ===========
(5 minutes 6 seconds) test (3.10, false, --ignore tests/mmm --ignore tests/clv)
============================= slowest 50 durations =============================
72.45s call     tests/test_mlflow.py::test_autolog_mmm
28.22s call     tests/test_mlflow.py::test_clv_fit_mcmc[BetaGeoModel]
19.01s call     tests/test_mlflow.py::test_clv_fit_map[BetaGeoModel]
14.07s call     tests/test_mlflow.py::test_file_system_uri_supported
13.38s call     tests/test_mlflow.py::test_autolog_pymc_model[nutpie]
9.25s call     tests/test_prior.py::test_censored_logp[scalar]
8.25s setup    tests/customer_choice/test_mv_its.py::test_MVITS_saturated[plot_fit]
6.51s setup    tests/test_prior.py::test_censored_normal_logp[-10]
3.57s call     tests/test_mlflow.py::test_autolog_pymc_model[numpyro]
3.36s call     tests/test_mlflow.py::test_autolog_pymc_model[blackjax]
3.22s call     tests/test_mlflow.py::test_log_with_data_in_likelihood
2.39s call     tests/customer_choice/test_mv_its.py::test_MVITS_unsaturated[plot_fit-unsaturated_model_bad]
2.35s call     tests/test_mlflow.py::test_log_data_no_data
1.39s call     tests/test_mlflow.py::test_logging_callback
1.12s call     tests/test_mlflow.py::test_autolog_pymc_model[pymc]
1.08s call     tests/test_prior.py::test_censored_logp[vector]
0.94s call     tests/test_model_builder.py::test_xarray_model_builder[DataArray]
0.72s setup    tests/test_model_builder.py::test_save_input_params
0.64s call     tests/test_plot.py::test_plot_functions[legend-different_axes-plot_hdi]
0.64s call     tests/test_prior.py::test_censored_sample_prior
0.58s call     tests/customer_choice/test_mv_its.py::test_save_load
0.56s call     tests/customer_choice/test_mv_its.py::test_MVITS_unsaturated[plot_fit-unsaturated_model_good]
0.56s call     tests/test_prior.py::test_custom_transform
0.36s call     tests/test_plot.py::test_plot_curve_custom_sel_to_string
0.36s call     tests/test_plot.py::test_plot_curve_custom_colors
0.36s call     tests/test_plot.py::test_plot_curve
0.33s call     tests/test_plot.py::test_plot_functions[no_legend-different_axes-plot_samples]
0.33s call     tests/test_plot.py::test_plot_functions[legend-different_axes-plot_samples]
0.32s call     tests/test_model_builder.py::test_fit_after_prior_keeps_prior
0.29s call     tests/test_model_builder.py::test_save_load
0.27s call     tests/customer_choice/test_mv_its.py::test_MVITS_unsaturated[plot_causal_impact_market_share-unsaturated_model_bad]
0.19s call     tests/test_plot.py::test_plot_curve_supply_axes_same_axes
0.19s call     tests/test_model_builder.py::test_second_fit
0.17s call     tests/test_plot.py::test_plot_functions[legend-same_axes-plot_samples]
0.17s call     tests/test_model_builder.py::test_sample_xxx_predictive_keeps_second[prior_predictive]
0.17s call     tests/test_model_builder.py::test_X_pred_prior_deprecation
0.16s call     tests/test_model_builder.py::test_fit_random_seed_reproducibility[int]
0.16s call     tests/test_plot.py::test_plot_functions[no_legend-same_axes-plot_samples]
0.16s call     tests/test_model_builder.py::test_fit_random_seed_reproducibility[rng]
0.16s call     tests/test_model_builder.py::test_fit_sampler_config_seed_reproducibility
0.14s call     tests/test_model_builder.py::test_sample_xxx_predictive_keeps_second[posterior_predictive]
0.12s call     tests/test_model_builder.py::test_fit
0.11s call     tests/test_model_builder.py::test_xarray_model_builder[Dataset]
0.10s call     tests/test_prior.py::test_create_variable
0.10s call     tests/test_model_builder.py::test_incorrect_set_idata_attrs_override
0.10s call     tests/test_model_builder.py::test_empty_sampler_config_fit
0.10s call     tests/test_model_builder.py::test_fit_no_t
0.10s call     tests/test_mlflow.py::test_log_mmm_evaluation_metrics
0.10s call     tests/test_model_builder.py::test_insufficient_attrs
0.09s call     tests/customer_choice/test_mv_its.py::test_MVITS_unsaturated[plot_counterfactual-unsaturated_model_bad]
================= 228 passed, 32 warnings in 206.70s (0:03:26) =================
(12 minutes 9 seconds) test (3.12, true, tests/mmm --ignore tests/mmm/test_tvp.py --ignore tests/mmm/test_budget_optimiz...
============================= slowest 50 durations =============================
22.42s call     tests/mmm/test_mmm.py::TestMMM::test_allocate_budget_to_maximize_response
18.00s call     tests/mmm/test_mmm.py::TestMMM::test_init[no_time_varying_media-time_varying_intercept-no_yearly_seasonality-single_channel-no_control-adstock_max_lag=1]
13.30s call     tests/mmm/test_lift_test.py::test_works_with_negative_delta
11.15s call     tests/mmm/test_multidimensional.py::test_fit[tvp intercept only, no fourier-Marginal model]
10.73s call     tests/mmm/components/test_adstock.py::test_apply[x1-channel-delayed]
10.66s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-delayed]
9.32s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-weibull_pdf]
8.96s call     tests/mmm/test_linear_trend.py::test_apply[with_intercept]
8.06s setup    tests/mmm/test_multidimensional.py::test_simple_fit
7.54s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[scalar-vector]
7.49s call     tests/mmm/test_multidimensional.py::test_fit[tvp intercept only, no fourier-County model]
7.25s call     tests/mmm/test_mmm.py::TestMMM::test_channel_contributions_forward_pass_is_consistent
6.42s call     tests/mmm/test_utility.py::test_mean_tightness_score
6.34s call     tests/mmm/test_linear_trend.py::test_apply_additional_dims[scalar]
6.11s call     tests/mmm/test_utility.py::test_conditional_value_at_risk
5.97s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[scalar-scalar]
4.65s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-no_yearly_seasonality-single_channel-no_control-adstock_max_lag=1]
4.27s call     tests/mmm/test_multidimensional.py::test_mmm_with_events
4.16s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[vector-scalar]
3.81s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-weibull_cdf]
3.66s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-geometric]
3.59s call     tests/mmm/test_fourier.py::test_sample_curve[yearly]
3.57s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-no_time_varying_intercept-no_yearly_seasonality-single_channel-no_control-adstock_max_lag=1]
3.54s call     tests/mmm/test_multidimensional.py::test_fit[tvp media only with fourier-Marginal model]
3.48s setup    tests/mmm/test_mmm.py::test_save_load_with_media_transformation
3.47s call     tests/mmm/test_multidimensional.py::test_fit[no tvps or fourier-Marginal model]
3.38s call     tests/mmm/components/test_base.py::test_transform_sample_curve_with_variable_factory
3.35s call     tests/mmm/test_mmm.py::test_save_load_with_tvp[True-True]
3.30s call     tests/mmm/test_multidimensional.py::test_fit[tvp media only with fourier-County model]
3.28s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-yearly_seasonality-single_channel-one_control-adstock_max_lag=1]
3.11s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-no_yearly_seasonality-single_channel-no_control-adstock_max_lag=4]
3.10s call     tests/mmm/test_utility.py::test_sharpe_ratio
3.09s call     tests/mmm/test_utility.py::test_covariance_matrix_matches_numpy[data0]
2.95s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-no_yearly_seasonality-single_channel-two_controls-adstock_max_lag=1]
2.88s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-no_yearly_seasonality-single_channel-one_control-adstock_max_lag=1]
2.87s call     tests/mmm/test_mmm.py::TestMMM::test_init[no_time_varying_media-no_time_varying_intercept-yearly_seasonality-single_channel-no_control-adstock_max_lag=1]
2.83s call     tests/mmm/test_multidimensional.py::test_fit[tvps and fourier-County model]
2.83s call     tests/mmm/test_multidimensional.py::test_fit[tvps and fourier-Marginal model]
2.68s setup    tests/mmm/test_mmm.py::test_plotting_media_transform_workflow[adstock]
2.59s call     tests/mmm/test_events.py::test_gaussian_basis_curve_sampling
2.55s call     tests/mmm/test_mmm.py::test_plotting_media_transform_workflow[adstock]
2.50s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-yearly_seasonality-single_channel-two_controls-adstock_max_lag=4]
2.48s call     tests/mmm/test_mmm.py::TestMMM::test_init[no_time_varying_media-time_varying_intercept-yearly_seasonality-single_channel-one_control-adstock_max_lag=1]
2.45s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-yearly_seasonality-multiple_channel-two_controls-adstock_max_lag=4]
2.39s call     tests/mmm/test_multidimensional.py::test_fit[no tvps with fourier-County model]
2.39s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-yearly_seasonality-multiple_channel-one_control-adstock_max_lag=1]
2.38s call     tests/mmm/test_mmm.py::test_save_load_with_tvp[True-False]
2.38s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-no_yearly_seasonality-multiple_channel-two_controls-adstock_max_lag=4]
2.32s call     tests/mmm/test_mmm.py::test_save_load_with_tvp[False-True]
2.32s call     tests/mmm/test_mmm.py::TestMMM::test_init[no_time_varying_media-time_varying_intercept-no_yearly_seasonality-single_channel-one_control-adstock_max_lag=1]
================ 1085 passed, 185 warnings in 611.68s (0:10:11) ================
(7 minutes 47 seconds) test (3.12, true, tests/mmm/test_tvp.py tests/mmm/test_budget_optimizer.py tests/mmm/test_hsgp.py...
============================= slowest 50 durations =============================
61.61s call     tests/mmm/test_tvp.py::test_time_varying_prior
32.62s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget[default_minimizer_kwargs]
26.23s call     tests/mmm/test_tvp.py::test_multidimensional
17.55s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_adstock_vectorized[PDF]
15.44s call     tests/mmm/test_hsgp.py::test_curve_workflow[HSGP]
13.25s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_custom_response_constraint[budget=10->resp=5]
9.27s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_adstock_vectorized[CDF]
9.00s call     tests/mmm/test_hsgp.py::test_curve_workflow[HSGPPeriodic]
8.23s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x0-1-1-4]
7.80s call     tests/mmm/test_hsgp.py::test_higher_dimension_hsgp
7.68s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[After]
6.45s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x2-0.7-1-100]
5.32s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_geometric_adstock_vectorized_logistic_saturation
4.56s call     tests/mmm/test_hsgp.py::test_hsgp_with_shared_data
4.36s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x2-0.7-1-100]
4.10s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[After]
3.90s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_parameterization_transformation[x0-20-0.5]
3.81s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x6-0.8-1-50]
3.59s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x0-1-1-4]
3.34s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x1-0.7-100]
3.32s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_infeasible_constraints[100-budget_bounds0-parameters0-custom_constraints0]
3.13s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x1-0.3-0.5-10]
3.12s call     tests/mmm/test_transformers.py::test_batched_convolution_broadcasting
2.92s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_x_zero[After]
2.78s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_parameterization_transformation[x1-100-0.5]
2.45s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[Overlap]
2.45s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_custom_response_constraint[budget=50->resp=10]
2.30s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x3-0.5-7]
2.22s call     tests/mmm/test_tvp.py::test_calling_without_default_args
2.21s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[Before]
2.17s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_bad_alpha[less_than_zero_1]
2.17s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_delayed_adstock_vectorized_logistic_saturation
2.14s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x5-0.8-50]
2.10s call     tests/mmm/test_transformers.py::test_batched_convolution[ndarray-0-After]
1.98s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x1-0.3-0.5-10]
1.95s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_baselined_range[x0-10-0.5-0.5]
1.94s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_hill_sigmoid_monotonicity[1-1-0]
1.89s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget[zero_total_budget]
1.84s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[Overlap]
1.71s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x0-0.5-1.0]
1.68s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x0-0.3-10]
1.68s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x1-0.2-19.0]
1.65s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_custom_minimize_args
1.64s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget[custom_minimizer_kwargs]
1.64s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x6-0.8-1-50]
1.63s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x2-0.6-5.0]
1.63s call     tests/mmm/test_transformers.py::test_batched_convolution[TensorVariable--1-After]
1.60s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[Before]
1.47s call     tests/mmm/test_hsgp.py::test_hsgp_class_method_ls_upper_changes_distribution
1.45s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_output_type[After]
================= 210 passed, 13 warnings in 349.15s (0:05:49) =================
(10 minutes 32 seconds) test (3.12, true, tests/clv)
============================= slowest 50 durations =============================
43.37s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_covariate_model_convergence
39.35s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[1]
24.04s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_expected_purchases[1]
22.89s call     tests/clv/test_utils.py::TestCustomerLifetimeValue::test_customer_lifetime_value_as_gg_method[fitted_pnbd]
22.32s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_posterior_distributions[map]
22.19s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_distribution_customer_churn_time
19.74s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_posterior_distributions[map]
16.40s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value3-0.3-p3-10-logp3]
14.61s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_logp
14.23s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_lifetimes[batch_shape1]
14.17s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_spend[True]
11.99s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[3]
11.95s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[6]
11.72s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[1]
10.99s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_expectation_method
10.89s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[6]
10.56s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[3]
10.50s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_save_load
10.11s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_distribution_new_customer
9.91s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_distribution_method
9.10s call     tests/clv/models/test_basic.py::TestCLVModel::test_fit_demz
7.94s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value4-0.55-10.58-0.61-beta4-12]
7.73s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_lifetimes[batch_shape0]
7.59s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_excel
7.31s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value0-0.4-0.15-10--8.39147106159807]
7.02s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_numerically_stable_logp[0-0--0.59947382]
6.55s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value1-lam1-0.15-10-logp1]
6.24s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_invalid_value_logp
4.33s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value1-r1-10.58-0.61-11.67-12]
2.81s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value2-lam2-0.15-10-logp2]
2.80s call     tests/clv/test_distributions.py::TestContContract::test_continuous_contractual[value3-0.3-p3-10-logp3]
2.68s call     tests/clv/models/test_basic.py::TestCLVModel::test_fit_mcmc
2.57s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_spend[False]
2.51s call     tests/clv/test_distributions.py::TestBetaGeoNBD::test_bg_nbd[value0-0.55-10.58-0.61-11.67-12]
2.32s call     tests/clv/test_distributions.py::TestBetaGeoNBD::test_bg_nbd[value3-0.55-11.67-0.61-10.58-T3]
2.31s setup    tests/clv/test_plotting.py::test_plot_expected_purchases_ppc_exceptions
2.27s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_posterior_distributions[mcmc]
2.19s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_extract_predictive_covariates
2.18s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_posterior_distributions[mcmc]
2.17s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value0-0.55-10.58-0.61-11.67-12]
2.11s call     tests/clv/test_plotting.py::test_plot_expected_purchases_ppc[prior-10-100-None]
1.94s call     tests/clv/models/test_modified_beta_geo.py::TestModifiedBetaGeoModel::test_numerically_stable_logp[0-0--0.41792826]
1.84s call     tests/clv/test_distributions.py::TestContContract::test_continuous_contractual[value1-lam1-0.15-10-logp1]
1.70s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value2-r2-10.58-s2-11.67-12]
1.67s call     tests/clv/test_distributions.py::TestModifiedBetaGeoNBD::test_invalid_value_logp
1.57s call     tests/clv/test_distributions.py::TestBetaGeoNBD::test_bg_nbd[value2-r2-10.58-a2-11.67-12]
1.52s call     tests/clv/test_utils.py::TestCustomerLifetimeValue::test_clv_after_thinning[fitted_pnbd]
1.49s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_beta_geo_beta_binom_sample_prior[None-None-gamma_size3-delta_size3-beta_geo_beta_binom_size3-expected_size3]
1.48s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModelIndividual::test_save_load
1.44s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_fit_result_without_fit
=========== 269 passed, 12 skipped, 41 warnings in 502.86s (0:08:22) ===========
(5 minutes 29 seconds) test (3.12, true, --ignore tests/mmm --ignore tests/clv)
============================= slowest 50 durations =============================
75.06s call     tests/test_mlflow.py::test_autolog_mmm
29.59s call     tests/test_mlflow.py::test_clv_fit_mcmc[BetaGeoModel]
21.38s call     tests/test_mlflow.py::test_clv_fit_map[BetaGeoModel]
14.42s call     tests/test_mlflow.py::test_file_system_uri_supported
13.56s call     tests/test_mlflow.py::test_autolog_pymc_model[nutpie]
9.33s call     tests/test_prior.py::test_censored_logp[scalar]
7.51s setup    tests/customer_choice/test_mv_its.py::test_MVITS_saturated[plot_fit]
6.09s setup    tests/test_prior.py::test_censored_normal_logp[-10]
4.93s call     tests/test_mlflow.py::test_autolog_pymc_model[numpyro]
3.93s call     tests/test_mlflow.py::test_autolog_pymc_model[blackjax]
3.67s call     tests/test_mlflow.py::test_log_with_data_in_likelihood
2.44s call     tests/test_mlflow.py::test_log_data_no_data
2.11s call     tests/customer_choice/test_mv_its.py::test_MVITS_unsaturated[plot_fit-unsaturated_model_bad]
1.54s call     tests/test_mlflow.py::test_logging_callback
1.23s call     tests/test_mlflow.py::test_autolog_pymc_model[pymc]
1.15s call     tests/test_prior.py::test_censored_logp[vector]
0.95s call     tests/test_model_builder.py::test_xarray_model_builder[DataArray]
0.85s call     tests/test_model_builder.py::test_fit_after_prior_keeps_prior
0.76s setup    tests/test_model_builder.py::test_save_input_params
0.68s call     tests/test_prior.py::test_censored_sample_prior
0.62s call     tests/customer_choice/test_mv_its.py::test_save_load
0.62s call     tests/customer_choice/test_mv_its.py::test_MVITS_unsaturated[plot_fit-unsaturated_model_good]
0.59s call     tests/test_prior.py::test_custom_transform
0.39s call     tests/test_plot.py::test_plot_curve
0.39s call     tests/test_plot.py::test_plot_curve_custom_sel_to_string
0.39s call     tests/test_plot.py::test_plot_curve_custom_colors
0.35s call     tests/test_plot.py::test_plot_functions[no_legend-different_axes-plot_samples]
0.35s call     tests/test_plot.py::test_plot_functions[legend-different_axes-plot_samples]
0.31s call     tests/test_model_builder.py::test_save_load
0.21s call     tests/test_plot.py::test_plot_curve_supply_axes_same_axes
0.20s call     tests/test_model_builder.py::test_second_fit
0.18s call     tests/test_plot.py::test_plot_functions[legend-same_axes-plot_samples]
0.18s call     tests/test_model_builder.py::test_X_pred_prior_deprecation
0.18s call     tests/test_model_builder.py::test_sample_xxx_predictive_keeps_second[prior_predictive]
0.18s call     tests/test_model_builder.py::test_fit_random_seed_reproducibility[int]
0.18s call     tests/test_model_builder.py::test_fit_random_seed_reproducibility[rng]
0.17s call     tests/test_model_builder.py::test_fit_sampler_config_seed_reproducibility
0.17s call     tests/test_plot.py::test_plot_functions[no_legend-same_axes-plot_samples]
0.14s call     tests/test_model_builder.py::test_sample_xxx_predictive_keeps_second[posterior_predictive]
0.13s call     tests/test_model_builder.py::test_fit
0.13s call     tests/customer_choice/test_mv_its.py::test_MVITS_saturated[plot_fit]
0.12s call     tests/test_model_builder.py::test_xarray_model_builder[Dataset]
0.12s call     tests/test_prior.py::test_create_variable
0.11s call     tests/test_model_builder.py::test_fit_no_t
0.11s call     tests/test_model_builder.py::test_empty_sampler_config_fit
0.11s call     tests/test_model_builder.py::test_insufficient_attrs
0.11s call     tests/test_model_builder.py::test_incorrect_set_idata_attrs_override
0.10s call     tests/test_prior.py::test_create_variable_multiple_times
0.10s call     tests/customer_choice/test_mv_its.py::test_MVITS_unsaturated[plot_counterfactual-unsaturated_model_bad]
0.10s call     tests/customer_choice/test_mv_its.py::test_MVITS_unsaturated[plot_counterfactual-unsaturated_model_good]
================= 228 passed, 35 warnings in 215.27s (0:03:35) =================

You can find more information on how to contribute here.

Automatically generated by GitHub Action
Latest run date: 2025-02-10

@wd60622 wd60622 changed the title Address slowest tests Slowest test times (Last Updated: 2024-11-03) Nov 4, 2024
@wd60622 wd60622 changed the title Slowest test times (Last Updated: 2024-11-03) Slowest test times Nov 4, 2024
@wd60622 wd60622 pinned this issue Nov 4, 2024
@wd60622 wd60622 changed the title Slowest test times Speed up test times 🚀 Nov 4, 2024
@wd60622 wd60622 added the help wanted Extra attention is needed label Nov 5, 2024
@ColtAllen
Copy link
Collaborator

Related to #1227

@ricardoV94
Copy link
Contributor

@wd60622 can we have this automated issue on pymc/pytensor 🤤 ?

@wd60622
Copy link
Contributor Author

wd60622 commented Dec 13, 2024

@wd60622 can we have this automated issue on pymc/pytensor 🤤 ?

Yes

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

3 participants