refactor: Removed unused imports

mindee · Sep 20, 2021 · 1bcbbe4 · 1bcbbe4
1 parent 6a572a0
commit 1bcbbe4
Show file tree

Hide file tree

Showing 5 changed files with 1 addition and 7 deletions.
diff --git a/doctr/models/predictor/pytorch.py b/doctr/models/predictor/pytorch.py
@@ -9,7 +9,6 @@
 from typing import List, Any, Union
 
 from doctr.io.elements import Document
-from doctr.models.preprocessor import PreProcessor
 from doctr.models.builder import DocumentBuilder
 from doctr.models.detection.predictor import DetectionPredictor
 from doctr.models.recognition.predictor import RecognitionPredictor

diff --git a/doctr/models/predictor/tensorflow.py b/doctr/models/predictor/tensorflow.py
@@ -9,7 +9,6 @@
 
 from doctr.io.elements import Document
 from doctr.utils.repr import NestedObject
-from doctr.models.preprocessor import PreProcessor
 from doctr.models.builder import DocumentBuilder
 from doctr.models.detection.predictor import DetectionPredictor
 from doctr.models.recognition.predictor import RecognitionPredictor

diff --git a/doctr/models/recognition/predictor/tensorflow.py b/doctr/models/recognition/predictor/tensorflow.py
@@ -5,7 +5,6 @@
 
 import numpy as np
 import tensorflow as tf
-from tensorflow import keras
 from typing import List, Any, Union, Tuple
 
 from doctr.utils.repr import NestedObject

diff --git a/scripts/analyze.py b/scripts/analyze.py
@@ -18,8 +18,6 @@
     gpu_devices = tf.config.experimental.list_physical_devices('GPU')
     if any(gpu_devices):
         tf.config.experimental.set_memory_growth(gpu_devices[0], True)
-else:
-    import torch
 
 
 def main(args):

diff --git a/test/common/test_models.py b/test/common/test_models.py
@@ -4,8 +4,7 @@
 import requests
 import cv2
 
-from doctr.io import reader, Document, DocumentFile
-from doctr import models
+from doctr.io import reader, DocumentFile
 from doctr.models._utils import extract_crops, extract_rcrops, get_bitmap_angle, estimate_orientation