Fix tuple sort-order in test
@@ -34,9 +34,9 @@ def test_align_word_pieces(spacy_tokens, wp_tokens, expected_alignment):
@pytest.mark.parametrize(
"lengths,max_words,expected",
[
- ([1, 2, 2, 4], 4, [[3], [1, 2], [0]]),
+ ([1, 2, 2, 4], 4, [[3], [2], [0, 1]]),
([1, 2, 2, 4], 3, [[3], [2], [1], [0]]),
- ([4, 2, 2, 1], 4, [[0], [1, 2], [3]]),
+ ([4, 2, 2, 1], 4, [[0], [2], [1, 3]]),
([4, 4, 2, 2, 1], 6, [[1], [0], [2, 3, 4]]),
([10, 7, 2, 2, 1], 10, [[0], [1], [2, 3, 4]]),
],
