BetaGeoBetaBinomModel Testing Speedup

[ColtAllen]

Description

The slow test test_model_convergence for the recently added BetaGeoBetaBinomModel takes over 90 minutes in the CI for the following reasons:

• The discrete distributions in this model require NUTS to default to compound sampling
• The test dataset is rather large (11k rows)

Merging #707 will make MCMC sampling 3x faster, and although the full test dataset is still required elsewhere so that tests match research results and lifetimes outputs, for this particular slow test we can get away with a smaller dataset.

test_model_convergence is now being ran on a dataframe of 1k rows sampled from the full dataset. I had to bump up the rtol from 0.1 to 0.3 so that it passes, but added a comment the test did previously pass with the full dataset.

Related Issue

• Related to Replace scan in BetaGeoBetaBinom #707

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Modules affected

• MMM
• CLV

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify): Tests

---

📚 Documentation preview 📚: https://pymc-marketing--1039.org.readthedocs.build/en/1039/

[juanitorduz]

Thanks @ColtAllen !