Skip to content

Commit 2764037

Browse files
authored
fix-1535-Exception-in-the-fit()-call-of-AutoSklearn (#1539)
* Create PR * Fix test fixture
1 parent 4f691a1 commit 2764037

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

test/fixtures/ensemble_building.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,10 @@ def _make(
231231
# Hence, we take the y_train of the datamanager and use that as the
232232
# the targets
233233
if "Y_train" in datamanager.data:
234-
backend.save_targets_ensemble(datamanager.data["Y_train"])
234+
backend.save_additional_data(
235+
datamanager.data["Y_train"],
236+
what="targets_ensemble",
237+
)
235238

236239
return EnsembleBuilderManager(
237240
backend=backend,

0 commit comments

Comments
 (0)