Fix use metric_kwargs in check_array_api_metric_pairwise (scikit-lear…

…n#29045)
neurodata · May 20, 2024 · 3a023b0 · 3a023b0
1 parent 18c1972
commit 3a023b0
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/sklearn/metrics/tests/test_common.py b/sklearn/metrics/tests/test_common.py
@@ -1898,12 +1898,24 @@ def check_array_api_metric_pairwise(metric, array_namespace, device, dtype_name)
  if "dense_output" in signature(metric).parameters:
  metric_kwargs["dense_output"] = False
  check_array_api_metric(
- metric, array_namespace, device, dtype_name, a_np=X_np, b_np=Y_np
+ metric,
+ array_namespace,
+ device,
+ dtype_name,
+ a_np=X_np,
+ b_np=Y_np,
+ **metric_kwargs,
  )
  metric_kwargs["dense_output"] = True
 
  check_array_api_metric(
- metric, array_namespace, device, dtype_name, a_np=X_np, b_np=Y_np
+ metric,
+ array_namespace,
+ device,
+ dtype_name,
+ a_np=X_np,
+ b_np=Y_np,
+ **metric_kwargs,
  )