fix: guard against no circles found

agrc · Feb 1, 2023 · a4bb1ce · a4bb1ce
1 parent adb74e9
commit a4bb1ce
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/row.py b/row.py
@@ -338,6 +338,11 @@ def export_circles_from_image(circles, out_dir, file_name, cv2_image, height, wi
     Returns:
         list: a list of cv2 images
     """
+    if circles is None:
+        logging.warning("No circles detected for %s", file_name)
+
+        return []
+
     #: round the values to the nearest integer
     circles = np.uint16(np.around(circles))