include confidence_threshold in submit_image_query params

groundlight · Dec 5, 2024 · 9e316a5 · 9e316a5
1 parent c29298a
commit 9e316a5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/groundlight/client.py b/src/groundlight/client.py
@@ -669,6 +669,9 @@ def submit_image_query(  # noqa: PLR0913 # pylint: disable=too-many-arguments, t
         if patience_time is not None:
             params["patience_time"] = patience_time
 
+        if confidence_threshold is not None:
+            params["confidence_threshold"] = confidence_threshold
+
         if human_review is not None:
             params["human_review"] = human_review