--- a/search_model_validation.py	Tue Apr 13 18:45:35 2021 +0000
+++ b/search_model_validation.py	Tue Apr 13 22:04:06 2021 +0000
@@ -1,15 +1,27 @@
 import argparse
 import collections
+import json
+import os
+import pickle
+import sys
+import warnings
+
 import imblearn
 import joblib
-import json
 import numpy as np
-import os
 import pandas as pd
-import pickle
 import skrebate
-import sys
-import warnings
+from galaxy_ml.utils import (
+    clean_params,
+    get_cv,
+    get_main_estimator,
+    get_module,
+    get_scoring,
+    load_model,
+    read_columns,
+    SafeEval,
+    try_get_attr
+)
 from scipy.io import mmread
 from sklearn import (
     cluster,
@@ -20,21 +32,8 @@
     preprocessing,
 )
 from sklearn.exceptions import FitFailedWarning
+from sklearn.model_selection import _search, _validation
 from sklearn.model_selection._validation import _score, cross_validate
-from sklearn.model_selection import _search, _validation
-from sklearn.pipeline import Pipeline
-
-from galaxy_ml.utils import (
-    SafeEval,
-    get_cv,
-    get_scoring,
-    load_model,
-    read_columns,
-    try_get_attr,
-    get_module,
-    clean_params,
-    get_main_estimator,
-)
 
 
 _fit_and_score = try_get_attr("galaxy_ml.model_validations", "_fit_and_score")