rye fmt

Author: kvilon

libs/labelbox/src/labelbox/data/annotation_types/relationship.py

@@ -31,13 +31,17 @@ def check_readonly(self):
     @model_validator(mode="after")
     def validate_source_fields(self):
         if self.source is None and self.source_ontology_name is None:
-            raise ValueError("Either source or source_ontology_name must be provided")
+            raise ValueError(
+                "Either source or source_ontology_name must be provided"
+            )
         return self
 
     @model_validator(mode="after")
     def validate_source_consistency(self):
         if self.source is not None and self.source_ontology_name is not None:
-            raise ValueError("Only one of 'source' or 'source_ontology_name' may be provided")
+            raise ValueError(
+                "Only one of 'source' or 'source_ontology_name' may be provided"
+            )
         return self
 
 

libs/labelbox/src/labelbox/data/serialization/ndjson/label.py

@@ -230,7 +230,10 @@ def _create_relationship_annotations(
                             f"Unable to create relationship with invalid source. For PDF targets, "
                             f"source must be ObjectAnnotation or ClassificationAnnotation. Got: {type(source)}"
                         )
-                    if source is None and annotation.value.source_ontology_name is None:
+                    if (
+                        source is None
+                        and annotation.value.source_ontology_name is None
+                    ):
                         raise ValueError(
                             "Unable to create relationship - either source or source_ontology_name must be provided"
                         )

libs/labelbox/src/labelbox/data/serialization/ndjson/relationship.py

@@ -53,7 +53,9 @@ def from_common(
             name=annotation.name,
             dataRow=DataRow(id=data.uid, global_key=data.global_key),
             relationship=_Relationship(
-                source=str(relationship.source._uuid) if relationship.source else None,
+                source=str(relationship.source._uuid)
+                if relationship.source
+                else None,
                 target=str(relationship.target._uuid),
                 sourceOntologyName=relationship.source_ontology_name,
                 type=relationship.type.value,

libs/labelbox/tests/data/annotation_import/test_relationships.py

@@ -437,7 +437,10 @@ def test_relationship_missing_source_validation():
         value=TextEntity(start=30, end=35),
     )
 
-    with pytest.raises(ValueError, match="Either source or source_ontology_name must be provided"):
+    with pytest.raises(
+        ValueError,
+        match="Either source or source_ontology_name must be provided",
+    ):
         RelationshipAnnotation(
             name="rel",
             value=Relationship(
@@ -458,7 +461,10 @@ def test_relationship_both_sources_validation():
         value=TextEntity(start=30, end=35),
     )
 
-    with pytest.raises(ValueError, match="Only one of 'source' or 'source_ontology_name' may be provided"):
+    with pytest.raises(
+        ValueError,
+        match="Only one of 'source' or 'source_ontology_name' may be provided",
+    ):
         RelationshipAnnotation(
             name="rel",
             value=Relationship(

libs/labelbox/tests/data/serialization/ndjson/test_relationship.py

@@ -383,7 +383,8 @@ def test_source_ontology_name_relationship():
 
     # Verify relationship with source_ontology_name
     assert (
-        ontology_rel_serialized["relationship"]["sourceOntologyName"] == "Person"
+        ontology_rel_serialized["relationship"]["sourceOntologyName"]
+        == "Person"
     )
     assert (
         ontology_rel_serialized["relationship"]["target"]
@@ -404,7 +405,10 @@ def test_source_ontology_name_relationship():
         )
         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" in str(e)
+        assert (
+            "Value error, Only one of 'source' or 'source_ontology_name' may be provided"
+            in str(e)
+        )
 
     # Test that providing neither source nor source_ontology_name raises an error
     try:
@@ -417,4 +421,7 @@ def test_source_ontology_name_relationship():
         )
         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" in str(e)
+        assert (
+            "Value error, Either source or source_ontology_name must be provided"
+            in str(e)
+        )