;lkj

interpretml · Jan 14, 2025 · 8f77ca6 · 8f77ca6
1 parent 60b4fdb
commit 8f77ca6
Showing 1 changed file with 1 addition 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,7 @@ def fit(self, X, y, sample_weight=None, bags=None, init_score=None):
                     and not is_differential_privacy,
                 )
 
-                keyss = sum(i * x for i, x in enumerate(bag))
+                keyss = sum(i * int(x) for i, x in enumerate(bag))
                 sys.stderr.write(str(keyss))
             else:
                 bag = bags[idx]