function without raise/except calling

Author: dvro

imblearn/under_sampling/instance_hardness_threshold.py

@@ -7,21 +7,23 @@
 
 import numpy as np
 from six import string_types
+import sklearn
 from sklearn.base import ClassifierMixin
 from sklearn.ensemble import RandomForestClassifier
 
 from ..base import BaseBinarySampler
 
 
 def _get_cv_splits(X, y, cv, random_state):
-    try:
+    if hasattr(sklearn, 'model_selection'):
         from sklearn.model_selection import StratifiedKFold
         cv_iterator = StratifiedKFold(
             n_splits=cv, shuffle=False, random_state=random_state).split(X, y)
-    except:
+    else:
         from sklearn.cross_validation import StratifiedKFold
         cv_iterator = StratifiedKFold(
             y, n_folds=cv, shuffle=False, random_state=random_state)
+
     return cv_iterator