File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
cmd/suggestion/optuna/v1beta1
pkg/suggestion/v1beta1/optuna Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 1
1
grpcio == 1.41.1
2
2
protobuf == 3.19.1
3
3
googleapis-common-protos == 1.53.0
4
- optuna < 3.0.0
4
+ optuna >= 3.0.0
Original file line number Diff line number Diff line change @@ -91,9 +91,9 @@ def _get_optuna_search_space(self):
91
91
search_space = {}
92
92
for param in self .search_space .params :
93
93
if param .type == INTEGER :
94
- search_space [param .name ] = optuna .distributions .IntUniformDistribution (int (param .min ), int (param .max ))
94
+ search_space [param .name ] = optuna .distributions .IntDistribution (int (param .min ), int (param .max ))
95
95
elif param .type == DOUBLE :
96
- search_space [param .name ] = optuna .distributions .UniformDistribution (float (param .min ), float (param .max ))
96
+ search_space [param .name ] = optuna .distributions .FloatDistribution (float (param .min ), float (param .max ))
97
97
elif param .type == CATEGORICAL or param .type == DISCRETE :
98
98
search_space [param .name ] = optuna .distributions .CategoricalDistribution (param .list )
99
99
return search_space
You can’t perform that action at this time.
0 commit comments