The code fails to run because optuna is not installed. Is optuna a critical requirement? In that case, shall we add it to pyproject?