handle edge case

Signed-off-by: Shashank Mittal <[email protected]>
shashank-iitbhu · Aug 26, 2024 · fab5639 · fab5639
1 parent 099d016
commit fab5639
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/suggestion/v1beta1/hyperopt/base_service.py b/pkg/suggestion/v1beta1/hyperopt/base_service.py
@@ -93,7 +93,12 @@ def create_hyperopt_domain(self):
                         hyperopt_search_space[param.name] = hyperopt.hp.loguniform(
                         param.name,
                         float(param.min),
-                        float(param.max))            
+                        float(param.max))
+                else:
+                    hyperopt_search_space[param.name] = hyperopt.hp.uniform(
+                        param.name,
+                        float(param.min),
+                        float(param.max))           
             elif param.type == CATEGORICAL or param.type == DISCRETE:
                 hyperopt_search_space[param.name] = hyperopt.hp.choice(
                     param.name, param.list