Skip to content

Commit c2d4c78

Browse files
committed
update linear_boost.py
1 parent 0fdd4c1 commit c2d4c78

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/linearboost/linear_boost.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -276,7 +276,9 @@ def fit(self, X, y, sample_weight=None) -> Self:
276276
if sample_weight is not None:
277277
sample_weight = np.asarray(sample_weight)
278278
if sample_weight.shape[0] != X.shape[0]:
279-
raise ValueError(f"sample_weight.shape == {sample_weight.shape} is incompatible with X.shape == {X.shape}")
279+
raise ValueError(
280+
f"sample_weight.shape == {sample_weight.shape} is incompatible with X.shape == {X.shape}"
281+
)
280282
nonzero_mask = sample_weight != 0
281283
X = X[nonzero_mask]
282284
y = y[nonzero_mask]

0 commit comments

Comments
 (0)