black

Author: christopher-wild

causal_testing/generation/abstract_causal_test_case.py

@@ -28,7 +28,7 @@ class AbstractCausalTestCase:
     """
 
     def __init__(
-        #pylint: disable=too-many-arguments
+        # pylint: disable=too-many-arguments
         self,
         scenario: Scenario,
         intervention_constraints: set[z3.ExprRef],
@@ -78,8 +78,11 @@ def sanitise(string):
         )
 
     def _generate_concrete_tests(
-        #pylint: disable=too-many-locals
-        self, sample_size: int, rct: bool = False, seed: int = 0
+        # pylint: disable=too-many-locals
+        self,
+        sample_size: int,
+        rct: bool = False,
+        seed: int = 0,
     ) -> tuple[list[CausalTestCase], pd.DataFrame]:
         """Generates a list of `num` concrete test cases.
 
@@ -153,7 +156,7 @@ def _generate_concrete_tests(
         return concrete_tests, pd.DataFrame(runs, columns=run_columns + ["bin"])
 
     def generate_concrete_tests(
-        #pylint: disable=too-many-arguments, too-many-locals
+        # pylint: disable=too-many-arguments, too-many-locals
         self,
         sample_size: int,
         target_ks_score: float = None,

causal_testing/testing/causal_test_case.py

@@ -10,7 +10,7 @@
 
 
 class CausalTestCase:
-    #pylint: disable=too-many-instance-attributes
+    # pylint: disable=too-many-instance-attributes
     """
     A CausalTestCase extends the information held in a BaseTestCase. As well as storing the treatment and outcome
     variables, a CausalTestCase stores the values of these variables. Also the outcome variable and value are
@@ -23,7 +23,7 @@ class CausalTestCase:
     """
 
     def __init__(
-        #pylint: disable=too-many-arguments
+        # pylint: disable=too-many-arguments
         self,
         base_test_case: BaseTestCase,
         expected_causal_effect: CausalTestOutcome,

causal_testing/testing/estimators.py

@@ -18,7 +18,7 @@
 
 
 class Estimator(ABC):
-    #pylint: disable=too-many-instance-attributes
+    # pylint: disable=too-many-instance-attributes
     """An estimator contains all of the information necessary to compute a causal estimate for the effect of changing
     a set of treatment variables to a set of values.
 
@@ -36,7 +36,7 @@ class Estimator(ABC):
     """
 
     def __init__(
-        #pylint: disable=too-many-arguments
+        # pylint: disable=too-many-arguments
         self,
         treatment: str,
         treatment_value: float,