Merge pull request #201 from nabenabe0928/fix-mypy-in-ctpe

Fix a typing error
optuna · Dec 10, 2024 · 660c61a · 660c61a
2 parents 5cb16be + 2ac3c4e
commit 660c61a
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: