Skip to content

Commit

Permalink
Also delete duplicated import
Browse files Browse the repository at this point in the history
  • Loading branch information
gen740 committed Feb 2, 2024
1 parent fcb0ff8 commit dd7a784
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tests/importance_tests/test_init.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@

import numpy as np
import optuna
from optuna import samplers
from optuna.importance import get_param_importances
from optuna.samplers import RandomSampler
from optuna.study import create_study
Expand Down Expand Up @@ -61,7 +60,7 @@ def objective(trial: Trial) -> float:
return value

with StorageSupplier(storage_mode) as storage:
study = create_study(storage=storage, sampler=samplers.RandomSampler())
study = create_study(storage=storage, sampler=RandomSampler())
study.optimize(objective, n_trials=3)

param_importance = get_param_importances(
Expand Down

0 comments on commit dd7a784

Please sign in to comment.