tests

Signed-off-by: Shashank Mittal <[email protected]>
shashank-iitbhu · Aug 27, 2024 · f0e8a12 · f0e8a12
1 parent fab5639
commit f0e8a12
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git 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
@@ -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)
                         )
                     ]
                 )