diff --git a/examples/v1beta1/hp-tuning/random.yaml b/examples/v1beta1/hp-tuning/random.yaml index 0c9705d64f5..44a67c1dd86 100644 --- a/examples/v1beta1/hp-tuning/random.yaml +++ b/examples/v1beta1/hp-tuning/random.yaml @@ -20,6 +20,8 @@ spec: feasibleSpace: min: "0.01" max: "0.05" + step: "0.01" + distribution: "logUniform" - name: momentum parameterType: double feasibleSpace: diff --git a/test/unit/v1beta1/suggestion/test_hyperopt_service.py b/test/unit/v1beta1/suggestion/test_hyperopt_service.py index 30486efa5d4..f865f75ba09 100644 --- a/test/unit/v1beta1/suggestion/test_hyperopt_service.py +++ b/test/unit/v1beta1/suggestion/test_hyperopt_service.py @@ -16,6 +16,7 @@ import grpc import grpc_testing +from pkg.suggestion.v1beta1.internal.constant import LOG_UNIFORM import utils from pkg.apis.manager.v1beta1.python import api_pb2 @@ -176,6 +177,12 @@ def test_get_suggestion(self): parameter_type=api_pb2.DOUBLE, feasible_space=api_pb2.FeasibleSpace( max="5", min="1", list=[]) + ), + api_pb2.ParameterSpec( + name="param-5", + parameter_type=api_pb2.DOUBLE, + feasible_space=api_pb2.FeasibleSpace( + max="5", min="1", list=[], step="0.5", distribution=api_pb2.LOG_UNIFORM) ) ] )