pylint

Author: jmafoster1

causal_testing/estimation/effect_estimate.py

@@ -2,8 +2,8 @@
 This module contains the EffectEstimate dataclass.
 """
 
-import pandas as pd
 from dataclasses import dataclass
+import pandas as pd
 
 
 @dataclass
@@ -39,4 +39,5 @@ def to_dict(self) -> dict:
         return d
 
     def to_df(self) -> pd.DataFrame:
+        """Return representation as a pandas dataframe."""
         return pd.DataFrame({"effect_estimate": self.value, "ci_low": self.ci_low, "ci_high": self.ci_high})

causal_testing/testing/causal_test_result.py

@@ -53,8 +53,6 @@ def push(s, inc="  "):
         if self.effect_estimate.ci_valid():
             ci_str = f"CI low: {self.effect_estimate.ci_low.to_dict}\n"
             ci_str += f"CI high: {self.effect_estimate.ci_high.to_dict}\n"
-            if "\n" in ci_str:
-                ci_str = " " + push(pd.DataFrame(self.confidence_intervals).transpose().to_string(header=False))
             confidence_str += f"Confidence intervals:{ci_str}\n"
             confidence_str += f"Alpha:{self.estimator.alpha}\n"
         adequacy_str = ""