bart

Author: xadupre

_unittests/ut_tasks/test_tasks.py

@@ -207,6 +207,7 @@ def test_fill_mask(self):
             )
 
     @hide_stdout()
+    @requires_transformers("4.53.99")
     def test_feature_extraction_bart_base(self):
         mid = "facebook/bart-base"
         data = get_untrained_model_with_inputs(mid, verbose=1, add_second_input=True)

onnx_diagnostic/torch_export_patches/onnx_export_errors.py

@@ -33,7 +33,7 @@ def get_patches(mod, verbose: int = 0) -> Tuple[str, List[Any]]:
                 doc = v.__doc__.lstrip()
                 if doc.startswith("manual patch"):
                     continue
-                reg = re.compile("[[]patch:([a-z_A-Z.]+)[]]")
+                reg = re.compile("[\\[]patch:([a-z_A-Z.]+)[\\]]")
                 fall = reg.findall(doc)
                 assert (
                     len(fall) == 1