From 7accd30fb7c095d9e29d9335dec36b6cade6cdba Mon Sep 17 00:00:00 2001 From: "niklas.finken" Date: Thu, 29 Feb 2024 10:47:42 +0100 Subject: [PATCH] fix linting --- tests/use_cases/qa/test_single_chunk_qa.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/use_cases/qa/test_single_chunk_qa.py b/tests/use_cases/qa/test_single_chunk_qa.py index 19b3fa467..752bdc54c 100644 --- a/tests/use_cases/qa/test_single_chunk_qa.py +++ b/tests/use_cases/qa/test_single_chunk_qa.py @@ -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)