Skip to content

Commit 334b121

Browse files
committed
Support max num results
1 parent b7a6059 commit 334b121

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/test_query.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,14 @@ def test_custom(
9191
width=1,
9292
image=high_quality_image,
9393
)
94+
target_id_3 = client.add_target(
95+
name=uuid.uuid4().hex,
96+
width=1,
97+
image=high_quality_image,
98+
)
9499
client.wait_for_target_processed(target_id=target_id)
95100
client.wait_for_target_processed(target_id=target_id_2)
101+
client.wait_for_target_processed(target_id=target_id_3)
96102
matches = cloud_reco_client.query(image=high_quality_image, max_num_results=2)
97103
assert len(matches) == 2
98104

0 commit comments

Comments
 (0)