Update _h_selection.py

rmj3197 · May 29, 2024 · 6dcdd3f · 6dcdd3f
1 parent de6b015
commit 6dcdd3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/QuadratiK/kernel_test/_h_selection.py b/QuadratiK/kernel_test/_h_selection.py
@@ -254,7 +254,7 @@ def _objective_two_sample(
         random_state=np.random.default_rng(random_state),
     )
 
-    statistic = stat_two_sample(xnew, ynew, h, np.repeat(0,d), np.diag(d))
+    statistic = stat_two_sample(xnew, ynew, h, np.repeat(0,d), np.eye(d))
     cv = cv_twosample(
         num_iter,
         quantile,