Fix xgboost ci step

Author: olegkkruglov

sklbench/benchmarks/sklearn_estimator.py

@@ -505,7 +505,7 @@ def function(*args, **kwargs):
 
         return data_descriptor
 
-    return function()
+    return function
 
 
 def measure_sklearn_estimator(

sklbench/datasets/special_params.py

@@ -213,7 +213,7 @@ def assign_case_special_values_on_run(
         and library == "xgboost"
         and estimator == "XGBClassifier"
     ):
-        y_train = convert_to_numpy(data[1])
+        y_train = convert_to_numpy(data[2])
         value_counts = pd.value_counts(y_train).sort_index()
         if len(value_counts) != 2:
             logger.info(

sklbench/datasets/transformer.py

@@ -183,6 +183,6 @@ def split_and_transform_data(dataset_info, data, device, data_description):
                 data_description[subset_name]["features"] = converted_data.shape[1]
 
     return (
-        (data_dict[name] for name in ["x_train", "x_test", "y_train", "y_test"]),
+        [data_dict[name] for name in ["x_train", "x_test", "y_train", "y_test"]],
         data_description,
     )