Skip to content

Commit c5339ac

Browse files
committed
for aug data splits should be consistent with original data
1 parent 40af1bc commit c5339ac

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

chebai/preprocessing/datasets/chebi.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,12 @@ def __init__(
146146
int(aug_smiles_variations) > 0
147147
), "Number of variations must be greater than 0"
148148
aug_smiles_variations = int(aug_smiles_variations)
149+
150+
if not kwargs.get("splits_file_path", None):
151+
raise ValueError(
152+
"When using SMILES augmentation, a splits_file_path must be provided to ensure consistent splits."
153+
)
154+
149155
reader_kwargs = kwargs.get("reader_kwargs", {})
150156
reader_kwargs["canonicalize_smiles"] = False
151157
kwargs["reader_kwargs"] = reader_kwargs

0 commit comments

Comments
 (0)