Skip to content
GitLab
Projects Groups Snippets
  • /
  • Help
    • Help
    • Support
    • Community forum
    • Submit feedback
    • Contribute to GitLab
  • Sign in
  • bob.pipelines bob.pipelines
  • Project information
    • Project information
    • Activity
    • Labels
    • Members
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
  • Issues 5
    • Issues 5
    • List
    • Boards
    • Service Desk
    • Milestones
  • Merge requests 0
    • Merge requests 0
  • CI/CD
    • CI/CD
    • Pipelines
    • Jobs
    • Schedules
  • Deployments
    • Deployments
    • Environments
    • Releases
  • Packages and registries
    • Packages and registries
    • Package Registry
    • Infrastructure Registry
  • Monitor
    • Monitor
    • Incidents
  • Analytics
    • Analytics
    • Value stream
    • CI/CD
    • Repository
  • Activity
  • Graph
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
Collapse sidebar
  • bobbob
  • bob.pipelinesbob.pipelines
  • Merge requests
  • !88

Add support for fitting estimators on dask bags

  • Review changes

  • Download
  • Email patches
  • Plain diff
Merged Amir MOHAMMADI requested to merge fit-dask-bags into master May 09, 2022
  • Overview 0
  • Commits 2
  • Pipelines 2
  • Changes 3

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.
Assignee
Assign to
Reviewers
Request review from
Time tracking
Source branch: fit-dask-bags