Skip to content

Commit

Permalink
Correct linting errors
Browse files Browse the repository at this point in the history
  • Loading branch information
jdalch committed Nov 5, 2024
1 parent 08af1b8 commit 5ed6b96
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion deel/puncc/api/nonconformity_scores.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ def lac_score(
raise NotImplementedError(
"LAC nonconformity score only implemented for ndarrays"
)

# Compute and return the LAC nonconformity score
return 1 - Y_pred[np.arange(y_true.shape[0]), y_true]

Expand Down
2 changes: 1 addition & 1 deletion deel/puncc/api/prediction_sets.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ def lac_set(
prediction_sets = [
np.where(Y_pred[i] >= 1 - scores_quantile)[0].tolist() for i in range(pred_len)
]

return (prediction_sets,)


Expand Down

0 comments on commit 5ed6b96

Please sign in to comment.