Skip to content

Commit

Permalink
Fix discrete setting
Browse files Browse the repository at this point in the history
  • Loading branch information
YamLyubov committed Aug 23, 2023
1 parent eba4897 commit fe151ba
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions golem/core/tuning/iopt_tuner.py
Original file line number Diff line number Diff line change
Expand Up @@ -235,8 +235,8 @@ def get_node_parameters_for_iopt(search_space: SearchSpace, node_id: int, operat

parameter_type = parameter_properties.get('type')
if parameter_type == 'discrete':
discrete_parameters_dict.update({node_op_parameter_name: [range(*parameter_properties
.get('sampling-scope'))]})
discrete_parameters_dict.update({node_op_parameter_name: list(range(*parameter_properties
.get('sampling-scope')))})
elif parameter_type == 'continuous':
float_parameters_dict.update({node_op_parameter_name: parameter_properties
.get('sampling-scope')})
Expand Down

0 comments on commit fe151ba

Please sign in to comment.