Merge pull request #68 from gen740/fix_import_statement

Fix unused (and unintended) import
optuna · Feb 6, 2024 · 6462be5 · 6462be5
2 parents b9f6072 + dd7a784
commit 6462be5
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/importance_tests/test_init.py b/tests/importance_tests/test_init.py
@@ -2,9 +2,7 @@
 
 import numpy as np
 import optuna
-from optuna import samplers
 from optuna.importance import get_param_importances
-import optuna.integration.shap
 from optuna.samplers import RandomSampler
 from optuna.study import create_study
 from optuna.testing.objectives import pruned_objective
@@ -62,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(