Skip to content

Commit bce7fb6

Browse files
committed
Merge remote-tracking branch 'origin/aad_loss_function' into aad_loss_function
2 parents 5388e11 + 152093b commit bce7fb6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/coniferest/aadforest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ def __init__(
206206
random_seed=None,
207207
sampletrees_per_batch=1 << 20,
208208
map_value=None,
209-
loss='hinge',
209+
loss="hinge",
210210
):
211211
super().__init__(
212212
trees=[],
@@ -250,7 +250,7 @@ def __init__(
250250

251251
LOSSES = ["hinge"]
252252

253-
if not (loss in LOSSES):
253+
if loss not in LOSSES:
254254
raise ValueError(f"loss is not one of {', '.join(LOSSES.keys())}.")
255255

256256
self.loss = loss

0 commit comments

Comments
 (0)