Fix preferential continuous check

2026-09-12 12:40:33 +08:00 · 2024-02-06 20:09:26 +09:00
parent cfadf70b4a
commit c70cd09fab
1 changed files with 4 additions and 1 deletions
@@ -389,7 +389,10 @@ class PreferentialGPSampler(optuna.samplers.BaseSampler):
            def get_all_possible_params(dist: optuna.distributions.BaseDistribution) -> list[Any]:
                if isinstance(dist, CategoricalDistribution):
                    return list(dist.choices)
-                elif isinstance(dist, (IntDistribution, FloatDistribution)):
+                elif (
+                    isinstance(dist, (IntDistribution, FloatDistribution))
+                    and dist.step is not None
+                ):
                    return list(np.arange(dist.low, dist.high, dist.step))
                else:
                    return []