Fix

optuna · Dec 10, 2024 · fedd54f · fedd54f
1 parent adc56e6
commit fedd54f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/package/samplers/hebo/sampler.py b/package/samplers/hebo/sampler.py
@@ -195,11 +195,13 @@ def _convert_to_hebo_design_space(
                 else:
                     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"
+                        config["type"] = (
+                            "pow_int" if isinstance(distribution, IntDistribution) else "pow"
+                        )
                     else:
-                        config["type"] = "int" if is_int else "num"
+                        assert not isinstance(distribution, IntDistribution)
+                        config["type"] = "num"
             elif isinstance(distribution, CategoricalDistribution):
                 config["type"] = "cat"
                 config["categories"] = distribution.choices