[IV 212] adding axis labels with Ben's comments

Signed-off-by: Nathaniel <[email protected]>

Author: NathanielF

causalpy/pymc_experiments.py

@@ -1002,11 +1002,12 @@ def _input_validation(self):
                 as an outcome variable and in the data object to be used as a covariate.
                 """
             )
-        check_binary = len(self.data[treatment.strip()].unique()) > 2
+        Z = self.data[treatment.strip()]
+        check_binary = len(np.unique(Z)) > 2
         if check_binary:
             warnings.warn(
                 """Warning. The treatment variable is not Binary.
-                This is not necessarily aproblem but it violates
+                This is not necessarily a problem but it violates
                 the assumption of a simple IV experiment.
-                Change your interpretation of model coefficients accordingly."""
+                The coefficients should be interpreted appropriately."""
             )

causalpy/pymc_models.py

@@ -136,8 +136,7 @@ def build_model(self, X, Z, y, t, coords, priors):
         :param priors: An optional dictionary of priors for the mus and
                       sigmas of both regressions
 
-        e.g priors = {'mus': [[10, 0], [2, 0]], 'sigmas': [[1, 1], [1, 1]]
-                      'eta': 2, 'lkj_sd': 2}
+        :code:`priors = {"mus": [0, 0], "sigmas": [1, 1], "eta": 2, "lkj_sd": 2}`
 
         """