MAINT Hide train_test_split from nose test discovery. (#10223)

scikit-learn · Dec 6, 2017 · ea6ea81 · ea6ea81
1 parent 00878d0
commit ea6ea81
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sklearn/model_selection/_split.py b/sklearn/model_selection/_split.py
@@ -2059,6 +2059,9 @@ def train_test_split(*arrays, **options):
                                      safe_indexing(a, test)) for a in arrays))
 
 
+# Tell nose that train_test_split is not a test
+train_test_split.__test__ = False
+
 def _build_repr(self):
     # XXX This is copied from BaseEstimator's get_params
     cls = self.__class__