22
33import pytest
44from groundlight import ApiException , ExperimentalApi
5+
56from test .retry_decorator import retry_on_failure
67
78
@@ -24,7 +25,9 @@ def test_binary_labels(gl_experimental: ExperimentalApi, detector_name: Callable
2425
2526@retry_on_failure ()
2627def test_counting_labels (gl_experimental : ExperimentalApi , detector_name : Callable ):
27- det = gl_experimental .create_counting_detector (detector_name ("Test counting labels" ), "test_query" , "test_object_class" )
28+ det = gl_experimental .create_counting_detector (
29+ detector_name ("Test counting labels" ), "test_query" , "test_object_class"
30+ )
2831 iq1 = gl_experimental .submit_image_query (det , "test/assets/cat.jpeg" )
2932
3033 gl_experimental .add_label (iq1 , 0 )
@@ -71,7 +74,9 @@ def test_multiclass_labels(gl_experimental: ExperimentalApi, detector_name: Call
7174
7275@retry_on_failure ()
7376def test_bounding_box_labels (gl_experimental : ExperimentalApi , detector_name : Callable ):
74- det = gl_experimental .create_bounding_box_detector (detector_name ("Test bounding box labels" ), "test_query" , "test_class" )
77+ det = gl_experimental .create_bounding_box_detector (
78+ detector_name ("Test bounding box labels" ), "test_query" , "test_class"
79+ )
7580 iq1 = gl_experimental .submit_image_query (det , "test/assets/cat.jpeg" )
7681 gl_experimental .add_label (iq1 , "NO_OBJECTS" )
7782 iq1 = gl_experimental .get_image_query (iq1 .id )
0 commit comments