Mute `Sampling: []` stdout when calling `get_channel_contributions_forward_pass_grid` - Githubissues

pymc-labs / pymc-marketing

Bayesian marketing toolbox in PyMC. Media Mix (MMM), customer lifetime value (CLV), buy-till-you-die (BTYD) models and more.

https://www.pymc-marketing.io/

Apache License 2.0

705 stars 198 forks source link

Mute `Sampling: []` stdout when calling `get_channel_contributions_forward_pass_grid` #1043

Closed PabloRoque closed 1 month ago

PabloRoque commented 1 month ago

Mutes repetitive Sampling: [] message displayed when calling plot_channel_contributions_grid

Description

Adds disable_logger_stdout flag, an Optional[bool], to channel_contributions_forward_pass.
When True, it disables the pymc.sampling.forward logger propagation [as suggested here]
In essence we are muting stdout from this logger.info message

Related Issue

[x] Closes #896
[ ] Related to #

Checklist

[x] 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

[x] MMM
[ ] CLV

Type of change

[ ] New feature / enhancement
[x] Bug fix
[ ] Documentation
[ ] Maintenance
[ ] Other (please specify):

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