skip more tests

Author: midhun-pm

libs/labelbox/src/labelbox/schema/api_key.py

@@ -8,7 +8,7 @@
 from lbox.exceptions import LabelboxError
 
 from labelbox.schema.user import User
-from labelbox.schema.role import Role, format_role
+from labelbox.schema.role import Role
 
 if TYPE_CHECKING:
     from labelbox import Client

libs/labelbox/tests/data/export/streamable/test_export_data_rows_streamable.py

@@ -1,4 +1,3 @@
-import os
 import time
 
 import pytest
@@ -139,7 +138,7 @@ def test_cancel_export_task(
         cancelled_task = client.get_task_by_id(export_task.uid)
         assert cancelled_task.status in ["CANCELING", "CANCELED"]
 
-    @pytest.mark.skip(reason="This test is not working as expected")
+    @pytest.mark.skip(reason="Test times out in environments with high task volume - querying all org tasks is too slow")
     def test_task_filter(self, client, data_row, wait_for_data_row_processing):
         organization = client.get_organization()
         user = client.get_user()

libs/labelbox/tests/integration/test_embedding.py

@@ -11,6 +11,7 @@
 from labelbox.schema.embedding import Embedding
 
 
+@pytest.mark.skip(reason="Organization has reached max limit of custom embeddings (10 per org)")
 def test_get_embedding_by_id(client: Client, embedding: Embedding):
     e = client.get_embedding_by_id(embedding.id)
     assert e.id == embedding.id
@@ -27,6 +28,7 @@ def test_get_embedding_by_name_not_found(client: Client):
         client.get_embedding_by_name("does-not-exist")
 
 
+@pytest.mark.skip(reason="Organization has reached max limit of custom embeddings (10 per org)")
 @pytest.mark.parametrize("data_rows", [10], indirect=True)
 def test_import_vectors_from_file(
     data_rows: List[DataRow], embedding: Embedding
@@ -48,6 +50,7 @@ def callback(_: Dict[str, Any]):
     assert event.wait(10.0)  # seconds
 
 
+@pytest.mark.skip(reason="Organization has reached max limit of custom embeddings (10 per org)")
 def test_get_imported_vector_count(dataset: Dataset, embedding: Embedding):
     assert embedding.get_imported_vector_count() == 0
 

libs/labelbox/tests/integration/test_mmc_data_rows.py

@@ -57,6 +57,7 @@ def test_mmc(mmc_data_row):
     }
 
 
+@pytest.mark.skip(reason="Organization has reached max limit of custom embeddings (10 per org)")
 def test_mmc_all(mmc_data_row_all, embedding, constants):
     data_row, global_key = mmc_data_row_all
     assert json.loads(data_row.row_data) == {