model_selection: move model_selection module to experimental

basf · Oct 9, 2024 · 8a8aafa · 8a8aafa
1 parent 7b4ee34
commit 8a8aafa
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 10 deletions.
diff --git a/molpipeline/experimental/model_selection/__init__.py b/molpipeline/experimental/model_selection/__init__.py
@@ -0,0 +1,6 @@
+"""Model selection module."""
+
+from molpipeline.experimental.model_selection.splitter import GroupShuffleSplit
+
+
+__all__ = ["GroupShuffleSplit"]
diff --git a/molpipeline/model_selection/splitter.py → .../experimental/model_selection/splitter.py b/molpipeline/model_selection/splitter.py → .../experimental/model_selection/splitter.py
diff --git a/molpipeline/model_selection/__init__.py b/molpipeline/model_selection/__init__.py
diff --git a/tests/test_experimental/test_model_selection/__init__.py b/tests/test_experimental/test_model_selection/__init__.py
diff --git a/tests/test_model_selection/test_splitter.py → ...tal/test_model_selection/test_splitter.py b/tests/test_model_selection/test_splitter.py → ...tal/test_model_selection/test_splitter.py
@@ -4,10 +4,7 @@
 from numpy.testing import assert_array_equal
 from sklearn.model_selection import GroupShuffleSplit
 
-from molpipeline import Pipeline
-from molpipeline.any2mol import AutoToMol
-from molpipeline.model_selection.splitter import GroupShuffleSplit
-from molpipeline.mol2any.mol2bool import MolToBool
+from molpipeline.experimental.model_selection.splitter import GroupShuffleSplit
 
 
 class TestSampleBasedGroupShuffleSplit(unittest.TestCase):