From c142d50300f3b23e9127113e1b0d56b1fce54da0 Mon Sep 17 00:00:00 2001 From: wizardlyrocket8 Date: Mon, 16 Feb 2026 09:38:25 +0300 Subject: [PATCH] Fix RangeParameter from_config.py returns ChoiceParameter --- ax/api/utils/instantiation/from_config.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/ax/api/utils/instantiation/from_config.py b/ax/api/utils/instantiation/from_config.py index 705d2affecb..7c34f342087 100644 --- a/ax/api/utils/instantiation/from_config.py +++ b/ax/api/utils/instantiation/from_config.py @@ -50,21 +50,21 @@ def parameter_from_config( "The range of the parameter must be evenly divisible by the " "step size." ) - - return ChoiceParameter( + return RangeParameter( name=config.name, parameter_type=_parameter_type_converter(config.parameter_type), - values=[*np.arange(lower, upper + step_size, step_size)], - is_ordered=True, + lower=lower, + upper=upper, + log_scale=config.scaling == "log", ) - - return RangeParameter( + return ChoiceParameter( name=config.name, parameter_type=_parameter_type_converter(config.parameter_type), - lower=lower, - upper=upper, - log_scale=config.scaling == "log", + values=[*np.arange(lower, upper + step_size, step_size)], + is_ordered=True, ) + + elif isinstance(config, DerivedParameterConfig): return DerivedParameter( name=config.name,