jrbourbeau / dask-optuna

Scale Optuna with Dask
https://jrbourbeau.github.io/dask-optuna/
MIT License
35 stars 6 forks source link

Multiple DaskStorage instances #5

Closed jrbourbeau closed 3 years ago

jrbourbeau commented 3 years ago

This allows multiple DaskStorage instances with the same type of underlying Optuna storage class by adding a new unique name identifier. Today if one does:

import dask_optuna

s1 = dask_optuna.DaskStorage()
s2 = dask_optuna.DaskStorage()

then the OptunaSchedulerExtension registered on the scheduler will store only a single Optuna InMemoryStorage class in it's .storages dictionary.