format

Author: Gabefire

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

@@ -19,7 +19,7 @@ class Type(Enum):
     type: Type = Type.UNIDIRECTIONAL
     readonly: Optional[bool] = None
 
-    @model_validator(mode='after')
+    @model_validator(mode="after")
     def check_readonly(self):
         if self.readonly is True:
             warnings.warn(

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

@@ -16,6 +16,7 @@ class _Relationship(BaseModel):
     type: str
     readonly: Optional[bool] = None
 
+
 class NDRelationship(NDAnnotation):
     relationship: _Relationship
 

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

@@ -395,7 +395,10 @@ def test_relationship_readonly_explicit_true():
         value=TextEntity(start=30, end=35),
     )
 
-    with pytest.warns(UserWarning, match="Creating a relationship with readonly=True is in beta.*"):
+    with pytest.warns(
+        UserWarning,
+        match="Creating a relationship with readonly=True is in beta.*",
+    ):
         relationship = RelationshipAnnotation(
             name="rel",
             value=Relationship(

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

@@ -261,13 +261,27 @@ def test_readonly_relationships():
     )
 
     # Verify readonly relationship
-    assert readonly_rel_serialized["relationship"]["source"] == ner_source_serialized["uuid"]
-    assert readonly_rel_serialized["relationship"]["target"] == ner_target_serialized["uuid"]
+    assert (
+        readonly_rel_serialized["relationship"]["source"]
+        == ner_source_serialized["uuid"]
+    )
+    assert (
+        readonly_rel_serialized["relationship"]["target"]
+        == ner_target_serialized["uuid"]
+    )
     assert readonly_rel_serialized["relationship"]["type"] == "unidirectional"
     assert readonly_rel_serialized["relationship"]["readonly"] is True
 
     # Verify non-readonly relationship
-    assert non_readonly_rel_serialized["relationship"]["source"] == ner_source_serialized["uuid"]
-    assert non_readonly_rel_serialized["relationship"]["target"] == ner_target_serialized["uuid"]
-    assert non_readonly_rel_serialized["relationship"]["type"] == "bidirectional"
+    assert (
+        non_readonly_rel_serialized["relationship"]["source"]
+        == ner_source_serialized["uuid"]
+    )
+    assert (
+        non_readonly_rel_serialized["relationship"]["target"]
+        == ner_target_serialized["uuid"]
+    )
+    assert (
+        non_readonly_rel_serialized["relationship"]["type"] == "bidirectional"
+    )
     assert non_readonly_rel_serialized["relationship"]["readonly"] is False