Fix example

optuna · Nov 22, 2024 · e75c251 · e75c251
1 parent ea70e5f
commit e75c251
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/package/samplers/nsgaii_with_tpe_warmup/README.md b/package/samplers/nsgaii_with_tpe_warmup/README.md
@@ -34,7 +34,7 @@ def objective(trial: optuna.Trial) -> tuple[float, float]:
 
 package_name = "samplers/nsgaii_with_tpe_warmup"
 sampler = optunahub.load_module(package=package_name).NSGAIIWithTPEWarmupSampler()
-study = optuna.create_study(sampler=sampler)
+study = optuna.create_study(sampler=sampler, directions=["minimize"]*2)
 study.optimize(objective, n_trials=60)
 
 ```
diff --git a/package/samplers/nsgaii_with_tpe_warmup/example.py b/package/samplers/nsgaii_with_tpe_warmup/example.py
@@ -12,5 +12,5 @@ def objective(trial: optuna.Trial) -> tuple[float, float]:
 
 package_name = "samplers/nsgaii_with_tpe_warmup"
 sampler = optunahub.load_module(package=package_name).NSGAIIWithTPEWarmupSampler()
-study = optuna.create_study(sampler=sampler)
+study = optuna.create_study(sampler=sampler, directions=["minimize"] * 2)
 study.optimize(objective, n_trials=60)