From 4ea1f4dec59faf6c30e549c459769b0013dc7367 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Thu, 7 Dec 2023 13:23:59 +0100 Subject: [PATCH] fix --- tests/pipelines/test_pipelines_fill_mask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/pipelines/test_pipelines_fill_mask.py b/tests/pipelines/test_pipelines_fill_mask.py index 125902cd2c6e1e..5bf77fe43db738 100644 --- a/tests/pipelines/test_pipelines_fill_mask.py +++ b/tests/pipelines/test_pipelines_fill_mask.py @@ -223,7 +223,7 @@ def run_large_test(self, unmasker): ) simplified = nested_simplify(outputs, decimals=6) self.assertEqual( - [{"sequence", x["sequence"][:100]} for x in simplified], + [{"sequence": x["sequence"][:100]} for x in simplified], [ {"sequence": f"My name is,{dummy_str}"[:100]}, {"sequence": f"My name is:,{dummy_str}"[:100]},