@@ -93,9 +93,12 @@ def test_create_detector(gl: Groundlight):
9393 count_detector = gl .create_detector (name = name , query = query , mode = ModeEnum .COUNT , class_names = "dog" )
9494 assert str (count_detector )
9595 name = f"Test { datetime .utcnow ()} " # Need a unique name
96- multiclass_detector = gl .create_detector (name = name , query = query , mode = ModeEnum .MULTI_CLASS , class_names = ["dog" , "cat" ])
96+ multiclass_detector = gl .create_detector (
97+ name = name , query = query , mode = ModeEnum .MULTI_CLASS , class_names = ["dog" , "cat" ]
98+ )
9799 assert str (multiclass_detector )
98100
101+
99102def test_create_detector_with_pipeline_config (gl : Groundlight ):
100103 # "never-review" is a special model that always returns the same result with 100% confidence.
101104 # It's useful for testing.
@@ -790,6 +793,7 @@ def test_submit_image_query_with_empty_inspection_id(gl: Groundlight, detector:
790793 inspection_id = "" ,
791794 )
792795
796+
793797def test_binary_detector (gl : Groundlight ):
794798 """
795799 verify that we can create and submit to a binary detector
@@ -800,6 +804,7 @@ def test_binary_detector(gl: Groundlight):
800804 binary_iq = gl .submit_image_query (created_detector , "test/assets/dog.jpeg" )
801805 assert binary_iq .result .label is not None
802806
807+
803808def test_counting_detector (gl : Groundlight ):
804809 """
805810 verify that we can create and submit to a counting detector
@@ -842,4 +847,4 @@ def test_multiclass_detector(gl: Groundlight):
842847 assert created_detector is not None
843848 mc_iq = gl .submit_image_query (created_detector , "test/assets/dog.jpeg" )
844849 assert mc_iq .result .label is not None
845- assert mc_iq .result .label in class_names
850+ assert mc_iq .result .label in class_names
0 commit comments