Internal function bugfix (#296)

Considering a case where ask_async returned image queries have no result

---------

Co-authored-by: Auto-format Bot <[email protected]>

Author: brandon-wada

pyproject.toml

@@ -9,7 +9,7 @@ packages = [
     {include = "**/*.py", from = "src"},
 ]
 readme = "README.md"
-version = "0.21.0"
+version = "0.21.1"
 
 [tool.poetry.dependencies]
 # For certifi, use ">=" instead of "^" since it upgrades its "major version" every year, not really following semver

src/groundlight/internalapi.py

@@ -65,13 +65,17 @@ def iq_is_confident(iq: ImageQuery, confidence_threshold: float) -> bool:
     The only subtlety here is that currently confidence of None means
     human label, which is treated as confident.
     """
+    if not iq.result:
+        return False
     return iq.result.confidence >= confidence_threshold  # type: ignore
 
 
 def iq_is_answered(iq: ImageQuery) -> bool:
     """Returns True if the image query has a ML or human label.
     Placeholder and special labels (out of domain) have confidences exactly 0.5
     """
+    if not iq.result:
+        return False
     if (iq.result.source == Source.STILL_PROCESSING) or (iq.result.source is None):  # Should never be None
         return False
     return True

test/unit/test_internalapi.py

@@ -0,0 +1,14 @@
+from groundlight import ExperimentalApi
+from groundlight.internalapi import iq_is_answered, iq_is_confident
+
+
+def test_iq_is_confident(gl_experimental: ExperimentalApi):
+    det = gl_experimental.get_or_create_detector("Test", "test_query")
+    iq = gl_experimental.ask_async(det, image="test/assets/dog.jpeg", wait=10)
+    assert not iq_is_confident(iq, 0.9)
+
+
+def test_iq_is_answered(gl_experimental: ExperimentalApi):
+    det = gl_experimental.get_or_create_detector("Test", "test_query")
+    iq = gl_experimental.ask_async(det, image="test/assets/dog.jpeg", wait=10)
+    assert not iq_is_answered(iq)