diff --git a/python/interpret-core/interpret/glassbox/_ebm/_boost.py b/python/interpret-core/interpret/glassbox/_ebm/_boost.py index cbea7cdb3..6bb64e48f 100644 --- 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)