Add pylint disable messages

Author: christopher-wild

causal_testing/generation/abstract_causal_test_case.py

@@ -28,6 +28,7 @@ class AbstractCausalTestCase:
     """
 
     def __init__(
+        #pylint: disable=too-many-arguments
         self,
         scenario: Scenario,
         intervention_constraints: set[z3.ExprRef],
@@ -77,6 +78,7 @@ def sanitise(string):
         )
 
     def _generate_concrete_tests(
+        #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.
@@ -151,6 +153,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
         self,
         sample_size: int,
         target_ks_score: float = None,

causal_testing/testing/causal_test_case.py

@@ -10,6 +10,7 @@
 
 
 class CausalTestCase:
+    #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
@@ -22,6 +23,7 @@ class CausalTestCase:
     """
 
     def __init__(
+        #pylint: disable=too-many-arguments
         self,
         base_test_case: BaseTestCase,
         expected_causal_effect: CausalTestOutcome,

causal_testing/testing/estimators.py

@@ -18,6 +18,7 @@
 
 
 class Estimator(ABC):
+    #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.
 
@@ -35,6 +36,7 @@ class Estimator(ABC):
     """
 
     def __init__(
+        #pylint: disable=too-many-arguments
         self,
         treatment: str,
         treatment_value: float,
@@ -92,6 +94,7 @@ class LogisticRegressionEstimator(Estimator):
     """
 
     def __init__(
+        # pylint: disable=too-many-arguments
         self,
         treatment: str,
         treatment_value: float,
@@ -291,6 +294,7 @@ class LinearRegressionEstimator(Estimator):
     """
 
     def __init__(
+        # pylint: disable=too-many-arguments
         self,
         treatment: str,
         treatment_value: float,
@@ -549,6 +553,7 @@ class InstrumentalVariableEstimator(Estimator):
     """
 
     def __init__(
+        # pylint: disable=too-many-arguments
         self,
         treatment: str,
         treatment_value: float,