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 19 seconds) test_slow (tests/clv/models/test_beta_geo_beta_binom.py)
============================= slowest 50 durations =============================
695.10s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_model_convergence[mcmc-0.3]
17.27s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_model_convergence[map-0.2]
0.05s 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 714.06s (0:11:54) =============
(10 minutes 9 seconds) test (3.10, false, tests/mmm --ignore tests/mmm/test_tvp.py --ignore tests/mmm/test_budget_optimi...
============================= slowest 50 durations =============================
27.77s call     tests/mmm/test_mmm.py::TestMMM::test_allocate_budget_to_maximize_response
19.29s 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]
17.66s call     tests/mmm/test_lift_test.py::test_works_with_negative_delta
13.90s call     tests/mmm/test_base.py::test_calling_fit_result_before_fit_raises_error
10.32s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-delayed]
9.85s call     tests/mmm/components/test_adstock.py::test_apply[x1-channel-delayed]
9.16s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-weibull_pdf]
7.92s call     tests/mmm/test_mmm.py::TestMMM::test_channel_contributions_forward_pass_is_consistent
7.82s call     tests/mmm/test_linear_trend.py::test_apply[with_intercept]
7.71s call     tests/mmm/test_utility.py::test_mean_tightness_score
7.49s call     tests/mmm/test_linear_trend.py::test_apply_additional_dims[scalar]
7.17s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[scalar-vector]
6.59s call     tests/mmm/test_utility.py::test_conditional_value_at_risk
6.24s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[scalar-scalar]
6.07s 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.55s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[vector-scalar]
4.16s call     tests/mmm/test_mmm.py::test_save_load_with_tvp[True-True]
4.01s call     tests/mmm/components/test_adstock.py::test_adstock_sample_curve[weibull_pdf]
3.81s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-weibull_cdf]
3.80s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-geometric]
3.80s call     tests/mmm/test_fourier.py::test_sample_curve
3.68s setup    tests/mmm/test_mmm.py::test_save_load_with_media_transformation
3.41s 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.33s call     tests/mmm/test_utility.py::test_covariance_matrix_matches_numpy[data0]
3.29s call     tests/mmm/test_mmm.py::test_save_load_with_tvp[True-False]
3.03s call     tests/mmm/test_mmm.py::test_save_load_with_tvp[False-True]
2.87s call     tests/mmm/test_mmm.py::test_channel_contributions_forward_pass_time_varying_media
2.86s 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]
2.80s call     tests/mmm/test_mmm.py::TestMMM::test_save_load_with_not_serializable_model_config
2.73s call     tests/mmm/test_mmm.py::test_plotting_media_transform_workflow[adstock]
2.70s call     tests/mmm/test_utility.py::test_sharpe_ratio
2.66s 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.58s 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.58s 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=4]
2.43s setup    tests/mmm/test_mmm.py::test_plotting_media_transform_workflow[adstock]
2.41s call     tests/mmm/test_fourier.py::test_plot_curve
2.40s call     tests/mmm/test_fourier.py::test_fourier_modes_shape[periods2-1-expected_shape2]
2.39s call     tests/mmm/test_utility.py::test_tail_distance[100-30-100-50-greater]
2.37s 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.27s 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.20s 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.16s call     tests/mmm/test_linear_trend.py::test_apply_additional_dims[1d]
2.16s call     tests/mmm/components/test_adstock.py::test_apply[x1-channel-weibull_pdf]
2.13s call     tests/mmm/test_fourier.py::test_additional_dimension
2.10s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-no_time_varying_intercept-yearly_seasonality-single_channel-one_control-adstock_max_lag=1]
2.01s call     tests/mmm/components/test_saturation.py::test_apply_method[x0-None-logistic]
1.98s 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]
1.97s 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]
1.96s 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]
1.96s 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]
================= 815 passed, 20 warnings in 523.44s (0:08:43) =================
(8 minutes 42 seconds) test_slow (tests/clv/models/test_pareto_nbd.py)
============================= slowest 50 durations =============================
403.93s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_model_convergence[mcmc-0.1]
22.81s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_model_convergence[map-0.2]
10.51s 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, 4001 warnings in 438.95s (0:07:18) ===========
(6 minutes 54 seconds) test (3.10, false, tests/mmm/test_tvp.py tests/mmm/test_budget_optimizer.py tests/mmm/test_hsgp.p...
============================= slowest 50 durations =============================
67.78s call     tests/mmm/test_tvp.py::test_time_varying_prior
34.08s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget[default_minimizer_kwargs]
31.31s call     tests/mmm/test_tvp.py::test_multidimensional
16.70s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_adstock_vectorized[PDF]
13.82s call     tests/mmm/test_hsgp.py::test_curve_workflow[HSGP]
12.64s call     tests/mmm/test_hsgp.py::test_curve_workflow[HSGPPeriodic]
9.09s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[After]
8.62s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_adstock_vectorized[CDF]
8.60s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x0-1-1-4]
6.52s call     tests/mmm/test_hsgp.py::test_higher_dimension_hsgp
6.09s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x2-0.7-1-100]
5.39s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_geometric_adstock_vectorized_logistic_saturation
3.94s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[After]
3.76s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x1-0.3-0.5-10]
3.72s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x0-1-1-4]
3.49s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_x_zero[After]
3.47s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x6-0.8-1-50]
3.43s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_parameterization_transformation[x0-20-0.5]
3.41s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x2-0.7-1-100]
3.02s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x1-0.7-100]
2.85s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_parameterization_transformation[x1-100-0.5]
2.40s call     tests/mmm/test_transformers.py::test_batched_convolution[ndarray-0-After]
2.22s call     tests/mmm/test_tvp.py::test_calling_without_default_args
2.12s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[Overlap]
2.00s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_hill_sigmoid_monotonicity[1-1-0]
2.00s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x5-0.8-50]
1.91s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[Before]
1.90s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x0-0.5-1.0]
1.75s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x1-0.3-0.5-10]
1.70s call     tests/mmm/test_transformers.py::test_batched_convolution[TensorVariable--1-After]
1.67s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_inverse_scaled_logistic_saturation_min_max_value
1.67s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x1-0.2-19.0]
1.66s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_bad_alpha[less_than_zero_1]
1.66s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget[custom_minimizer_kwargs]
1.65s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_delayed_adstock_vectorized_logistic_saturation
1.65s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x6-0.8-1-50]
1.64s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_custom_minimize_args
1.62s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x3-0.5-7]
1.58s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x0-0.3-10]
1.57s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[Before]
1.48s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_output_type[After]
1.47s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_infeasible_constraints[100-budget_bounds0-parameters0-custom_constraints0]
1.43s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_zero_total[0-budget_bounds0-parameters0-expected_optimal0-2.38e-10]
1.41s call     tests/mmm/test_hsgp.py::test_hsgp_class_method_ls_upper_changes_distribution
1.37s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[Overlap]
1.37s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_baselined_range[x2-10-0.001-0.01]
1.32s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_hill_sigmoid_sigma_upper_bound[5-2-0.5-1]
1.30s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x4-0.001-0.01]
1.23s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x4-0.5-0.8-7]
1.09s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x3-0.2-0.2-5]
================= 198 passed, 3 warnings in 330.02s (0:05:30) ==================
(4 minutes 39 seconds) test_slow (--ignore tests/clv/models/test_beta_geo_beta_binom.py --ignore tests/clv/models/test_p...
============================= slowest 50 durations =============================
53.88s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_model_convergence[mcmc-0.1]
48.33s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_model_convergence
43.54s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModelIndividual::test_model_convergence
34.92s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_model_convergence
14.20s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_model_convergence[map-0.2]
0.69s setup    tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_model_convergence
0.32s setup    tests/mmm/test_plotting.py::TestBasePlotting::test_plots[without_controls-target_transform-seed_0-plot_prior_predictive-kwargs_plot8]
0.04s setup    tests/clv/models/test_gamma_gamma.py::TestGammaGammaModelIndividual::test_model_convergence
0.01s setup    tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_model_convergence[mcmc-0.1]

0.005s hidden.  Use -vv to show these durations.)
=========== 5 passed, 1346 skipped, 17 warnings in 202.47s (0:03:22) ===========
(8 minutes 25 seconds) test (3.10, false, tests/clv)
============================= slowest 50 durations =============================
42.34s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[1]
40.93s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_covariate_model_convergence
25.19s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_expected_purchases[1]
23.69s call     tests/clv/test_utils.py::TestCustomerLifetimeValue::test_customer_lifetime_value_as_gg_method[fitted_pnbd]
20.05s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_distribution_customer_churn_time
17.59s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_posterior_distributions[map]
15.45s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_spend[True]
14.23s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_logp
12.88s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_lifetimes[batch_shape0]
12.74s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value3-0.3-p3-10-logp3]
11.83s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_lifetimes[batch_shape1]
10.81s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_expectation_method
9.71s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value4-0.55-10.58-0.61-beta4-12]
9.43s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_distribution_method
9.32s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_distribution_new_customer
8.88s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_excel
8.50s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_save_load
8.06s call     tests/clv/models/test_basic.py::TestCLVModel::test_fit_demz
7.02s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_invalid_value_logp
6.32s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value0-0.4-0.15-10--8.39147106159807]
5.68s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[6]
5.64s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[1]
5.62s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[3]
5.48s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value1-lam1-0.15-10-logp1]
4.65s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value2-lam2-0.15-10-logp2]
4.61s call     tests/clv/test_distributions.py::TestContContract::test_continuous_contractual[value3-0.3-p3-10-logp3]
4.59s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[6]
4.57s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[3]
4.42s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_spend[False]
4.39s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value1-r1-10.58-0.61-11.67-12]
3.96s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_numerically_stable_logp[0-0--0.59947382]
3.48s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_distribution_new_customer
2.44s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value3-0.55-11.67-0.61-10.58-T3]
2.25s call     tests/clv/test_distributions.py::TestContContract::test_continuous_contractual[value1-lam1-0.15-10-logp1]
2.24s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value0-0.55-10.58-0.61-11.67-12]
1.95s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value2-r2-10.58-s2-11.67-12]
1.90s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_posterior_distributions[mcmc]
1.83s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_extract_predictive_covariates
1.82s call     tests/clv/test_plotting.py::test_plot_expected_purchases_ppc[prior-10-100-None]
1.44s call     tests/clv/test_distributions.py::TestContContract::test_continuous_contractual[value0-0.3-0.15-10--10.45705972]
1.40s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModelIndividual::test_save_load
1.31s 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.30s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_fit_result_without_fit
1.26s 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]
1.22s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_numerically_stable_logp[200-38-100.7957]
1.06s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_expected_probability_alive
1.04s call     tests/clv/test_utils.py::TestCustomerLifetimeValue::test_clv_after_thinning[fitted_pnbd]
0.99s call     tests/clv/test_plotting.py::test_plot_expected_purchases_ppc[posterior-20-50-subplot1]
0.97s setup    tests/clv/test_plotting.py::test_plot_expected_purchases_ppc_exceptions
0.78s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_expected_purchase_probability[2-2]
=========== 222 passed, 10 skipped, 10 warnings in 421.64s (0:07:01) ===========
(4 minutes 30 seconds) test (3.10, false, --ignore tests/mmm --ignore tests/clv)
============================= slowest 50 durations =============================
82.72s call     tests/test_mlflow.py::test_autolog_mmm
17.15s call     tests/test_mlflow.py::test_file_system_uri_supported
10.80s call     tests/test_model_builder.py::test_second_fit
9.86s call     tests/test_mlflow.py::test_autolog_pymc_model[nutpie]
9.21s setup    tests/test_model_builder.py::test_save_input_params
7.78s call     tests/test_model_builder.py::test_fit_random_seed_reproducibility[rng]
6.38s call     tests/test_model_builder.py::test_fit_sampler_config_seed_reproducibility
5.99s call     tests/test_model_builder.py::test_fit_random_seed_reproducibility[int]
5.80s call     tests/test_model_builder.py::test_fit_after_prior_keeps_prior
4.92s call     tests/test_model_builder.py::test_empty_sampler_config_fit
4.70s call     tests/test_mlflow.py::test_log_data_no_data
3.66s call     tests/test_mlflow.py::test_autolog_pymc_model[blackjax]
3.59s call     tests/test_mlflow.py::test_autolog_pymc_model[numpyro]
3.28s call     tests/test_model_builder.py::test_fit_no_t
2.62s call     tests/test_mlflow.py::test_autolog_pymc_model[pymc]
1.60s call     tests/test_model_builder.py::test_save_load
0.67s setup    tests/test_mlflow.py::test_multi_likelihood_type
0.66s call     tests/test_prior.py::test_custom_transform_comes_first
0.64s call     tests/test_prior.py::test_custom_transform
0.31s call     tests/test_plot.py::test_plot_curve
0.30s call     tests/test_plot.py::test_plot_curve_custom_sel_to_string
0.30s call     tests/test_plot.py::test_plot_curve_custom_colors
0.27s call     tests/test_plot.py::test_plot_functions[no_legend-different_axes-plot_samples]
0.27s call     tests/test_plot.py::test_plot_functions[legend-different_axes-plot_samples]
0.15s call     tests/test_model_builder.py::test_sample_xxx_predictive_keeps_second[prior_predictive]
0.14s call     tests/test_plot.py::test_plot_curve_supply_axes_same_axes
0.14s call     tests/test_plot.py::test_plot_functions[legend-same_axes-plot_samples]
0.12s call     tests/test_plot.py::test_plot_functions[no_legend-same_axes-plot_samples]
0.12s call     tests/test_model_builder.py::test_fit
0.11s call     tests/test_model_builder.py::test_sample_xxx_predictive_keeps_second[posterior_predictive]
0.10s call     tests/test_model_builder.py::test_insufficient_attrs
0.10s call     tests/test_model_builder.py::test_incorrect_set_idata_attrs_override
0.08s call     tests/test_prior.py::test_create_variable
0.08s call     tests/test_model_builder.py::test_fit_sampler_config_with_rng_fails
0.08s call     tests/test_plot.py::test_plot_functions[no_legend-different_axes-plot_hdi]
0.08s call     tests/test_plot.py::test_plot_functions[legend-different_axes-plot_hdi]
0.07s call     tests/test_prior.py::test_create_variable_multiple_times
0.06s call     tests/test_model_builder.py::test_graphviz
0.06s call     tests/test_model_builder.py::test_predict
0.06s call     tests/test_model_builder.py::test_sample_posterior_predictive[True]
0.05s call     tests/test_model_builder.py::test_prediction_kwarg
0.05s call     tests/test_model_builder.py::test_sample_posterior_predictive[False]
0.05s call     tests/test_prior.py::test_sample_prior
0.05s call     tests/test_mlflow.py::test_log_model_graph_no_graphviz[render_error]
0.04s call     tests/test_mlflow.py::test_multi_likelihood_type
0.04s call     tests/test_plot.py::test_plot_functions[legend-same_axes-plot_hdi]
0.03s call     tests/test_plot.py::test_plot_functions[no_legend-same_axes-plot_hdi]
0.03s call     tests/test_prior.py::test_to_graph
0.03s call     tests/test_mlflow.py::test_log_model_graph_no_graphviz[no_graphviz]
0.02s setup    tests/test_mlflow.py::test_file_system_uri_supported
================= 149 passed, 21 warnings in 189.25s (0:03:09) =================
(12 minutes 5 seconds) test (3.12, true, tests/mmm --ignore tests/mmm/test_tvp.py --ignore tests/mmm/test_budget_optimiz...
============================= slowest 50 durations =============================
30.07s call     tests/mmm/test_mmm.py::TestMMM::test_allocate_budget_to_maximize_response
21.25s 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]
19.62s call     tests/mmm/test_base.py::test_calling_fit_result_before_fit_raises_error
18.93s call     tests/mmm/test_lift_test.py::test_works_with_negative_delta
11.63s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-delayed]
10.34s call     tests/mmm/components/test_adstock.py::test_apply[x1-channel-delayed]
9.98s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-weibull_pdf]
8.12s call     tests/mmm/test_linear_trend.py::test_apply[with_intercept]
8.00s call     tests/mmm/test_utility.py::test_mean_tightness_score
7.99s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[scalar-vector]
7.91s call     tests/mmm/test_linear_trend.py::test_apply_additional_dims[scalar]
7.51s call     tests/mmm/test_mmm.py::TestMMM::test_channel_contributions_forward_pass_is_consistent
6.88s call     tests/mmm/test_utility.py::test_conditional_value_at_risk
6.78s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[scalar-scalar]
6.39s 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.79s call     tests/mmm/test_mmm.py::test_save_load_with_tvp[True-True]
4.69s call     tests/mmm/test_media_transformation.py::test_apply_media_transformation[vector-scalar]
4.51s call     tests/mmm/components/test_adstock.py::test_adstock_sample_curve[weibull_pdf]
4.17s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-weibull_cdf]
4.15s call     tests/mmm/components/test_adstock.py::test_apply[x0-None-geometric]
4.03s call     tests/mmm/test_mmm.py::test_save_load_with_tvp[True-False]
4.02s setup    tests/mmm/test_mmm.py::test_save_load_with_media_transformation
4.00s call     tests/mmm/test_fourier.py::test_sample_curve
3.93s 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.72s call     tests/mmm/test_mmm.py::test_save_load_with_tvp[False-True]
3.52s call     tests/mmm/test_mmm.py::test_channel_contributions_forward_pass_time_varying_media
3.42s call     tests/mmm/test_utility.py::test_covariance_matrix_matches_numpy[data0]
3.30s call     tests/mmm/test_mmm.py::TestMMM::test_save_load_with_not_serializable_model_config
3.23s 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.07s call     tests/mmm/test_mmm.py::test_plotting_media_transform_workflow[adstock]
2.95s 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=4]
2.89s 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.86s call     tests/mmm/test_utility.py::test_sharpe_ratio
2.81s 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.68s setup    tests/mmm/test_mmm.py::test_plotting_media_transform_workflow[adstock]
2.58s 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.55s 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.52s call     tests/mmm/test_fourier.py::test_plot_curve
2.48s call     tests/mmm/components/test_adstock.py::test_apply[x1-channel-weibull_pdf]
2.46s call     tests/mmm/test_fourier.py::test_fourier_modes_shape[periods2-1-expected_shape2]
2.45s 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=4]
2.37s 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]
2.35s 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.34s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-no_time_varying_intercept-yearly_seasonality-single_channel-one_control-adstock_max_lag=1]
2.30s call     tests/mmm/test_utility.py::test_tail_distance[100-30-100-50-greater]
2.27s 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.26s 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.22s call     tests/mmm/test_mmm.py::test_missing_attrs_to_defaults
2.20s call     tests/mmm/test_mmm.py::test_new_spend_contributions_prior[True]
2.20s call     tests/mmm/test_mmm.py::TestMMM::test_init[time_varying_media-time_varying_intercept-no_yearly_seasonality-multiple_channel-one_control-adstock_max_lag=4]
================= 815 passed, 25 warnings in 613.18s (0:10:13) =================
(8 minutes 2 seconds) test (3.12, true, tests/mmm/test_tvp.py tests/mmm/test_budget_optimizer.py tests/mmm/test_hsgp.py...
============================= slowest 50 durations =============================
75.62s call     tests/mmm/test_tvp.py::test_time_varying_prior
37.25s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget[default_minimizer_kwargs]
34.46s call     tests/mmm/test_tvp.py::test_multidimensional
18.15s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_adstock_vectorized[PDF]
15.24s call     tests/mmm/test_hsgp.py::test_curve_workflow[HSGP]
13.61s call     tests/mmm/test_hsgp.py::test_curve_workflow[HSGPPeriodic]
10.15s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[After]
9.94s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_adstock_vectorized[CDF]
9.13s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x0-1-1-4]
6.96s call     tests/mmm/test_hsgp.py::test_higher_dimension_hsgp
6.71s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x2-0.7-1-100]
6.14s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_geometric_adstock_vectorized_logistic_saturation
4.55s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[After]
4.20s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x1-0.3-0.5-10]
4.15s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x6-0.8-1-50]
4.05s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x2-0.7-1-100]
3.96s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x0-1-1-4]
3.92s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_parameterization_transformation[x0-20-0.5]
3.89s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_x_zero[After]
3.74s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x1-0.7-100]
3.17s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_parameterization_transformation[x1-100-0.5]
2.66s call     tests/mmm/test_tvp.py::test_calling_without_default_args
2.66s call     tests/mmm/test_transformers.py::test_batched_convolution[ndarray-0-After]
2.54s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[Overlap]
2.30s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x5-0.8-50]
2.27s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_vectorized[Before]
2.19s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_hill_sigmoid_monotonicity[1-1-0]
2.04s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget[custom_minimizer_kwargs]
2.04s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_infeasible_constraints[100-budget_bounds0-parameters0-custom_constraints0]
2.04s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_delayed_adstock_vectorized_logistic_saturation
2.03s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_bad_alpha[less_than_zero_1]
1.99s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_cdf_adsotck[x1-0.3-0.5-10]
1.97s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x0-0.5-1.0]
1.93s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[Before]
1.92s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x6-0.8-1-50]
1.92s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x1-0.2-19.0]
1.86s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_inverse_scaled_logistic_saturation_min_max_value
1.83s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_zero_total[0-budget_bounds0-parameters0-expected_optimal0-2.38e-10]
1.82s call     tests/mmm/test_transformers.py::test_batched_convolution[TensorVariable--1-After]
1.81s call     tests/mmm/test_budget_optimizer.py::test_allocate_budget_custom_minimize_args
1.78s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x0-0.3-10]
1.78s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_geometric_adstock_good_alpha[x3-0.5-7]
1.72s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_vectorized[Overlap]
1.65s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_delayed_adstock_output_type[After]
1.53s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_tanh_saturation_baselined_range[x2-10-0.001-0.01]
1.48s call     tests/mmm/test_hsgp.py::test_hsgp_class_method_ls_upper_changes_distribution
1.45s call     tests/mmm/test_transformers.py::TestSaturationTransformers::test_hill_sigmoid_sigma_upper_bound[5-2-0.5-1]
1.40s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x2-0.6-5.0]
1.40s call     tests/mmm/test_transformers.py::TestsAdstockTransformers::test_weibull_pdf_adstock[x4-0.5-0.8-7]
1.33s call     tests/mmm/test_transformers.py::TestTransformersComposition::test_logistic_saturation_geometric_adstock_composition[x4-0.001-0.01]
================= 198 passed, 4 warnings in 369.72s (0:06:09) ==================
(10 minutes 19 seconds) test (3.12, true, tests/clv)
============================= slowest 50 durations =============================
53.82s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[1]
43.18s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_covariate_model_convergence
26.01s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_expected_purchases[1]
24.64s call     tests/clv/test_utils.py::TestCustomerLifetimeValue::test_customer_lifetime_value_as_gg_method[fitted_pnbd]
21.99s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_distribution_customer_churn_time
19.85s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_posterior_distributions[map]
17.95s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_spend[True]
14.71s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_logp
13.33s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_lifetimes[batch_shape0]
13.08s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[6]
13.07s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[3]
13.06s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_probability_alive[1]
12.66s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_lifetimes[batch_shape1]
11.93s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value3-0.3-p3-10-logp3]
11.81s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[3]
11.77s call     tests/clv/models/test_beta_geo_beta_binom.py::TestBetaGeoBetaBinomModel::test_expected_purchases[6]
11.60s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_expectation_method
10.70s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_distribution_method
10.36s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_distribution_new_customer
9.81s call     tests/clv/models/test_shifted_beta_geo.py::TestShiftedBetaGeoModel::test_save_load
9.36s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value4-0.55-10.58-0.61-beta4-12]
9.28s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_logp_matches_excel
8.63s call     tests/clv/models/test_basic.py::TestCLVModel::test_fit_demz
8.07s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_invalid_value_logp
6.53s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value0-0.4-0.15-10--8.39147106159807]
6.34s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModel::test_spend[False]
5.70s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value1-lam1-0.15-10-logp1]
4.99s call     tests/clv/test_distributions.py::TestContNonContract::test_continuous_non_contractual[value2-lam2-0.15-10-logp2]
4.64s call     tests/clv/test_distributions.py::TestContContract::test_continuous_contractual[value3-0.3-p3-10-logp3]
4.19s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_numerically_stable_logp[0-0--0.59947382]
3.89s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_distribution_new_customer
3.89s 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::TestParetoNBD::test_pareto_nbd[value3-0.55-11.67-0.61-10.58-T3]
2.33s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value0-0.55-10.58-0.61-11.67-12]
2.33s call     tests/clv/test_distributions.py::TestContContract::test_continuous_contractual[value1-lam1-0.15-10-logp1]
2.12s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_posterior_distributions[mcmc]
2.09s call     tests/clv/test_plotting.py::test_plot_expected_purchases_ppc[prior-10-100-None]
1.99s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModelWithCovariates::test_extract_predictive_covariates
1.76s call     tests/clv/test_distributions.py::TestParetoNBD::test_pareto_nbd[value2-r2-10.58-s2-11.67-12]
1.60s 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.60s call     tests/clv/test_distributions.py::TestContContract::test_continuous_contractual[value0-0.3-0.15-10--10.45705972]
1.56s 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]
1.50s call     tests/clv/test_utils.py::TestCustomerLifetimeValue::test_clv_after_thinning[fitted_pnbd]
1.45s call     tests/clv/models/test_gamma_gamma.py::TestGammaGammaModelIndividual::test_save_load
1.27s call     tests/clv/test_plotting.py::test_plot_expected_purchases_ppc[posterior-20-50-subplot1]
1.16s call     tests/clv/models/test_beta_geo.py::TestBetaGeoModel::test_numerically_stable_logp[200-38-100.7957]
1.10s setup    tests/clv/test_plotting.py::test_plot_expected_purchases_ppc_exceptions
1.00s call     tests/clv/models/test_pareto_nbd.py::TestParetoNBDModel::test_expected_probability_alive
0.92s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_beta_geo_beta_binom_sample_prior[alpha_size1-None-None-None-None-expected_size1]
0.91s call     tests/clv/test_distributions.py::TestBetaGeoBetaBinom::test_beta_geo_beta_binom_sample_prior[None-beta_size2-None-None-beta_geo_beta_binom_size2-expected_size2]
=========== 222 passed, 10 skipped, 11 warnings in 495.74s (0:08:15) ===========
(5 minutes 45 seconds) test (3.12, true, --ignore tests/mmm --ignore tests/clv)
============================= slowest 50 durations =============================
91.16s call     tests/test_mlflow.py::test_autolog_mmm
18.75s call     tests/test_mlflow.py::test_file_system_uri_supported
14.73s call     tests/test_model_builder.py::test_second_fit
14.23s call     tests/test_mlflow.py::test_autolog_pymc_model[nutpie]
12.01s call     tests/test_model_builder.py::test_fit_random_seed_reproducibility[rng]
11.14s setup    tests/test_model_builder.py::test_save_input_params
10.12s call     tests/test_model_builder.py::test_empty_sampler_config_fit
9.07s call     tests/test_model_builder.py::test_fit_random_seed_reproducibility[int]
8.70s call     tests/test_model_builder.py::test_fit_sampler_config_seed_reproducibility
8.26s call     tests/test_model_builder.py::test_fit_no_t
6.54s call     tests/test_model_builder.py::test_fit_after_prior_keeps_prior
5.37s call     tests/test_mlflow.py::test_log_data_no_data
4.67s call     tests/test_mlflow.py::test_autolog_pymc_model[blackjax]
4.64s call     tests/test_mlflow.py::test_autolog_pymc_model[numpyro]
2.79s call     tests/test_mlflow.py::test_autolog_pymc_model[pymc]
1.79s call     tests/test_model_builder.py::test_save_load
0.83s call     tests/test_plot.py::test_plot_curve
0.80s call     tests/test_model_builder.py::test_sample_xxx_predictive_keeps_second[prior_predictive]
0.71s call     tests/test_prior.py::test_custom_transform_comes_first
0.70s setup    tests/test_mlflow.py::test_multi_likelihood_type
0.66s call     tests/test_prior.py::test_custom_transform
0.39s call     tests/test_plot.py::test_plot_curve_custom_colors
0.39s call     tests/test_plot.py::test_plot_curve_custom_sel_to_string
0.36s call     tests/test_plot.py::test_plot_functions[legend-different_axes-plot_samples]
0.36s call     tests/test_plot.py::test_plot_functions[no_legend-different_axes-plot_samples]
0.21s call     tests/test_plot.py::test_plot_curve_supply_axes_same_axes
0.21s call     tests/test_plot.py::test_plot_functions[legend-same_axes-plot_samples]
0.18s call     tests/test_model_builder.py::test_incorrect_set_idata_attrs_override
0.18s call     tests/test_model_builder.py::test_insufficient_attrs
0.17s call     tests/test_plot.py::test_plot_functions[no_legend-same_axes-plot_samples]
0.17s call     tests/test_model_builder.py::test_fit
0.16s call     tests/test_model_builder.py::test_sample_xxx_predictive_keeps_second[posterior_predictive]
0.10s call     tests/test_model_builder.py::test_fit_sampler_config_with_rng_fails
0.09s call     tests/test_prior.py::test_create_variable
0.09s call     tests/test_plot.py::test_plot_functions[legend-different_axes-plot_hdi]
0.09s call     tests/test_plot.py::test_plot_functions[no_legend-different_axes-plot_hdi]
0.09s call     tests/test_model_builder.py::test_predict
0.09s call     tests/test_model_builder.py::test_sample_posterior_predictive[True]
0.08s call     tests/test_prior.py::test_create_variable_multiple_times
0.08s call     tests/test_model_builder.py::test_prediction_kwarg
0.08s call     tests/test_model_builder.py::test_sample_posterior_predictive[False]
0.07s call     tests/test_model_builder.py::test_graphviz
0.06s call     tests/test_prior.py::test_sample_prior
0.06s call     tests/test_mlflow.py::test_log_model_graph_no_graphviz[render_error]
0.05s call     tests/test_mlflow.py::test_multi_likelihood_type
0.05s call     tests/test_plot.py::test_plot_functions[legend-same_axes-plot_hdi]
0.04s call     tests/test_plot.py::test_plot_functions[no_legend-same_axes-plot_hdi]
0.04s call     tests/test_prior.py::test_to_graph
0.03s call     tests/test_mlflow.py::test_log_model_graph_no_graphviz[no_graphviz]
0.02s setup    tests/test_mlflow.py::test_file_system_uri_supported
================= 149 passed, 24 warnings in 236.00s (0:03:56) =================

You can find more information on how to contribute here.

Automatically generated by GitHub Action
Latest run date: 2024-12-24

@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