mbreorder/enhancment: fix missing imports

(not sure if these models really need that, though)

src/eynollah/image_enhancer.py

@@ -6,23 +6,23 @@ from logging import Logger
 import os
 import time
 from typing import Optional
-import atexit
-from functools import partial
 from pathlib import Path
-from multiprocessing import cpu_count
 import gc
+
 import cv2
 import numpy as np
 from ocrd_utils import getLogger, tf_disable_interactive_logs
 import tensorflow as tf
 from skimage.morphology import skeletonize
 from tensorflow.keras.models import load_model
+
 from .utils.resize import resize_image
 from .utils.pil_cv2 import pil2cv
 from .utils import (
     is_image_filename,
     crop_image_inside_box
 )
+from .eynollah import PatchEncoder, Patches
 
 DPI_THRESHOLD = 298
 KERNEL = np.ones((5, 5), np.uint8)

src/eynollah/mb_ro_on_layout.py

@@ -6,25 +6,24 @@ from logging import Logger
 import os
 import time
 from typing import Optional
-import atexit
-from functools import partial
 from pathlib import Path
-from multiprocessing import cpu_count
 import xml.etree.ElementTree as ET
+
 import cv2
 import numpy as np
 from ocrd_utils import getLogger
 import statistics
 import tensorflow as tf
 from tensorflow.keras.models import load_model
-from .utils.resize import resize_image
 
+from .utils.resize import resize_image
 from .utils.contour import (
     find_new_features_of_contours,
     return_contours_of_image,
     return_parent_contours,
 )
 from .utils import is_xml_filename
+from .eynollah import PatchEncoder, Patches
 
 DPI_THRESHOLD = 298
 KERNEL = np.ones((5, 5), np.uint8)

tests/test_smoke.py

@@ -2,6 +2,5 @@ def test_utils_import():
     import eynollah.utils
     import eynollah.utils.contour
     import eynollah.utils.drop_capitals
-    import eynollah.utils.drop_capitals
     import eynollah.utils.is_nan
     import eynollah.utils.rotate