@@ -53,12 +53,7 @@ def run_task(task, model):
5353
5454 # execute the run
5555 run = OpenMLRun (task_id = task .task_id , flow_id = None , dataset_id = dataset .dataset_id , model = model )
56-
57- try :
58- run .data_content , run .trace_content = _run_task_get_arffcontent (model , task , class_labels )
59- except PyOpenMLError as message :
60- run .error_message = str (message )
61- warnings .warn ("Run terminated with error: %s" % run .error_message )
56+ run .data_content , run .trace_content = _run_task_get_arffcontent (model , task , class_labels )
6257
6358 # now generate the flow
6459 flow = sklearn_to_flow (model )
@@ -131,17 +126,13 @@ def _run_task_get_arffcontent(model, task, class_labels):
131126 testX = X [test_indices ]
132127 testY = Y [test_indices ]
133128
134- try :
135- model .fit (trainX , trainY )
136-
137- if isinstance (model , BaseSearchCV ):
138- _add_results_to_arfftrace (arff_tracecontent , fold_no , model , rep_no )
139- model_classes = model .best_estimator_ .classes_
140- else :
141- model_classes = model .classes_
142- except AttributeError as e :
143- # typically happens when training a regressor on classification task
144- raise PyOpenMLError (str (e ))
129+ model .fit (trainX , trainY )
130+
131+ if isinstance (model , BaseSearchCV ):
132+ _add_results_to_arfftrace (arff_tracecontent , fold_no , model , rep_no )
133+ model_classes = model .best_estimator_ .classes_
134+ else :
135+ model_classes = model .classes_
145136
146137 ProbaY = model .predict_proba (testX )
147138 PredY = model .predict (testX )
0 commit comments