From 3fc7cafc0018e053b231bbcae68eedbefc1d8bf2 Mon Sep 17 00:00:00 2001 From: Will Dean <57733339+wd60622@users.noreply.github.com> Date: Sat, 22 Jun 2024 10:25:31 +0200 Subject: [PATCH] remove warnings on init (#771) --- pymc_marketing/mmm/delayed_saturated_mmm.py | 18 +----------------- 1 file changed, 1 insertion(+), 17 deletions(-) diff --git a/pymc_marketing/mmm/delayed_saturated_mmm.py b/pymc_marketing/mmm/delayed_saturated_mmm.py index 526a5f8e..75069f79 100644 --- a/pymc_marketing/mmm/delayed_saturated_mmm.py +++ b/pymc_marketing/mmm/delayed_saturated_mmm.py @@ -426,11 +426,6 @@ def build_model( ) ): if self.model_config["gamma_control"].get("dims") != "control": - msg = ( - "The 'dims' key in gamma_control must be 'control'." - " This will be fixed automatically." - ) - warnings.warn(msg, stacklevel=2) self.model_config["gamma_control"]["dims"] = "control" gamma_control = create_distribution_from_config( @@ -470,11 +465,6 @@ def build_model( ) if self.model_config["gamma_fourier"].get("dims") != "fourier_mode": - msg = ( - "The 'dims' key in gamma_fourier must be 'fourier_mode'." - " This will be fixed automatically." - ) - warnings.warn(msg, stacklevel=2) self.model_config["gamma_fourier"]["dims"] = "fourier_mode" gamma_fourier = create_distribution_from_config( @@ -548,14 +538,8 @@ def default_model_config(self) -> dict: } for media_transform in [self.adstock, self.saturation]: - for param, config in media_transform.function_priors.items(): + for config in media_transform.function_priors.values(): if "dims" not in config: - msg = ( - f"{param} doesn't have a 'dims' key in config. Setting to channel." - f" Set priors explicitly in {media_transform.__class__.__name__}" - " to avoid this warning." - ) - warnings.warn(msg, stacklevel=2) config["dims"] = "channel" return {