Merge pull request #83 from Alnusjaponica/fix-import-path

Fix import path
optuna · Feb 13, 2024 · 3a6b327 · 3a6b327
2 parents 33a2d2d + 3f0344d
commit 3a6b327
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optuna_integration/xgboost.py b/optuna_integration/xgboost.py
@@ -2,12 +2,12 @@
 
 import optuna
 
-import optuna_integration
+from optuna_integration._imports import try_import
 
 
 use_callback_cls = True
 
-with optuna_integration._imports.try_import() as _imports:
+with try_import() as _imports:
     import xgboost as xgb
 
     xgboost_version = xgb.__version__.split(".")