make tests more robust (#5747)

This commit is contained in:
Stefano Fiorucci
2023-09-08 15:50:56 +02:00
committed by GitHub
parent b5987a6d8d
commit d860a5c604
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -161,7 +161,7 @@ def test_webqa_pipeline():
assert isinstance(result, dict)
assert len(result["results"]) == 1
answer = result["results"][0]
assert "Stark" in answer or "NED" in answer
assert "stark" in answer.lower() or "ned" in answer.lower()
def test_faq_pipeline_batch():
+1 -1
View File
@@ -90,4 +90,4 @@ def test_webqa_pipeline():
assert isinstance(result, dict)
assert len(result["results"]) == 1
answer = result["results"][0]
assert "Stark" in answer or "NED" in answer
assert "stark" in answer.lower() or "ned" in answer.lower()