mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-14 20:58:08 +00:00
Fix dummy creation script (#17304)
This commit is contained in:
parent
986dd5c5bf
commit
032d63b976
1 changed files with 3 additions and 1 deletions
|
|
@ -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+is\_[a-z]*\_available\(\)")
|
||||
_re_test_backend = re.compile(r"^\s+if\s+not\s+is\_[a-z]*\_available\(\)")
|
||||
|
||||
|
||||
DUMMY_CONSTANT = """
|
||||
|
|
@ -73,6 +73,8 @@ def read_init():
|
|||
# If the line is an if is_backend_available, we grab all objects associated.
|
||||
backend = find_backend(lines[line_index])
|
||||
if backend is not None:
|
||||
while not lines[line_index].startswith(" else:"):
|
||||
line_index += 1
|
||||
line_index += 1
|
||||
|
||||
objects = []
|
||||
|
|
|
|||
Loading…
Reference in a new issue