Skip to content

Commit 52e2424

Browse files
committed
updata test to use new test parameters as per #102
1 parent d2d7c3d commit 52e2424

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/unit/dataset_classes/testTox21MolNet.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,11 @@ def test_setup_processed_with_group_split(
126126
mock_torch_save (MagicMock): Mocked `torch.save` function to avoid actual file writes.
127127
mock_load_file (MagicMock): Mocked `_load_data_from_file` to provide custom data.
128128
"""
129-
self.data_module.train_split = 0.5
129+
# self.data_module.train_split = 0.5
130+
# To get the train split as 50%, set test and validation splits to 25% each
131+
# Refer: https://github.com/ChEB-AI/python-chebai/pull/102
132+
self.data_module.test_split = 0.25
133+
self.data_module.validation_split = 0.25
130134
self.data_module.setup_processed()
131135

132136
# Verify if torch.save was called for each split

0 commit comments

Comments
 (0)