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