Fix custom function

IntelPython · Sep 23, 2024 · 5986180 · 5986180
1 parent d921b1b
commit 5986180
Showing 1 changed file with 2 additions and 1 deletion.
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
     )