UPD: take TCO comments into account

scikit-learn-contrib · Jul 19, 2023 · aa3431d · aa3431d
1 parent 93ed05b
commit aa3431d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mapie/conformity_scores/residual_conformity_scores.py b/mapie/conformity_scores/residual_conformity_scores.py
@@ -392,7 +392,7 @@ def get_signed_conformity_scores(
  )
 
  # reconstruct array with nan and conformity scores
- complete_signed_cs = np.zeros(y_pred.shape)
+ complete_signed_cs = np.zeros_like(y_pred)
  complete_signed_cs[cal_indexes] = signed_conformity_scores
  complete_signed_cs[train_indexes] = np.full(
  (train_indexes.shape[0],),