Fix

optuna · Dec 10, 2024 · adc56e6 · adc56e6
1 parent 7ebafa6
commit adc56e6
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/package/samplers/hebo/sampler.py b/package/samplers/hebo/sampler.py
@@ -193,14 +193,13 @@ def _convert_to_hebo_design_space(
                     config["lb"] = 0
                     config["ub"] = n_steps - 1
                 else:
-                    config["type"] = "_".join(
-                        [
-                            "pow" if distribution.log else "",
-                            "int" if isinstance(distribution, IntDistribution) else "num",
-                        ]
-                    )
                     config["lb"] = distribution.low
                     config["ub"] = distribution.high
+                    is_int = isinstance(distribution, IntDistribution)
+                    if distribution.log:
+                        config["type"] = "pow_int" if is_int else "pow"
+                    else:
+                        config["type"] = "int" if is_int else "num"
             elif isinstance(distribution, CategoricalDistribution):
                 config["type"] = "cat"
                 config["categories"] = distribution.choices