Skip to content

Commit fb3aa86

Browse files
committed
fix(classifier): make log_prob an explicit constructor option
1 parent 79fdfdd commit fb3aa86

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

delphi/scorers/classifier/classifier.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ def __init__(
2222
verbose: bool,
2323
n_examples_shown: int,
2424
seed: int = 42,
25+
log_prob: bool = False,
2526
**generation_kwargs,
2627
):
2728
"""
@@ -41,7 +42,7 @@ def __init__(
4142
self.verbose = verbose
4243
self.n_examples_shown = n_examples_shown
4344
self.generation_kwargs = generation_kwargs
44-
self.log_prob = False
45+
self.log_prob = log_prob
4546
self.rng = random.Random(seed)
4647

4748
async def __call__(

0 commit comments

Comments
 (0)