Skip to content

Commit 8897829

Browse files
authored
Fix failing model CI tests (#189)
1 parent 18f7ccd commit 8897829

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

tests/modelci/test_unit_test.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,7 @@
99
)
1010

1111

12-
@pytest.mark.skip("Assigned slice has no 2D annotations")
13-
def test_unit_test_metric_creation(CLIENT, unit_test):
12+
def test_unit_test_metric_creation(CLIENT, annotations, unit_test):
1413
# create some dataset_items for the unit test to reference
1514
iou = CLIENT.modelci.eval_functions.bbox_iou
1615
unit_test_metric = unit_test.add_criterion(iou() > EVAL_FUNCTION_THRESHOLD)
@@ -24,8 +23,7 @@ def test_unit_test_metric_creation(CLIENT, unit_test):
2423
assert unit_test_metric in criteria
2524

2625

27-
@pytest.mark.skip("Assigned slice has no 2D annotations")
28-
def test_list_unit_test(CLIENT, test_slice):
26+
def test_list_unit_test(CLIENT, test_slice, annotations):
2927
test_name = "unit_test_" + get_uuid() # use uuid to make unique
3028

3129
e = CLIENT.modelci.eval_functions
@@ -59,7 +57,7 @@ def test_unit_test_items(CLIENT, test_slice, slice_items, annotations):
5957
CLIENT.modelci.delete_unit_test(unit_test.id)
6058

6159

62-
def test_no_criteria_raises_error(CLIENT, test_slice):
60+
def test_no_criteria_raises_error(CLIENT, test_slice, annotations):
6361
test_name = "unit_test_" + get_uuid() # use uuid to make unique
6462
with pytest.raises(CreateUnitTestError):
6563
CLIENT.modelci.create_unit_test(

0 commit comments

Comments
 (0)