Move dependency to call method (#15941)

e8efaecb · NielsRogge · GitHub · 5c6f57ee · e8efaecb
Unverified Commit e8efaecb authored Mar 04, 2022 by NielsRogge Committed by GitHub Mar 04, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

src/transformers/models/layoutlmv2/feature_extraction_layoutlmv2.py ...ormers/models/layoutlmv2/feature_extraction_layoutlmv2.py +1 -2

No files found.
--- a/src/transformers/models/layoutlmv2/feature_extraction_layoutlmv2.py
+++ b/src/transformers/models/layoutlmv2/feature_extraction_layoutlmv2.py
@@ -120,8 +120,6 @@ class LayoutLMv2FeatureExtractor(FeatureExtractionMixin, ImageFeatureExtractionM
        self.resample = resample
        self.apply_ocr = apply_ocr
        self.ocr_lang = ocr_lang
-        if apply_ocr:
-            requires_backends(self, "pytesseract")

    def __call__(
        self, images: ImageInput, return_tensors: Optional[Union[str, TensorType]] = None, **kwargs
@@ -200,6 +198,7 @@ class LayoutLMv2FeatureExtractor(FeatureExtractionMixin, ImageFeatureExtractionM

        # Tesseract OCR to get words + normalized bounding boxes
        if self.apply_ocr:
+            requires_backends(self, "pytesseract")
            words_batch = []
            boxes_batch = []
            for image in images: