Binary features now allow bounds

cog-imperial · Dec 31, 2024 · e1ce8d8 · e1ce8d8
1 parent 4057ae3
commit e1ce8d8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/entmoot/problem_config.py b/entmoot/problem_config.py
@@ -284,10 +284,11 @@ def add_feature(
         if name is None:
             name = f"feat_{len(self.feat_list)}"
 
+        if feat_type == "binary":
+            self._feat_list.append(Binary(name=name))
+            return
+
         if bounds is None:
-            if feat_type == "binary":
-                self._feat_list.append(Binary(name=name))
-                return
             raise ValueError(
                 "Please provide bounds for feature types in '(real, integer, categorical)'"
             )