Skip to content

Commit

Permalink
FIX: lint for ill-sorted predictions warnings in test_utils.py
Browse files Browse the repository at this point in the history
  • Loading branch information
LacombeLouis committed Feb 7, 2024
1 parent 02fc562 commit adc6c99
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mapie/tests/test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -230,7 +230,7 @@ def test_valid_verbose(verbose: Any) -> None:
def test_initial_low_high_pred() -> None:
"""Test lower/upper predictions of the quantiles regression crossing"""
y_preds = np.array([[4, 5, 2], [4, 4, 4], [2, 3, 4]])
with pytest.warns(UserWarning, match=r"WARNING: The predictions are ill-sorted."):
with pytest.warns(UserWarning, match=r"WARNING: The predictions are*"):
check_lower_upper_bounds(y_preds[0], y_preds[1], y_preds[2])


Expand All @@ -241,7 +241,7 @@ def test_final_low_high_pred() -> None:
)
y_pred_low = np.array([4, 7, 2])
y_pred_up = np.array([3, 3, 3])
with pytest.warns(UserWarning, match=r"WARNING: The predictions are ill-sorted."):
with pytest.warns(UserWarning, match=r"WARNING: The predictions are*"):
check_lower_upper_bounds(y_pred_low, y_pred_up, y_preds[2])


Expand Down

0 comments on commit adc6c99

Please sign in to comment.