removed get_traceble_model

Author: janvanrijn

openml/flows/sklearn_converter.py

@@ -551,27 +551,3 @@ def _deserialize_cross_validator(value, **kwargs):
 
 def _format_external_version(model_package_name, model_package_version_number):
     return '%s==%s' % (model_package_name, model_package_version_number)
-
-
-def get_traceble_model(model):
-    '''
-    Returns whether the model can produce an openml trace. If yes,
-    return that model. Returns false otherwise.
-    Clause holds true for instances of BaseSearchCV and Pipelines that
-    contain exactly one BaseSearchCV (no use to search deeper in the
-    Tree, maybe later)
-    '''
-    if isinstance(model, sklearn.model_selection._search.BaseSearchCV):
-        return True
-    count = 0
-    returnValue = None
-    if isinstance(model, sklearn.pipeline.Pipeline):
-        for step in model.steps:
-            if isinstance(step[1], sklearn.model_selection._search.BaseSearchCV):
-                returnValue = step[1]
-                count += 1
-    if count == 1:
-        return returnValue
-    else:
-        return False
-

openml/runs/functions.py

@@ -9,7 +9,6 @@
 from ..exceptions import PyOpenMLError
 from .. import config
 from ..flows import sklearn_to_flow, get_flow, flow_exists
-from ..flows.sklearn_converter import get_traceble_model
 from ..setups import setup_exists
 from ..exceptions import OpenMLCacheException, OpenMLServerException
 from ..util import URLError
@@ -169,7 +168,7 @@ def _run_task_get_arffcontent(model, task, class_labels):
             try:
                 model_fold.fit(trainX, trainY)
 
-                if get_traceble_model(model_fold):
+                if isinstance(model_fold, sklearn.model_selection._search.BaseSearchCV):
                     arff_tracecontent.extend(_extract_arfftrace(model_fold, rep_no, fold_no))
                     model_classes = model_fold.best_estimator_.classes_
                 else:
@@ -190,7 +189,7 @@ def _run_task_get_arffcontent(model, task, class_labels):
             fold_no = fold_no + 1
         rep_no = rep_no + 1
 
-    if get_traceble_model(model):
+    if isinstance(model_fold, sklearn.model_selection._search.BaseSearchCV):
         # arff_tracecontent is already set
         arff_trace_attributes = _extract_arfftrace_attributes(model_fold)
     else: