We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d2d7c3d commit 52e2424Copy full SHA for 52e2424
tests/unit/dataset_classes/testTox21MolNet.py
@@ -126,7 +126,11 @@ def test_setup_processed_with_group_split(
126
mock_torch_save (MagicMock): Mocked `torch.save` function to avoid actual file writes.
127
mock_load_file (MagicMock): Mocked `_load_data_from_file` to provide custom data.
128
"""
129
- self.data_module.train_split = 0.5
+ # 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
134
self.data_module.setup_processed()
135
136
# Verify if torch.save was called for each split
0 commit comments