Update test_estimator_list.py

Signed-off-by: AlxdrPolyakov <[email protected]>
py-why · Sep 9, 2024 · 2e2492e · 2e2492e
1 parent d7277e9
commit 2e2492e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/causaltune/test_estimator_list.py b/tests/causaltune/test_estimator_list.py
@@ -92,7 +92,7 @@ def test_invalid_choice(self):
     def test_invalid_choice_fitter(self):
         with pytest.raises(AssertionError):
             """tests if empty list is correctly handled"""
-            ct = CausalTune(components_time_budget=10)
+            ct = CausalTune(components_time_budget=10, outcome_model="auto")
             ct.fit(
                 pd.DataFrame(
                     {"treatment": [0, 1], "outcome": [0.5, 1.5], "dummy": [0.1, 0.2]}