Fix custom function

olegkkruglov · olegkkruglov · commit 59861804cc24 · 2024-09-23T07:13:24.000-07:00
diff --git a/sklbench/benchmarks/custom_function.py b/sklbench/benchmarks/custom_function.py
@@ -77,7 +77,8 @@ def main(bench_case: BenchCase, filters: List[BenchCase]):
     function_instance = get_function_instance(library_name, function_name)
 
     # load and transform data
-    data, data_description = load_dataset(bench_case)
+    dataset_info = get_bench_case_value("data")
+    data, data_description = load_dataset(bench_case, dataset_info)
     (x_train, x_test, y_train, y_test), data_description = split_and_transform_data(
         bench_case, data, data_description
     )

Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,8 @@ def main(bench_case: BenchCase, filters: List[BenchCase]):`
`77`	`77`	`function_instance = get_function_instance(library_name, function_name)`
`78`	`78`
`79`	`79`	`# load and transform data`
`80`		`- data, data_description = load_dataset(bench_case)`
	`80`	`+ dataset_info = get_bench_case_value("data")`
	`81`	`+ data, data_description = load_dataset(bench_case, dataset_info)`
`81`	`82`	`(x_train, x_test, y_train, y_test), data_description = split_and_transform_data(`
`82`	`83`	`bench_case, data, data_description`
`83`	`84`	`)`