Black

Author: jmafoster1

causal_testing/json_front/json_class.py

@@ -227,7 +227,9 @@ def _setup_test(
                 - estimation_model - Estimator instance for the test being run
         """
 
-        data_collector = ObservationalDataCollector(self.modelling_scenario, self.data.query(" & ".join(conditions)) if conditions else self.data)
+        data_collector = ObservationalDataCollector(
+            self.modelling_scenario, self.data.query(" & ".join(conditions)) if conditions else self.data
+        )
         causal_test_engine = CausalTestEngine(self.causal_specification, data_collector, index_col=0)
 
         minimal_adjustment_set = self.causal_specification.causal_dag.identification(causal_test_case.base_test_case)

causal_testing/specification/metamorphic_relation.py

@@ -136,14 +136,14 @@ def test_oracle(self, test_results):
     def to_json_stub(self, skip=True) -> dict:
         """Convert to a JSON frontend stub string for user customisation"""
         return {
-                "name": str(self),
-                "estimator": "LinearRegressionEstimator",
-                "estimate_type": "coefficient",
-                "effect": "direct",
-                "mutations": [self.treatment_var],
-                "expectedEffect": {self.output_var: "SomeEffect"},
-                "skip": skip
-              }
+            "name": str(self),
+            "estimator": "LinearRegressionEstimator",
+            "estimate_type": "coefficient",
+            "effect": "direct",
+            "mutations": [self.treatment_var],
+            "expectedEffect": {self.output_var: "SomeEffect"},
+            "skip": skip,
+        }
 
     def __str__(self):
         formatted_str = f"{self.treatment_var} --> {self.output_var}"
@@ -165,18 +165,17 @@ def test_oracle(self, test_results):
             len(test_results["fail"]) == 0
         ), f"{str(self)}: {len(test_results['fail'])}/{len(self.tests)} tests failed."
 
-
     def to_json_stub(self, skip=True) -> dict:
         """Convert to a JSON frontend stub string for user customisation"""
         return {
-                "name": str(self),
-                "estimator": "LinearRegressionEstimator",
-                "estimate_type": "coefficient",
-                "effect": "direct",
-                "mutations": [self.treatment_var],
-                "expectedEffect": {self.output_var: "NoEffect"},
-                "skip": skip
-              }
+            "name": str(self),
+            "estimator": "LinearRegressionEstimator",
+            "estimate_type": "coefficient",
+            "effect": "direct",
+            "mutations": [self.treatment_var],
+            "expectedEffect": {self.output_var: "NoEffect"},
+            "skip": skip,
+        }
 
     def __str__(self):
         formatted_str = f"{self.treatment_var} _||_ {self.output_var}"

causal_testing/testing/estimators.py

@@ -541,7 +541,9 @@ def estimate_coefficient(self, df):
         return ab / a
 
     def estimate_unit_ate(self, bootstrap_size=100):
-        bootstraps = sorted([self.estimate_coefficient(self.df.sample(len(self.df), replace=True)) for _ in range(bootstrap_size)])
+        bootstraps = sorted(
+            [self.estimate_coefficient(self.df.sample(len(self.df), replace=True)) for _ in range(bootstrap_size)]
+        )
         bound = ceil((bootstrap_size * 0.05) / 2)
         ci_low = bootstraps[bound]
         ci_high = bootstraps[bootstrap_size - bound]