Looks like Optuna moved third party integrations to optuna-integrations I think in some notebooks, this would require installing also optuna-integrations.

I can confirm that to be able to do optuna.integration.DaskStorage(), optuna-integration needs to be installed.

TODO:

• https://docs.rapids.ai/deployment/stable/examples/xgboost-gpu-hpo-job-parallel-ngc/notebook/#perform-hyperparameter-optimization-with-a-toy-example

Needs to suggest installing optuna and optuna-integration

• https://docs.rapids.ai/deployment/stable/examples/xgboost-gpu-hpo-job-parallel-k8s/notebook/#perform-hyperparameter-optimization-with-a-toy-example

Under install system packages, needs to add optuna-integration

I can't run the examples fully, because I don't have a GPU or a NGC account, I'm happy to open a PR, with the suggestions but someone else will need to test they run.