flip TermBoostFlags_MissingCategory

interpretml · Jan 2, 2025 · e19927a · e19927a
1 parent 5098d64
commit e19927a
Showing 1 changed file with 6 additions and 13 deletions.
diff --git a/python/interpret-core/interpret/glassbox/_ebm/_boost.py b/python/interpret-core/interpret/glassbox/_ebm/_boost.py
@@ -156,21 +156,14 @@ def boost(
                         )
 
                 if missing == "low":
-                    term_boost_flags_local |= (
-                        Native.TermBoostFlags_MissingLow
-                        | Native.TermBoostFlags_MissingCategory
-                    )
+                    term_boost_flags_local |= Native.TermBoostFlags_MissingLow
                 elif missing == "high":
-                    term_boost_flags_local |= (
-                        Native.TermBoostFlags_MissingHigh
-                        | Native.TermBoostFlags_MissingCategory
-                    )
+                    term_boost_flags_local |= Native.TermBoostFlags_MissingHigh
                 elif missing == "separate":
-                    term_boost_flags_local |= (
-                        Native.TermBoostFlags_MissingSeparate
-                        | Native.TermBoostFlags_MissingCategory
-                    )
-                elif missing != "gain":
+                    term_boost_flags_local |= Native.TermBoostFlags_MissingSeparate
+                elif missing == "gain":
+                    term_boost_flags_local |= Native.TermBoostFlags_MissingCategory
+                else:
                     msg = f"Unrecognized missing option {missing}."
                     raise Exception(msg)