Add LogisticRegressionEstimator to formula type check

Author: christopher-wild

causal_testing/json_front/json_class.py

@@ -22,7 +22,7 @@
 from causal_testing.specification.variable import Input, Meta, Output
 from causal_testing.testing.causal_test_case import CausalTestCase
 from causal_testing.testing.causal_test_result import CausalTestResult
-from causal_testing.testing.estimators import Estimator, LinearRegressionEstimator
+from causal_testing.testing.estimators import Estimator, LinearRegressionEstimator, LogisticRegressionEstimator
 from causal_testing.testing.base_test_case import BaseTestCase
 from causal_testing.testing.causal_test_adequacy import DataAdequacy
 
@@ -309,8 +309,10 @@ def _setup_test(self, causal_test_case: CausalTestCase, test: Mapping) -> Estima
         """
         estimator_kwargs = {}
         if "formula" in test:
-            if test["estimator"] != LinearRegressionEstimator:
-                raise TypeError("Currently only LinearRegressionEstimator supports the use of formulas")
+            if test["estimator"] != LinearRegressionEstimator or LogisticRegressionEstimator:
+                raise TypeError(
+                    "Currently only LinearRegressionEstimator and LogisticRegressionEstimator supports the use of formulas"
+                )
             estimator_kwargs["formula"] = test["formula"]
             estimator_kwargs["adjustment_set"] = {}
         else: