Update test_sklearn_propensity_model.py

Signed-off-by: AlxdrPolyakov <[email protected]>
py-why · Sep 9, 2024 · 7ab640f · 7ab640f
1 parent 2e2492e
commit 7ab640f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/causaltune/test_sklearn_propensity_model.py b/tests/causaltune/test_sklearn_propensity_model.py
@@ -39,6 +39,7 @@ def test_sklearn_propensity_model(self):
             verbose=3,
             components_verbose=2,
             resources_per_trial={"cpu": 0.5},
+            outcome_model="auto",
         )
 
         ct.fit(data)
@@ -61,6 +62,7 @@ def test_sklearn_propensity_model_multivalue(self):
             estimator_list="all",
             num_samples=len(estimator_list),
             components_time_budget=10,
+            outcome_model="auto",
         )
         ct.fit(data)
         # TODO add an effect() call and an effect_tt call