fixup

Author: matthiasschaub

sketch_map_tool/tasks.py

@@ -10,7 +10,7 @@
 from sam2.build_sam import build_sam2
 from sam2.sam2_image_predictor import SAM2ImagePredictor
 from ultralytics import YOLO
-from ultralytics_4bands import YOLO as YOLO_4
+from ultralytics_MB import YOLO as YOLO_4
 
 from sketch_map_tool import celery_app as celery
 from sketch_map_tool import get_config_value, map_generation

sketch_map_tool/upload_processing/detect_markings.py

@@ -5,7 +5,7 @@
 from PIL import Image, ImageEnhance
 from sam2.sam2_image_predictor import SAM2ImagePredictor
 from ultralytics import YOLO
-from ultralytics_4bands import YOLO as YOLO_4
+from ultralytics_MB import YOLO as YOLO_4
 
 
 def detect_markings(

tests/integration/upload_processing/test_detect_markings.py

@@ -6,7 +6,7 @@
 from sam2.build_sam import build_sam2
 from sam2.sam2_image_predictor import SAM2ImagePredictor
 from ultralytics import YOLO
-from ultralytics_4bands import YOLO as YOLO_4
+from ultralytics_MB import YOLO as YOLO_4
 
 from sketch_map_tool.config import get_config_value
 from sketch_map_tool.upload_processing.detect_markings import (