From 096def1e9d0b95cf3690734730f675ae5a74c0fd Mon Sep 17 00:00:00 2001
From: Robert Sachunsky <sachunsky@informatik.uni-leipzig.de>
Date: Wed, 8 Oct 2025 15:13:13 +0200
Subject: [PATCH] mbreorder/enhancment: fix missing imports

(not sure if these models really need that, though)
---
 src/eynollah/image_enhancer.py  | 6 +++---
 src/eynollah/mb_ro_on_layout.py | 7 +++----
 tests/test_smoke.py             | 1 -
 3 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/src/eynollah/image_enhancer.py b/src/eynollah/image_enhancer.py
index 89dde16..9247efe 100644
--- a/src/eynollah/image_enhancer.py
+++ b/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)
diff --git a/src/eynollah/mb_ro_on_layout.py b/src/eynollah/mb_ro_on_layout.py
index 45db8e4..218f973 100644
--- a/src/eynollah/mb_ro_on_layout.py
+++ b/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)
diff --git a/tests/test_smoke.py b/tests/test_smoke.py
index 252213f..e2b323a 100644
--- a/tests/test_smoke.py
+++ b/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