[IV 212] pretty print user warning

Signed-off-by: Nathaniel <[email protected]>
pymc-labs · Aug 9, 2023 · f737115 · f737115
1 parent efa3f6d
commit f737115
Show file tree

Hide file tree

Showing 3 changed files with 85 additions and 77 deletions.
diff --git a/causalpy/pymc_experiments.py b/causalpy/pymc_experiments.py
@@ -1005,7 +1005,8 @@ def _input_validation(self):
         check_binary = len(self.data[treatment.strip()].unique()) > 2
         if check_binary:
             warnings.warn(
-                """Warning. The treatment variable is not Binary. \n
-                          This is not necessarily a problem but it complicates \n
-                          the interpretation of the model coefficients."""
+                """Warning. The treatment variable is not Binary.
+                This is not necessarily aproblem but it violates
+                the assumption of a simple IV experiment.
+                Change your interpretation of model coefficients accordingly."""
             )
diff --git a/docs/source/_static/interrogate_badge.svg b/docs/source/_static/interrogate_badge.svg
diff --git a/docs/source/notebooks/iv_pymc.ipynb b/docs/source/notebooks/iv_pymc.ipynb