update tests

Author: Matt Sokoloff

tests/integration/bulk_import/conftest.py

@@ -317,8 +317,8 @@ def model_run(client, rand_gen, configured_project, annotation_submit_fn,
     model_run_s = model.create_model_run(name)
 
     time.sleep(3)
-    model_run_s.upsert_labels(
-        [label.uid for label in configured_project.labels()])
+    labels = configured_project.export_labels(download=True)
+    model_run_s.upsert_labels([label['ID'] for label in labels])
     time.sleep(3)
 
     yield model_run_s

tests/integration/test_label.py

@@ -9,6 +9,7 @@
 IMG_URL = "https://picsum.photos/200/300"
 
 
+@pytest.mark.skip("Cannot query for labels created with create_label")
 def test_labels(label_pack):
     project, dataset, data_row, label = label_pack
 
@@ -55,6 +56,7 @@ def test_label_update(label_pack):
     assert label.label == "something else"
 
 
+@pytest.mark.skip("Cannot query for labels created with create_label")
 def test_label_filter_order(client, project, rand_gen):
     dataset_1 = client.create_dataset(name=rand_gen(str), projects=project)
     dataset_2 = client.create_dataset(name=rand_gen(str), projects=project)
@@ -83,6 +85,7 @@ def test_label_filter_order(client, project, rand_gen):
     project.delete()
 
 
+@pytest.mark.skip("Cannot query for labels created with create_label")
 def test_label_bulk_deletion(project, rand_gen):
     dataset = project.client.create_dataset(name=rand_gen(str),
                                             projects=project)

tests/integration/test_labeler_performance.py

@@ -1,7 +1,10 @@
 from datetime import datetime, timezone, timedelta
 import time
+import pytest
 
 
+@pytest.mark.skip(
+    "Cannot query for labeler performance for labels created with create_label")
 def test_labeler_performance(label_pack):
     project, dataset, data_row, label = label_pack
     # Sleep a bit as it seems labeler performance isn't updated immediately.

tests/integration/test_model_run.py

@@ -13,12 +13,12 @@ def test_model_run(client, configured_project_with_label, rand_gen):
     assert model_run.model_id == model.uid
     assert model_run.created_by_id == client.get_user().uid
 
-    label = list(project.labels())[0]
-    model_run.upsert_labels([label.uid])
+    label = project.export_labels(download=True)[0]
+    model_run.upsert_labels([label['ID']])
     time.sleep(3)
 
     annotation_group = next(model_run.annotation_groups())
-    assert annotation_group.label_id == label.uid
+    assert annotation_group.label_id == label['ID']
     assert annotation_group.model_run_id == model_run.uid
     assert annotation_group.data_row().uid == next(
         next(project.datasets()).data_rows()).uid