Find out how to add pruner through hydra config

[robogast]

Pruners are not currently supported through the hydra optuna sweeper plugin: facebookresearch/hydra#1710 .

Adding the pruner to the sweeper ourselves might be doable, as it seems like a single argument added to https://github.com/facebookresearch/hydra/blob/30a8c8a070166ac2d3b19c4ca29af7aa500dc269/plugins/hydra_optuna_sweeper/hydra_plugins/hydra_optuna_sweeper/_impl.py#L183 .

Tasks:

• create fork of hydra: https://github.com/sara-nl/hydra/tree/optuna-pruner
• add pruner to OptunaSweeperConf: https://github.com/facebookresearch/hydra/blob/main/plugins/hydra_optuna_sweeper/hydra_plugins/hydra_optuna_sweeper/config.py#L133
• add pruner to study: https://github.com/facebookresearch/hydra/blob/30a8c8a070166ac2d3b19c4ca29af7aa500dc269/plugins/hydra_optuna_sweeper/hydra_plugins/hydra_optuna_sweeper/_impl.py#L183 .
• check functionality with submitit launcher
  • Fixed in hydra optuna implementation

[robogast]

Leaving open until a merge is performed following facebookresearch/hydra#1710