From b27655a7ddbb9bf3ceb2f393d0db29c17f5113c7 Mon Sep 17 00:00:00 2001 From: CoreyEWood Date: Fri, 21 Feb 2025 15:17:51 -0800 Subject: [PATCH] Always fetch detector confidence threshold if inputted value is None --- src/groundlight/client.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/groundlight/client.py b/src/groundlight/client.py index ff5a4f5a..9a55a906 100644 --- a/src/groundlight/client.py +++ b/src/groundlight/client.py @@ -1004,6 +1004,8 @@ def wait_for_confident_result( """ if isinstance(image_query, str): image_query = self.get_image_query(image_query) + + if confidence_threshold is None: confidence_threshold = self.get_detector(image_query.detector_id).confidence_threshold confidence_above_thresh = partial(iq_is_confident, confidence_threshold=confidence_threshold) # type: ignore