Skip to content

Commit

Permalink
fix linting
Browse files Browse the repository at this point in the history
  • Loading branch information
NickyHavoc committed Feb 29, 2024
1 parent 2bfa0b6 commit 7accd30
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion tests/use_cases/qa/test_single_chunk_qa.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,5 +77,8 @@ def test_qa_with_logit_bias_for_no_answer(
# on CI, this is tokenized as "nonononono" rather than "no no no no no"
# Likely, this is because some test changes the tokenizer state to remove the whitespace
# We should fix this, but for now, I'll assert both
acceptable_answers = [" ".join([first_token] * max_tokens), first_token*max_tokens]
acceptable_answers = [
" ".join([first_token] * max_tokens),
first_token * max_tokens,
]
assert any(output.answer == a for a in acceptable_answers)

0 comments on commit 7accd30

Please sign in to comment.