pylint

Author: jmafoster1

causal_testing/estimation/experimental_estimator.py

@@ -1,8 +1,8 @@
 """This module contains the ExperimentalEstimator class for directly interacting with the system under test."""
 
-import pandas as pd
 from typing import Any
 from abc import abstractmethod
+import pandas as pd
 
 from causal_testing.estimation.abstract_estimator import Estimator
 

causal_testing/testing/causal_test_case.py

@@ -2,7 +2,6 @@
 
 import logging
 from typing import Any
-import pandas as pd
 
 from causal_testing.specification.variable import Variable
 from causal_testing.testing.causal_test_outcome import CausalTestOutcome

causal_testing/testing/metamorphic_relation.py

@@ -47,7 +47,10 @@ def to_json_stub(self, skip=True) -> dict:
             "effect": "direct",
             "mutations": [self.base_test_case.treatment_variable],
             "expected_effect": {self.base_test_case.outcome_variable: "SomeEffect"},
-            "formula": f"{self.base_test_case.outcome_variable} ~ {' + '.join([self.base_test_case.treatment_variable] + self.adjustment_vars)}",
+            "formula": (
+                f"{self.base_test_case.outcome_variable} ~ "
+                f"{' + '.join([self.base_test_case.treatment_variable] + self.adjustment_vars)}"
+            ),
             "skip": skip,
         }
 
@@ -70,7 +73,10 @@ def to_json_stub(self, skip=True) -> dict:
             "effect": "direct",
             "mutations": [self.base_test_case.treatment_variable],
             "expected_effect": {self.base_test_case.outcome_variable: "NoEffect"},
-            "formula": f"{self.base_test_case.outcome_variable} ~ {' + '.join([self.base_test_case.treatment_variable] + self.adjustment_vars)}",
+            "formula": (
+                f"{self.base_test_case.outcome_variable} ~ "
+                f"{' + '.join([self.base_test_case.treatment_variable] + self.adjustment_vars)}"
+            ),
             "alpha": 0.05,
             "skip": skip,
         }