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]},