Skip to content

Commit 380dc83

Browse files
Refacto development automl common backend debug (#2)
* [ADD] debug information * [FIX] try fork for more stability
1 parent c784d34 commit 380dc83

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/test_ensemble/test_ensemble.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -718,10 +718,10 @@ def test_ensemble_builder_nbest_remembered(fit_ensemble, ensemble_backend, dask_
718718
max_iterations=None,
719719
)
720720

721-
manager.build_ensemble(dask_client, unit_test=True)
721+
manager.build_ensemble(dask_client, unit_test=True, pynisher_context='fork')
722722
future = manager.futures[0]
723723
dask.distributed.wait([future]) # wait for the ensemble process to finish
724-
assert future.result() == ([], 5, None, None)
724+
assert future.result() == ([], 5, None, None), vars(future.result())
725725
file_path = os.path.join(ensemble_backend.internals_directory, 'ensemble_read_preds.pkl')
726726
assert not os.path.exists(file_path)
727727

0 commit comments

Comments
 (0)