Merge branch 'main' into some-devops

Author: drbenvincent

causalpy/pymc_experiments.py

@@ -709,7 +709,7 @@ def summary(self, round_to=None) -> None:
         print(f"{self.expt_type:=^80}")
         print(f"Formula: {self.formula}")
         print("\nResults:")
-        print(round_num(self._causal_impact_summary_stat(), round_to))
+        print(self._causal_impact_summary_stat(round_to))
         self.print_coefficients(round_to)
 
 
@@ -903,7 +903,7 @@ def plot(self, round_to=None):
         )
         return fig, ax
 
-    def summary(self, round_to: None) -> None:
+    def summary(self, round_to=None) -> None:
         """
         Print text output summarising the results
 

causalpy/tests/test_pymc_experiments.py

@@ -17,8 +17,10 @@ def test_did_summary():
         group_variable_name="group",
         model=cp.pymc_models.LinearRegression(sample_kwargs=sample_kwargs),
     )
-    print(type(result._causal_impact_summary_stat()))
-    assert isinstance(result._causal_impact_summary_stat(), str)
+    round_to = None
+    print(type(result._causal_impact_summary_stat(round_to)))
+    print(result._causal_impact_summary_stat(round_to))
+    assert isinstance(result._causal_impact_summary_stat(round_to), str)
 
 
 def test_regression_kink_gradient_change():