check for YOLO

roboflow · Oct 24, 2024 · db94afd · db94afd
1 parent 74a19af
commit db94afd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference/core/workflows/core_steps/models/foundation/ultralytics/v1.py b/inference/core/workflows/core_steps/models/foundation/ultralytics/v1.py
@@ -147,7 +147,7 @@ def run(
         half_precision: bool,
         imgsz: int,
     ) -> BlockResult:
-        if "YOLO" not in locals():
+        if "YOLO" not in globals():
             raise RuntimeError(
                 "You must install ultralytics in order to use this block."
             )