Run format

midhun-pm · midhun-pm · commit e79af237dfe4 · 2025-10-14T10:48:33.000-07:00
diff --git a/libs/labelbox/src/labelbox/data/metrics/confusion_matrix/confusion_matrix.py b/libs/labelbox/src/labelbox/data/metrics/confusion_matrix/confusion_matrix.py
@@ -96,7 +96,7 @@ def _get_metric_name(
     if _is_classification(ground_truths, predictions):
         return "classification"
 
-    return f"{int(iou*100)}pct_iou"
+    return f"{int(iou * 100)}pct_iou"
 
 
 def _is_classification(
diff --git a/libs/labelbox/src/labelbox/data/serialization/ndjson/label.py b/libs/labelbox/src/labelbox/data/serialization/ndjson/label.py
@@ -187,7 +187,7 @@ def _create_non_video_annotations(cls, label: Label):
                 yield NDMessageTask.from_common(annotation, label.data)
             else:
                 raise TypeError(
-                    f"Unable to convert object to MAL format. `{type(getattr(annotation, 'value',annotation))}`"
+                    f"Unable to convert object to MAL format. `{type(getattr(annotation, 'value', annotation))}`"
                 )
 
     @classmethod
diff --git a/libs/labelbox/src/labelbox/orm/db_object.py b/libs/labelbox/src/labelbox/orm/db_object.py
@@ -76,8 +76,7 @@ def _set_field_values(self, field_values):
                     value = value.replace(tzinfo=timezone.utc)
                 except ValueError:
                     logger.warning(
-                        "Failed to convert value '%s' to datetime for "
-                        "field %s",
+                        "Failed to convert value '%s' to datetime for field %s",
                         value,
                         field,
                     )
diff --git a/libs/labelbox/src/labelbox/schema/workflow/filter_converters.py b/libs/labelbox/src/labelbox/schema/workflow/filter_converters.py
@@ -358,7 +358,7 @@ def _handle_feature_consensus_average(
             if isinstance(annotations[0], str):
                 # Simple ID list - convert to full format (placeholder names)
                 annotation_objects = [
-                    {"name": f"Feature {i+1}", "schemaNodeId": ann_id}
+                    {"name": f"Feature {i + 1}", "schemaNodeId": ann_id}
                     for i, ann_id in enumerate(annotations)
                 ]
             else:
diff --git a/libs/labelbox/src/labelbox/schema/workflow/filter_utils.py b/libs/labelbox/src/labelbox/schema/workflow/filter_utils.py
@@ -84,27 +84,27 @@ def build_metadata_items(
     """
     if item_type == "user":
         return [
-            {key_field: item_id, "email": f"user{i+1}@example.com"}
+            {key_field: item_id, "email": f"user{i + 1}@example.com"}
             for i, item_id in enumerate(ids)
         ]
     elif item_type == "dataset":
         return [
-            {key_field: item_id, "name": f"Dataset {i+1}"}
+            {key_field: item_id, "name": f"Dataset {i + 1}"}
             for i, item_id in enumerate(ids)
         ]
     elif item_type == "annotation":
         return [
-            {"name": f"Annotation {i+1}", "schemaNodeId": item_id}
+            {"name": f"Annotation {i + 1}", "schemaNodeId": item_id}
             for i, item_id in enumerate(ids)
         ]
     elif item_type == "issue":
         return [
-            {key_field: item_id, "name": f"Issue Category {i+1}"}
+            {key_field: item_id, "name": f"Issue Category {i + 1}"}
             for i, item_id in enumerate(ids)
         ]
     else:
         return [
-            {key_field: item_id, "name": f"{item_type.title()} {i+1}"}
+            {key_field: item_id, "name": f"{item_type.title()} {i + 1}"}
             for i, item_id in enumerate(ids)
         ]
 
diff --git a/libs/labelbox/src/labelbox/schema/workflow/workflow_utils.py b/libs/labelbox/src/labelbox/schema/workflow/workflow_utils.py
@@ -402,5 +402,5 @@ def print_filters(workflow: "ProjectWorkflow") -> None:
             if isinstance(node, LogicNode):
                 logger.info(f"Filters for node {node.id} ({node.name}):")
                 for i, f in enumerate(node.get_parsed_filters()):
-                    logger.info(f"  Filter {i+1}:")
+                    logger.info(f"  Filter {i + 1}:")
                     logger.info(f"  {json.dumps(f, indent=2)}")
diff --git a/libs/labelbox/tests/conftest.py b/libs/labelbox/tests/conftest.py
@@ -688,12 +688,12 @@ def create_label():
             predictions,
         )
         upload_task.wait_until_done(sleep_time_seconds=5)
-        assert (
-            upload_task.state == AnnotationImportState.FINISHED
-        ), "Label Import did not finish"
-        assert (
-            len(upload_task.errors) == 0
-        ), f"Label Import {upload_task.name} failed with errors {upload_task.errors}"
+        assert upload_task.state == AnnotationImportState.FINISHED, (
+            "Label Import did not finish"
+        )
+        assert len(upload_task.errors) == 0, (
+            f"Label Import {upload_task.name} failed with errors {upload_task.errors}"
+        )
 
     project.create_label = create_label
     project.create_label()
diff --git a/libs/labelbox/tests/data/annotation_import/conftest.py b/libs/labelbox/tests/data/annotation_import/conftest.py
@@ -1929,12 +1929,12 @@ def model_run_with_data_rows(
         model_run_predictions,
     )
     upload_task.wait_until_done()
-    assert (
-        upload_task.state == AnnotationImportState.FINISHED
-    ), "Label Import did not finish"
-    assert (
-        len(upload_task.errors) == 0
-    ), f"Label Import {upload_task.name} failed with errors {upload_task.errors}"
+    assert upload_task.state == AnnotationImportState.FINISHED, (
+        "Label Import did not finish"
+    )
+    assert len(upload_task.errors) == 0, (
+        f"Label Import {upload_task.name} failed with errors {upload_task.errors}"
+    )
     labels = wait_for_label_processing(configured_project)
     label_ids = [label.uid for label in labels]
     model_run.upsert_labels(label_ids)
@@ -1963,12 +1963,12 @@ def model_run_with_all_project_labels(
         model_run_predictions,
     )
     upload_task.wait_until_done()
-    assert (
-        upload_task.state == AnnotationImportState.FINISHED
-    ), "Label Import did not finish"
-    assert (
-        len(upload_task.errors) == 0
-    ), f"Label Import {upload_task.name} failed with errors {upload_task.errors}"
+    assert upload_task.state == AnnotationImportState.FINISHED, (
+        "Label Import did not finish"
+    )
+    assert len(upload_task.errors) == 0, (
+        f"Label Import {upload_task.name} failed with errors {upload_task.errors}"
+    )
     labels = wait_for_label_processing(configured_project)
     label_ids = [label.uid for label in labels]
     model_run.upsert_labels(label_ids)
diff --git a/libs/labelbox/tests/data/export/conftest.py b/libs/labelbox/tests/data/export/conftest.py
@@ -462,12 +462,12 @@ def model_run_with_data_rows(
         model_run_predictions,
     )
     upload_task.wait_until_done()
-    assert (
-        upload_task.state == AnnotationImportState.FINISHED
-    ), "Label Import did not finish"
-    assert (
-        len(upload_task.errors) == 0
-    ), f"Label Import {upload_task.name} failed with errors {upload_task.errors}"
+    assert upload_task.state == AnnotationImportState.FINISHED, (
+        "Label Import did not finish"
+    )
+    assert len(upload_task.errors) == 0, (
+        f"Label Import {upload_task.name} failed with errors {upload_task.errors}"
+    )
     labels = wait_for_label_processing(configured_project_with_ontology)
     label_ids = [label.uid for label in labels]
     model_run.upsert_labels(label_ids)
diff --git a/libs/labelbox/tests/data/export/streamable/test_export_data_rows_streamable.py b/libs/labelbox/tests/data/export/streamable/test_export_data_rows_streamable.py
@@ -138,7 +138,9 @@ 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="Test times out in environments with high task volume - querying all org tasks is too slow")
+    @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()
diff --git a/libs/labelbox/tests/data/metrics/confusion_matrix/test_confusion_matrix_data_row.py b/libs/labelbox/tests/data/metrics/confusion_matrix/test_confusion_matrix_data_row.py
@@ -38,9 +38,9 @@ def test_overlapping_objects(tool_examples):
                 ).values():
                     for idx in range(4):
                         expected[idx] += expected_values[idx]
-                assert score[0].value == tuple(
-                    expected
-                ), f"{example.predictions},{example.ground_truths}"
+                assert score[0].value == tuple(expected), (
+                    f"{example.predictions},{example.ground_truths}"
+                )
 
 
 @parametrize(
@@ -59,9 +59,9 @@ def test_overlapping_classifications(tool_examples):
             for expected_values in example.expected.values():
                 for idx in range(4):
                     expected[idx] += expected_values[idx]
-            assert score[0].value == tuple(
-                expected
-            ), f"{example.predictions},{example.ground_truths}"
+            assert score[0].value == tuple(expected), (
+                f"{example.predictions},{example.ground_truths}"
+            )
 
 
 def test_partial_overlap(pair_iou_thresholds):
@@ -70,6 +70,6 @@ def test_partial_overlap(pair_iou_thresholds):
             score = confusion_matrix_metric(
                 example.predictions, example.ground_truths, iou=iou
             )
-            assert score[0].value == tuple(
-                example.expected[iou]
-            ), f"{example.predictions},{example.ground_truths}"
+            assert score[0].value == tuple(example.expected[iou]), (
+                f"{example.predictions},{example.ground_truths}"
+            )
diff --git a/libs/labelbox/tests/data/metrics/confusion_matrix/test_confusion_matrix_feature.py b/libs/labelbox/tests/data/metrics/confusion_matrix/test_confusion_matrix_feature.py
@@ -33,9 +33,9 @@ def test_overlapping_objects(tool_examples):
             if len(getattr(example, expected_attr_name)) == 0:
                 assert len(metrics) == 0
             else:
-                assert metrics == getattr(
-                    example, expected_attr_name
-                ), f"{example.predictions},{example.ground_truths}"
+                assert metrics == getattr(example, expected_attr_name), (
+                    f"{example.predictions},{example.ground_truths}"
+                )
 
 
 @parametrize(
@@ -52,6 +52,6 @@ def test_overlapping_classifications(tool_examples):
         if len(example.expected) == 0:
             assert len(metrics) == 0
         else:
-            assert (
-                metrics == example.expected
-            ), f"{example.predictions},{example.ground_truths}"
+            assert metrics == example.expected, (
+                f"{example.predictions},{example.ground_truths}"
+            )
diff --git a/libs/labelbox/tests/data/serialization/ndjson/test_relationship.py b/libs/labelbox/tests/data/serialization/ndjson/test_relationship.py
@@ -403,7 +403,9 @@ def test_source_ontology_name_relationship():
                 type=Relationship.Type.UNIDIRECTIONAL,
             ),
         )
-        assert False, "Expected ValueError for providing both source and source_ontology_name"
+        assert False, (
+            "Expected ValueError for providing both source and source_ontology_name"
+        )
     except Exception as e:
         assert (
             "Value error, Only one of 'source' or 'source_ontology_name' may be provided"
@@ -419,7 +421,9 @@ def test_source_ontology_name_relationship():
                 type=Relationship.Type.UNIDIRECTIONAL,
             ),
         )
-        assert False, "Expected ValueError for providing neither source nor source_ontology_name"
+        assert False, (
+            "Expected ValueError for providing neither source nor source_ontology_name"
+        )
     except Exception as e:
         assert (
             "Value error, Either source or source_ontology_name must be provided"
diff --git a/libs/labelbox/tests/data/test_data_row_metadata.py b/libs/labelbox/tests/data/test_data_row_metadata.py
@@ -165,9 +165,9 @@ def test_large_bulk_upsert_datarow_metadata(big_dataset, mdo):
         for metadata in mdo.bulk_export(data_row_ids)
     }
     for data_row_id in data_row_ids:
-        assert len(
-            [f for f in metadata_lookup.get(data_row_id).fields]
-        ), metadata_lookup.get(data_row_id).fields
+        assert len([f for f in metadata_lookup.get(data_row_id).fields]), (
+            metadata_lookup.get(data_row_id).fields
+        )
 
 
 def test_upsert_datarow_metadata_by_name(data_row, mdo):
@@ -179,9 +179,9 @@ def test_upsert_datarow_metadata_by_name(data_row, mdo):
         metadata.data_row_id: metadata
         for metadata in mdo.bulk_export([data_row.uid])
     }
-    assert len(
-        [f for f in metadata_lookup.get(data_row.uid).fields]
-    ), metadata_lookup.get(data_row.uid).fields
+    assert len([f for f in metadata_lookup.get(data_row.uid).fields]), (
+        metadata_lookup.get(data_row.uid).fields
+    )
 
 
 def test_upsert_datarow_metadata_option_by_name(data_row, mdo):
diff --git a/libs/labelbox/tests/integration/test_api_keys.py b/libs/labelbox/tests/integration/test_api_keys.py
@@ -15,15 +15,15 @@ def test_create_api_key_success(client):
     key_name = f"Test Key {uuid.uuid4()}"
     user_email = client.get_user().email
 
-    assert (
-        client.get_user().org_role().name == "Admin"
-    ), "User must be an admin to create API keys"
+    assert client.get_user().org_role().name == "Admin", (
+        "User must be an admin to create API keys"
+    )
 
     # Get available roles and use the first one
     available_roles = ApiKey._get_available_api_key_roles(client)
-    assert (
-        len(available_roles) > 0
-    ), "No available roles found for API key creation"
+    assert len(available_roles) > 0, (
+        "No available roles found for API key creation"
+    )
 
     # Create the API key with a short validity period
     api_key_result = client.create_api_key(
@@ -35,13 +35,13 @@ def test_create_api_key_success(client):
     )
 
     # Verify the response format
-    assert isinstance(
-        api_key_result, dict
-    ), "API key result should be a dictionary"
+    assert isinstance(api_key_result, dict), (
+        "API key result should be a dictionary"
+    )
     assert "id" in api_key_result, "API key result should contain an 'id' field"
-    assert (
-        "jwt" in api_key_result
-    ), "API key result should contain a 'jwt' field"
+    assert "jwt" in api_key_result, (
+        "API key result should contain a 'jwt' field"
+    )
 
     # Verify the JWT token format (should be a JWT string)
     jwt = api_key_result["jwt"]
diff --git a/libs/labelbox/tests/integration/test_embedding.py b/libs/labelbox/tests/integration/test_embedding.py
@@ -11,7 +11,9 @@
 from labelbox.schema.embedding import Embedding
 
 
-@pytest.mark.skip(reason="Organization has reached max limit of custom embeddings (10 per org)")
+@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
@@ -28,7 +30,9 @@ 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.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
@@ -50,7 +54,9 @@ 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)")
+@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
 
diff --git a/libs/labelbox/tests/integration/test_invite.py b/libs/labelbox/tests/integration/test_invite.py
@@ -172,9 +172,9 @@ def test_project_invite_after_project_deletion(client, dummy_email):
     assert found_invite is not None, f"Invite for {dummy_email} not found"
 
     # Verify only one project role remains
-    assert (
-        len(found_invite.project_roles) == 1
-    ), "Expected only one project role"
+    assert len(found_invite.project_roles) == 1, (
+        "Expected only one project role"
+    )
     assert found_invite.project_roles[0].project.uid == project2.uid
 
     # Cleanup
diff --git a/libs/labelbox/tests/integration/test_mmc_data_rows.py b/libs/labelbox/tests/integration/test_mmc_data_rows.py
@@ -57,7 +57,9 @@ def test_mmc(mmc_data_row):
     }
 
 
-@pytest.mark.skip(reason="Organization has reached max limit of custom embeddings (10 per org)")
+@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) == {
diff --git a/libs/labelbox/tests/integration/test_user_management.py b/libs/labelbox/tests/integration/test_user_management.py
@@ -33,9 +33,9 @@ def org_invite(client, organization, environ, queries):
     invite_limit = organization.invite_limit()
 
     if environ.value == "prod":
-        assert (
-            invite_limit.remaining > 0
-        ), "No invites available for the account associated with this key."
+        assert invite_limit.remaining > 0, (
+            "No invites available for the account associated with this key."
+        )
     elif environ.value != "staging":
         # Cannot run against local
         return
@@ -102,9 +102,9 @@ def test_org_invite(client, organization, environ, queries, org_invite):
 
     assert found_invite is not None, "Invite not found"
     org_role = found_invite.organization_role_name.lower()
-    assert (
-        org_role == role.name.lower()
-    ), "Role should be labeler. Found {org_role} "
+    assert org_role == role.name.lower(), (
+        "Role should be labeler. Found {org_role} "
+    )
 
 
 def test_cancel_invite(
diff --git a/libs/labelbox/tests/integration/test_workflow.py b/libs/labelbox/tests/integration/test_workflow.py

Original file line number	Diff line number	Diff line change
`@@ -187,7 +187,7 @@ def _create_non_video_annotations(cls, label: Label):`
`187`	`187`	`yield NDMessageTask.from_common(annotation, label.data)`
`188`	`188`	`else:`
`189`	`189`	`raise TypeError(`
`190`		- f"Unable to convert object to MAL format. `{type(getattr(annotation, 'value',annotation))}`"
	`190`	+ f"Unable to convert object to MAL format. `{type(getattr(annotation, 'value', annotation))}`"
`191`	`191`	`)`
`192`	`192`
`193`	`193`	`@classmethod`
Original file line number	Diff line number	Diff line change
`@@ -358,7 +358,7 @@ def _handle_feature_consensus_average(`
`358`	`358`	`if isinstance(annotations[0], str):`
`359`	`359`	`# Simple ID list - convert to full format (placeholder names)`
`360`	`360`	`annotation_objects = [`
`361`		`- {"name": f"Feature {i+1}", "schemaNodeId": ann_id}`
	`361`	`+ {"name": f"Feature {i + 1}", "schemaNodeId": ann_id}`
`362`	`362`	`for i, ann_id in enumerate(annotations)`
`363`	`363`	`]`
`364`	`364`	`else:`