From 95e277d7984366294d3b2c6984155f3f59fc48f2 Mon Sep 17 00:00:00 2001 From: Paul Koch Date: Tue, 14 Jan 2025 02:23:27 -0800 Subject: [PATCH] asdf --- python/interpret-core/interpret/glassbox/_ebm/_ebm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/interpret-core/interpret/glassbox/_ebm/_ebm.py b/python/interpret-core/interpret/glassbox/_ebm/_ebm.py index 807d357c7..864a7c859 100644 --- 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]