From c70cd09fabc277c58bd5bc5d5ac9564f3caf7576 Mon Sep 17 00:00:00 2001 From: Contramundum Date: Tue, 6 Feb 2024 20:09:26 +0900 Subject: [PATCH] Fix preferential continuous check --- optuna_dashboard/preferential/samplers/gp.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/optuna_dashboard/preferential/samplers/gp.py b/optuna_dashboard/preferential/samplers/gp.py index 24cea9a1..188432b8 100644 --- a/optuna_dashboard/preferential/samplers/gp.py +++ b/optuna_dashboard/preferential/samplers/gp.py @@ -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 []