Skip to content

GitLab

Unverified Commit f2e90bcb authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Fix feature extraction utils import (#14515)

parent 6c4d688f
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
src/transformers/__init__.py
View file @ f2e90bcb
...@@ -94,7 +94,8 @@ _import_structure = { ...@@ -94,7 +94,8 @@ _import_structure = {
"DataCollatorWithPadding", "DataCollatorWithPadding",
"default_data_collator", "default_data_collator",
], ],
"feature_extraction_sequence_utils": ["BatchFeature", "SequenceFeatureExtractor"], "feature_extraction_sequence_utils": ["SequenceFeatureExtractor"],
"feature_extraction_utils": ["BatchFeature"],
"file_utils": [ "file_utils": [
"CONFIG_NAME", "CONFIG_NAME",
"MODEL_CARD_NAME", "MODEL_CARD_NAME",
...@@ -2059,9 +2060,10 @@ if TYPE_CHECKING: ...@@ -2059,9 +2060,10 @@ if TYPE_CHECKING:
DataCollatorWithPadding, DataCollatorWithPadding,
default_data_collator, default_data_collator,
) )
from .feature_extraction_sequence_utils import SequenceFeatureExtractor
# Feature Extractor # Feature Extractor
from .feature_extraction_utils import BatchFeature, SequenceFeatureExtractor from .feature_extraction_utils import BatchFeature
# Files and general utilities # Files and general utilities
from .file_utils import ( from .file_utils import (
......
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