Skip to content

Commit 8a8aafa

Browse files
model_selection: move model_selection module to experimental
1 parent 7b4ee34 commit 8a8aafa

File tree

5 files changed

+7
-10
lines changed

5 files changed

+7
-10
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
"""Model selection module."""
2+
3+
from molpipeline.experimental.model_selection.splitter import GroupShuffleSplit
4+
5+
6+
__all__ = ["GroupShuffleSplit"]

molpipeline/model_selection/__init__.py

-6
This file was deleted.

tests/test_experimental/test_model_selection/__init__.py

Whitespace-only changes.

tests/test_model_selection/test_splitter.py tests/test_experimental/test_model_selection/test_splitter.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,7 @@
44
from numpy.testing import assert_array_equal
55
from sklearn.model_selection import GroupShuffleSplit
66

7-
from molpipeline import Pipeline
8-
from molpipeline.any2mol import AutoToMol
9-
from molpipeline.model_selection.splitter import GroupShuffleSplit
10-
from molpipeline.mol2any.mol2bool import MolToBool
7+
from molpipeline.experimental.model_selection.splitter import GroupShuffleSplit
118

129

1310
class TestSampleBasedGroupShuffleSplit(unittest.TestCase):

0 commit comments

Comments
 (0)