[X-0] Add field definitions

Author: kozikkamil

libs/labelbox/src/labelbox/schema/project.py

@@ -83,7 +83,7 @@ def validate_labeling_parameter_overrides(
 
 
 class Project(DbObject, Updateable, Deletable):
-    """ A Project is a container that includes a labeling frontend, an ontology,
+    """A Project is a container that includes a labeling frontend, an ontology,
     datasets and labels.
 
     Attributes:
@@ -96,6 +96,8 @@ class Project(DbObject, Updateable, Deletable):
         queue_mode (string)
         auto_audit_number_of_labels (int)
         auto_audit_percentage (float)
+        is_benchmark_enabled (bool)
+        is_consensus_enabled (bool)
 
         created_by (Relationship): `ToOne` relationship to User
         organization (Relationship): `ToOne` relationship to Organization
@@ -123,6 +125,8 @@ class Project(DbObject, Updateable, Deletable):
     data_row_count = Field.Int("data_row_count")
     model_setup_complete: Field = Field.Boolean("model_setup_complete")
     upload_type: Field = Field.Enum(UploadType, "upload_type")
+    is_benchmark_enabled = Field.Boolean("is_benchmark_enabled")
+    is_consensus_enabled = Field.Boolean("is_consensus_enabled")
 
     # Relationships
     created_by = Relationship.ToOne("User", False, "created_by")

libs/labelbox/tests/integration/test_legacy_project.py

@@ -17,16 +17,14 @@ def test_project_dataset(client, rand_gen):
 
 def test_project_auto_audit_parameters(client, rand_gen):
     with pytest.raises(
-            ValueError,
-            match=
-            "quality_mode must be set instead of auto_audit_percentage or auto_audit_number_of_labels."
+        ValueError,
+        match="quality_modes must be set instead of auto_audit_percentage or auto_audit_number_of_labels.",
     ):
         client.create_project(name=rand_gen(str), auto_audit_percentage=0.5)
 
     with pytest.raises(
-            ValueError,
-            match=
-            "quality_mode must be set instead of auto_audit_percentage or auto_audit_number_of_labels."
+        ValueError,
+        match="quality_modes must be set instead of auto_audit_percentage or auto_audit_number_of_labels.",
     ):
         client.create_project(name=rand_gen(str), auto_audit_number_of_labels=2)