Fix feature extraction utils import (#14515)

f2e90bcb · Lysandre Debut · GitHub · 6c4d688f · f2e90bcb
Unverified Commit f2e90bcb authored Nov 24, 2021 by Lysandre Debut Committed by GitHub Nov 24, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

src/transformers/__init__.py src/transformers/__init__.py +4 -2

No files found.
--- a/src/transformers/__init__.py
+++ b/src/transformers/__init__.py
@@ -94,7 +94,8 @@ _import_structure = {
        "DataCollatorWithPadding",
        "default_data_collator",
    ],
-    "feature_extraction_sequence_utils": ["BatchFeature", "SequenceFeatureExtractor"],
+    "feature_extraction_sequence_utils": ["SequenceFeatureExtractor"],
+    "feature_extraction_utils": ["BatchFeature"],
    "file_utils": [
        "CONFIG_NAME",
        "MODEL_CARD_NAME",
@@ -2059,9 +2060,10 @@ if TYPE_CHECKING:
        DataCollatorWithPadding,
        default_data_collator,
    )
+    from .feature_extraction_sequence_utils import SequenceFeatureExtractor

    # Feature Extractor
-    from .feature_extraction_utils import BatchFeature, SequenceFeatureExtractor
+    from .feature_extraction_utils import BatchFeature

    # Files and general utilities
    from .file_utils import (