ENH: correcting big misktake in ltt

scikit-learn-contrib · Jul 12, 2023 · 71b4c67 · 71b4c67
1 parent 8a1d21d
commit 71b4c67
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mapie/control_risk/ltt.py b/mapie/control_risk/ltt.py
@@ -67,9 +67,10 @@ def ltt_procedure(
  + " controlling precision with LTT. "
  )
  p_values = compute_hoeffdding_bentkus_p_value(r_hat, n_obs, alpha_np)
+ N = len(p_values)
  valid_index = []
  for i in range(len(alpha_np)):
- l_index = np.where(p_values[:, i] <= delta/n_obs)[0].tolist()
+ l_index = np.where(p_values[:, i] <= delta/N)[0].tolist()
  valid_index.append(l_index)
  return valid_index, p_values