From 451df725d6a34b36be5d0c33c18cc98fb6cf9c31 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Wed, 21 Sep 2022 11:41:45 -0400 Subject: [PATCH] Fix dummy creation for multi-frameworks objects (#19144) --- utils/check_dummies.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/check_dummies.py b/utils/check_dummies.py index 484aac2545..6e03bfed8c 100644 --- a/utils/check_dummies.py +++ b/utils/check_dummies.py @@ -26,7 +26,7 @@ PATH_TO_TRANSFORMERS = "src/transformers" _re_backend = re.compile(r"is\_([a-z_]*)_available()") # Matches from xxx import bla _re_single_line_import = re.compile(r"\s+from\s+\S*\s+import\s+([^\(\s].*)\n") -_re_test_backend = re.compile(r"^\s+if\s+not\s+is\_[a-z_]*\_available\(\)") +_re_test_backend = re.compile(r"^\s+if\s+not\s+\(?is\_[a-z_]*\_available\(\)") DUMMY_CONSTANT = """