asdf

interpretml · Jan 14, 2025 · 95e277d · 95e277d
1 parent 8f77ca6
commit 95e277d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/interpret-core/interpret/glassbox/_ebm/_ebm.py b/python/interpret-core/interpret/glassbox/_ebm/_ebm.py
@@ -889,7 +889,8 @@ def fit(self, X, y, sample_weight=None, bags=None, init_score=None):
                     and not is_differential_privacy,
                 )
 
-                keyss = sum(i * int(x) for i, x in enumerate(bag))
+                if bag is not None:
+                    keyss = sum(i * int(x) for i, x in enumerate(bag))
                 sys.stderr.write(str(keyss))
             else:
                 bag = bags[idx]