optimize.py

from src.optimize_hyperparameters import objective

# we define a lambda function because study.optimize()
# expects the objective function to have only 1 input
# (trial), while our objective function hast 2 extra
# inputs I defined to add flexibility to the script
func = lambda trial: objective(trial,
                               force_linear_model=False,
                               n_episodes_to_train=200)
                               
study.optimize(func, n_trials=100)