Fix feature extraction utils import (#14515)

huggingface · Nov 24, 2021 · 118ae19 · 118ae19
1 parent dd54843
commit 118ae19
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -94,7 +94,8 @@
         "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 @@
         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 (