Introduce early detection for dependencies

optuna · Nov 15, 2024 · 3323b0f · 3323b0f
1 parent d4997d9
commit 3323b0f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/package/samplers/auto_sampler/_sampler.py b/package/samplers/auto_sampler/_sampler.py
@@ -6,6 +6,7 @@
 from typing import Any
 from typing import TYPE_CHECKING
 
+import cmaes as _  # NOQA
 from optuna.distributions import CategoricalDistribution
 from optuna.logging import get_logger
 from optuna.samplers import BaseSampler
@@ -18,6 +19,8 @@
 from optuna.samplers._lazy_random_state import LazyRandomState
 from optuna.search_space import IntersectionSearchSpace
 from optuna.trial import TrialState
+import scipy as _  # NOQA
+import torch as _  # NOQA
 
 
 if TYPE_CHECKING: