Merge pull request #137 from pymc-labs/notebook-fix-up

fix `results.summary` print out + rerun notebook

Author: drbenvincent

causalpy/pymc_experiments.py

@@ -38,23 +38,15 @@ def print_coefficients(self):
         for name in self.labels:
             coeff_samples = coeffs.sel(coeffs=name)
             print(
-                f"""
-                {name: <30}{coeff_samples.mean().data:.2f},
-                94% HDI [{coeff_samples.quantile(0.03).data:.2f},
-                {coeff_samples.quantile(1-0.03).data:.2f}]
-                """
+                f"{name: <30}{coeff_samples.mean().data:.2f}, 94% HDI [{coeff_samples.quantile(0.03).data:.2f}, {coeff_samples.quantile(1-0.03).data:.2f}]"  # noqa: E501
             )
         # add coeff for measurement std
         coeff_samples = az.extract(
             self.prediction_model.idata.posterior, var_names="sigma"
         )
         name = "sigma"
         print(
-            f"""
-            {name: <30}{coeff_samples.mean().data:.2f},
-            94% HDI [{coeff_samples.quantile(0.03).data:.2f},
-            {coeff_samples.quantile(1-0.03).data:.2f}]
-            """
+            f"{name: <30}{coeff_samples.mean().data:.2f}, 94% HDI [{coeff_samples.quantile(0.03).data:.2f}, {coeff_samples.quantile(1-0.03).data:.2f}]"  # noqa: E501
         )