Skip to content
Snippets Groups Projects
Commit 639cb0c6 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Add support for fitting estimators on dask bags

The estimators that can handle dask bags should set
the `bob_fit_supports_dask_bag` as True.
This commit also includes
* Adds a new tag: `bob_fit_supports_dask_bag`
* Adds a new tag: `bob_checkpoint_features` for when you want to always avoid
  checkpointing features for a specific estimator.
* Expose dask_tags, get_bob_tags in the main API
* The SampleWrapper was modified to support `bob_fit_supports_dask_bag`
* The CheckpointWrapper now loads estimators without losing references correctly.
parent fc80d8d5
No related branches found
No related tags found
1 merge request!88Add support for fitting estimators on dask bags
Pipeline #61012 failed
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment