Skip to content

Commit 3166bb3

Browse files
committed
add error when beta is set but not dataextractor
1 parent 1c629ff commit 3166bb3

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

chebai/loss/bce_weighted.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,11 @@ def __init__(
3232
if isinstance(data_extractor, LabeledUnlabeledMixed):
3333
data_extractor = data_extractor.labeled
3434
self.data_extractor = data_extractor
35+
36+
# If beta is provided, require a data_extractor.
37+
if self.beta is not None and self.data_extractor is None:
38+
raise ValueError("When 'beta' is set, 'data_extractor' must also be set.")
39+
3540
assert (
3641
isinstance(self.data_extractor, _ChEBIDataExtractor)
3742
or self.data_extractor is None

0 commit comments

Comments
 (0)