"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "e135a6c93125db85b291380186b840a6b488a151"
Unverified Commit bbcd5eea authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix init import_structure sorting (#20477)

* Fix init import_structure sorting

* Fix rebase
parent 3b91f96f
This diff is collapsed.
...@@ -47,7 +47,12 @@ except OptionalDependencyNotAvailable: ...@@ -47,7 +47,12 @@ except OptionalDependencyNotAvailable:
else: else:
_import_structure["feature_extraction_speech_to_text"] = ["Speech2TextFeatureExtractor"] _import_structure["feature_extraction_speech_to_text"] = ["Speech2TextFeatureExtractor"]
if is_sentencepiece_available(): try:
if not (is_speech_available() and is_sentencepiece_available()):
raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
pass
else:
_import_structure["processing_speech_to_text"] = ["Speech2TextProcessor"] _import_structure["processing_speech_to_text"] = ["Speech2TextProcessor"]
try: try:
...@@ -96,7 +101,12 @@ if TYPE_CHECKING: ...@@ -96,7 +101,12 @@ if TYPE_CHECKING:
else: else:
from .feature_extraction_speech_to_text import Speech2TextFeatureExtractor from .feature_extraction_speech_to_text import Speech2TextFeatureExtractor
if is_sentencepiece_available(): try:
if not (is_speech_available() and is_sentencepiece_available()):
raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
pass
else:
from .processing_speech_to_text import Speech2TextProcessor from .processing_speech_to_text import Speech2TextProcessor
try: try:
......
...@@ -200,9 +200,9 @@ def sort_imports(file, check_only=True): ...@@ -200,9 +200,9 @@ def sort_imports(file, check_only=True):
indent = get_indent(block_lines[1]) indent = get_indent(block_lines[1])
# Slit the internal block into blocks of indent level 1. # Slit the internal block into blocks of indent level 1.
internal_blocks = split_code_in_indented_blocks(internal_block_code, indent_level=indent) internal_blocks = split_code_in_indented_blocks(internal_block_code, indent_level=indent)
# We have two categories of import key: list or _import_structu[key].append/extend # We have two categories of import key: list or _import_structure[key].append/extend
pattern = _re_direct_key if "_import_structure" in block_lines[0] else _re_indirect_key pattern = _re_direct_key if "_import_structure = {" in block_lines[0] else _re_indirect_key
# Grab the keys, but there is a trap: some lines are empty or jsut comments. # Grab the keys, but there is a trap: some lines are empty or just comments.
keys = [(pattern.search(b).groups()[0] if pattern.search(b) is not None else None) for b in internal_blocks] keys = [(pattern.search(b).groups()[0] if pattern.search(b) is not None else None) for b in internal_blocks]
# We only sort the lines with a key. # We only sort the lines with a key.
keys_to_sort = [(i, key) for i, key in enumerate(keys) if key is not None] keys_to_sort = [(i, key) for i, key in enumerate(keys) if key is not None]
......
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