fix dependencies.

swisstopo · Apr 4, 2024 · 3521401 · 3521401
1 parent c430f15
commit 3521401
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/stratigraphy/line_detection.py b/src/stratigraphy/line_detection.py
@@ -101,7 +101,9 @@ def draw_lines_on_page(filename: str, page: fitz.Page, geometric_lines: list[Lin
     """
     if not mlflow_tracking:
         logger.warning("MLFlow tracking is not enabled. MLFLow is required to store the images.")
-    import mlflow
 
-    img = plot_lines(page, geometric_lines, scale_factor=line_detection_params["pdf_scale_factor"])
-    mlflow.log_image(img, f"pages/{filename}_page_{page.number + 1}_lines.png")
+    else:
+        import mlflow
+
+        img = plot_lines(page, geometric_lines, scale_factor=line_detection_params["pdf_scale_factor"])
+        mlflow.log_image(img, f"pages/{filename}_page_{page.number + 1}_lines.png")