Fix a typing error

optuna · Dec 8, 2024 · 2ac3c4e · 2ac3c4e
1 parent 8b92c6a
commit 2ac3c4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/package/samplers/ctpe/sampler.py b/package/samplers/ctpe/sampler.py
@@ -151,7 +151,7 @@ def _sample(
         )
         quantiles.append(len(below_trials) / len(trials))
 
-        _samples_below: dict[str, list[_ParzenEstimator]] = {
+        _samples_below: dict[str, list[np.ndarray]] = {
             param_name: [] for param_name in search_space
         }
         for mpe in mpes_below: