Skip to content

Commit 7796f95

Browse files
committed
align custom function with skl estimator
1 parent d3f0b85 commit 7796f95

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

sklbench/benchmarks/custom_function.py

+7-3
Original file line numberDiff line numberDiff line change
@@ -64,9 +64,13 @@ def get_function_args(bench_case: BenchCase, x_train, y_train, x_test, y_test) -
6464

6565
def measure_function_instance(bench_case, function_instance, args: Tuple, kwargs: Dict):
6666
metrics = dict()
67-
metrics["time[ms]"], metrics["time std[ms]"], _ = measure_case(
68-
bench_case, function_instance, *args, **kwargs
69-
)
67+
(
68+
metrics["time[ms]"],
69+
metrics["time std[ms]"],
70+
metrics["first iter[ms]"],
71+
metrics["box filter mean[ms]"],
72+
metrics["box filter std[ms]"],
73+
) = measure_case(bench_case, function_instance, *args, **kwargs)
7074
return metrics
7175

7276

0 commit comments

Comments
 (0)