fix

huggingface · Dec 7, 2023 · 4ea1f4d · 4ea1f4d
1 parent 21114e8
commit 4ea1f4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]},