Skip to content

Commit 4ac993c

Browse files
committed
create ROI correctly
1 parent 4de8352 commit 4ac993c

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

test/unit/test_labels.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,7 @@
22

33
import pytest
44
from groundlight import ApiException, ExperimentalApi
5-
from groundlight_openapi_client.model.b_box_geometry import BBoxGeometry
6-
from groundlight_openapi_client.model.roi import ROI
5+
from model import BBoxGeometryRequest, ROIRequest
76

87

98
def test_binary_labels(gl_experimental: ExperimentalApi):
@@ -78,7 +77,7 @@ def test_bounding_box_labels(gl_experimental: ExperimentalApi):
7877
gl_experimental.add_label(
7978
iq1,
8079
"BOUNDING_BOX",
81-
rois=[ROI(label="test_class", geometry=BBoxGeometry(left=0.1, top=0.1, right=0.6, bottom=0.6))],
80+
rois=[gl_experimental.create_roi(label="test_class", top_left=(0.1, 0.1), bottom_right=(0.6, 0.6))],
8281
)
8382
iq1 = gl_experimental.get_image_query(iq1.id)
8483
assert iq1.result.label == "BOUNDING_BOX"

0 commit comments

Comments
 (0)