Internal api bugfix fix (#297)

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

Author: brandon-wada

test/integration/test_groundlight.py

@@ -20,6 +20,7 @@
     CountingResult,
     Detector,
     ImageQuery,
+    MultiClassificationResult,
     PaginatedDetectorList,
     PaginatedImageQueryList,
 )
@@ -30,7 +31,11 @@
 
 def is_valid_display_result(result: Any) -> bool:
     """Is the image query result valid to display to the user?."""
-    if not isinstance(result, BinaryClassificationResult) and not isinstance(result, CountingResult):
+    if (
+        not isinstance(result, BinaryClassificationResult)
+        and not isinstance(result, CountingResult)
+        and not isinstance(result, MultiClassificationResult)
+    ):
         return False
     if not is_valid_display_label(result.label):
         return False

test/unit/test_internalapi.py

@@ -4,11 +4,11 @@
 
 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)
+    iq = gl_experimental.ask_async(det, image="test/assets/dog.jpeg")
     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)
+    iq = gl_experimental.ask_async(det, image="test/assets/dog.jpeg")
     assert not iq_is_answered(iq)