Fixed imports.

Author: ajaysaini725

.travis.yml

@@ -32,7 +32,7 @@ install:
   - conda info -a
 
   # Replace dep1 dep2 ... with your dependencies
-  - conda create -q -n test-environment python=$TRAVIS_PYTHON_VERSION scikit-learn==0.18.0 nose=1.3.7 pandas=0.18 numpy==1.11.0
+  - conda create -q -n test-environment python=$TRAVIS_PYTHON_VERSION scikit-learn==0.18.0 nose=1.3.7 pandas=0.18
   - source activate test-environment
 
 script:

python/requirements.txt

@@ -1,3 +1,2 @@
 # This file should list any python package dependencies.
 scikit-learn==0.18.1
-numpy==1.11.0

python/spark_sklearn/grid_search.py

@@ -8,20 +8,19 @@
 from collections import Sized, Mapping, namedtuple, defaultdict, Sequence
 from functools import partial
 import warnings
+
 import numpy as np
 from scipy.stats import rankdata
 
 from sklearn.base import BaseEstimator, is_classifier, clone
+from sklearn.metrics.scorer import check_scoring
 from sklearn.model_selection import KFold, check_cv, ParameterGrid
 from sklearn.model_selection._validation import _fit_and_score
-from sklearn.utils.metaestimators import _safe_split
 from sklearn.model_selection._search import BaseSearchCV, _check_param_grid, _CVScoreTuple
-from sklearn.metrics.scorer import check_scoring
 from sklearn.utils.fixes import MaskedArray
+from sklearn.utils.metaestimators import _safe_split
 from sklearn.utils.validation import _num_samples, indexable
 
-from sklearn.externals.joblib import Parallel, delayed
-
 
 class GridSearchCV(BaseSearchCV):
     """Exhaustive search over specified parameter values for an estimator, using Spark to