Skip to content

GitLab

Unverified Commit 032d63b9 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix dummy creation script (#17304)

parent 986dd5c5
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
utils/check_dummies.py
View file @ 032d63b9
...@@ -26,7 +26,7 @@ PATH_TO_TRANSFORMERS = "src/transformers" ...@@ -26,7 +26,7 @@ PATH_TO_TRANSFORMERS = "src/transformers"
_re_backend = re.compile(r"is\_([a-z_]*)_available()") _re_backend = re.compile(r"is\_([a-z_]*)_available()")
# Matches from xxx import bla # Matches from xxx import bla
_re_single_line_import = re.compile(r"\s+from\s+\S*\s+import\s+([^\(\s].*)\n") _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+is\_[a-z]*\_available\(\)") _re_test_backend = re.compile(r"^\s+if\s+not\s+is\_[a-z]*\_available\(\)")
DUMMY_CONSTANT = """ DUMMY_CONSTANT = """
...@@ -73,6 +73,8 @@ def read_init(): ...@@ -73,6 +73,8 @@ def read_init():
# If the line is an if is_backend_available, we grab all objects associated. # If the line is an if is_backend_available, we grab all objects associated.
backend = find_backend(lines[line_index]) backend = find_backend(lines[line_index])
if backend is not None: if backend is not None:
while not lines[line_index].startswith(" else:"):
line_index += 1
line_index += 1 line_index += 1
objects = [] objects = []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment