improved logging

src/stratigraphy/main.py

@@ -208,6 +208,7 @@ def start_pipeline(
                                 )
                                 mlflow.log_image(img, f"pages/{filename}_page_{page.number + 1}_lines.png")
 
+    logger.info("Writing predictions to JSON file %s", predictions_path)
     with open(predictions_path, "w") as file:
         file.write(json.dumps(predictions))
 

src/stratigraphy/util/coordinate_extraction.py

@@ -295,6 +295,7 @@ def extract_coordinates(self) -> Coordinate | None:
                     if coordinate.is_valid():
                         return coordinate
 
+                if len(coordinate_values):
                     logger.warning(f"Could not extract valid coordinates from {coordinate_values}")
 
         logger.info("No coordinates found in this borehole profile.")

src/stratigraphy/util/draw.py

@@ -43,6 +43,7 @@ def draw_predictions(predictions: list[FilePredictions], directory: Path, out_di
     if directory.is_file():  # deal with the case when we pass a file instead of a directory
         directory = directory.parent
     for file_name, file_prediction in predictions.items():
+        logger.info("Drawing predictions for file %s", file_name)
         with fitz.Document(directory / file_name) as doc:
             for page_index, page in enumerate(doc):
                 page_number = page_index + 1