feat(mmm): Replace ValueError with UserWarning for negative values

pymc-labs · Oct 23, 2024 · 6e5f622 · 6e5f622
1 parent 3f991d6
commit 6e5f622
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pymc_marketing/mmm/validating.py b/pymc_marketing/mmm/validating.py
@@ -14,6 +14,7 @@
 """Validating methods for MMM classes."""
 
 from collections.abc import Callable
+from warnings import warn
 
 import pandas as pd
 
@@ -116,8 +117,10 @@ def validate_channel_columns(self, data: pd.DataFrame) -> None:
                 f"channel_columns {self.channel_columns} contains duplicates"
             )
         if (data.filter(list(self.channel_columns)) < 0).any().any():
-            raise ValueError(
-                f"channel_columns {self.channel_columns} contains negative values"
+            warn(
+                f"channel_columns {self.channel_columns} contains negative values",
+                UserWarning,
+                stacklevel=2,
             )