Skip to content

Commit 9a39003

Browse files
committed
fix one more issue
1 parent 6a6b87c commit 9a39003

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

test/conftest.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -55,15 +55,17 @@ def fixture_image_query_no(gl: Groundlight, detector: Detector) -> ImageQuery:
5555

5656

5757
@pytest.fixture(name="image_query_one")
58-
def fixture_image_query_one(gl_experimental: Groundlight, detector: Detector) -> ImageQuery:
59-
iq = gl_experimental.submit_image_query(detector=detector.id, image="test/assets/dog.jpeg", human_review="NEVER")
58+
def fixture_image_query_one(gl_experimental: Groundlight, count_detector: Detector) -> ImageQuery:
59+
iq = gl_experimental.submit_image_query(
60+
detector=count_detector.id, image="test/assets/dog.jpeg", human_review="NEVER"
61+
)
6062
return iq
6163

6264

6365
@pytest.fixture(name="image_query_zero")
64-
def fixture_image_query_zero(gl_experimental: Groundlight, detector: Detector) -> ImageQuery:
66+
def fixture_image_query_zero(gl_experimental: Groundlight, count_detector: Detector) -> ImageQuery:
6567
iq = gl_experimental.submit_image_query(
66-
detector=detector.id, image="test/assets/no_dogs.jpeg", human_review="NEVER"
68+
detector=count_detector.id, image="test/assets/no_dogs.jpeg", human_review="NEVER"
6769
)
6870
return iq
6971

0 commit comments

Comments
 (0)