Merged with main tests pass

Author: jmafoster1

tests/testing_tests/test_estimators.py

@@ -371,13 +371,12 @@ def setUpClass(cls) -> None:
         df = pd.DataFrame({"X1": np.random.uniform(-1000, 1000, 1000), "X2": np.random.uniform(-1000, 1000, 1000)})
         df["Y"] = 2 * df["X1"] - 3 * df["X2"] + 2 * df["X1"] * df["X2"] + 10
         cls.df = df
-        print(df)
 
     def test_X1_effect(self):
         """When we fix the value of X2 to 0, the effect of X1 on Y should become ~2 (because X2 terms are cancelled)."""
         x2 = Input("X2", float)
         lr_model = LinearRegressionEstimator(
-            ("X1",), 1, 0, {"X2"}, ("Y",), effect_modifiers={x2: 0}, product_terms=[("X1", "X2")], df=self.df
+            "X1", 1, 0, {"X2"}, "Y", effect_modifiers={x2: 0}, product_terms=[("X1", "X2")], df=self.df
         )
         test_results = lr_model.estimate_ate()
         ate = test_results[0]