Enhanced cross-validation inputs for Dask-ML HyperbandSearchCV

Hello,
Is it possible to add the same possible "cv" inputs for Dask-ML HyperbandSearchCV that are currently available for Dask-ML RandomizedSearchCV and Dask-ML GridSearchCV? HyperbandCV currently only allows inputs for test_size, but I would like to be able to define my own cross-validation iterable which would contain the indicies of my training and validation splits for model tuning and evaluation. RandomizedSearchCV and GridSearchCV both allow this sort of cross validation input as seen below.


Thank you!

Start with the HyperbandSearchCV entry point and compare its current test_size handling with the cv inputs supported by Dask-ML RandomizedSearchCV and GridSearchCV. Trace how each search class consumes training and validation splits, then verify that HyperbandSearchCV accepts a user-provided cross-validation iterable and uses those splits for tuning and evaluation.

Written by the indexing model from the issue text.