pylint

Author: jmafoster1

causal_testing/surrogate/causal_surrogate_assisted.py

@@ -106,7 +106,8 @@ def execute(
                 )
                 test_result.relationship = (
                     f"{surrogate_model.base_test_case.treatment_variable.name} -> "
-                    f"{surrogate_model.base_test_case.outcome_variable.name} expected {surrogate_model.expected_relationship}"
+                    f"{surrogate_model.base_test_case.outcome_variable.name} expected "
+                    f"{surrogate_model.expected_relationship}"
                 )
                 return test_result, i + 1, df
 

causal_testing/testing/causal_test_result.py

@@ -129,8 +129,9 @@ def ci_valid(self) -> bool:
 
     def summary(self):
         """Summarise the causal test result as an intuitive sentence."""
+        treatment_variable = self.estimator.base_test_case.treatment_variable
         print(
-            f"The causal effect of changing {self.estimator.base_test_case.treatment_variable.name} = {self.estimator.control_value} to "
-            f"{self.estimator.base_test_case.treatment_variable.name}' = {self.estimator.treatment_value} is {self.test_value.value}"
+            f"The causal effect of changing {treatment_variable.name} = {self.estimator.control_value} to "
+            f"{treatment_variable.name}' = {self.estimator.treatment_value} is {self.test_value.value}"
             f"(95% confidence intervals: {self.confidence_intervals})."
         )